packet_test.go

// Copyright 2015 Tony Bai.
//
// Licensed under the Apache License, Vsion 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.

package cmpp

import (
	"encoding/binary"
	"errors"
	"io"
	"testing"
)

func TestTypeString(t *testing.T) {
	a, b, c := V30, V21, V20
	if a != 0x30 || b != 0x21 || c != 0x20 {
		t.Fatal("The value of var of Type is incorrect")
	}

	if a.String() != "cmpp30" || b.String() != "cmpp21" || c.String() != "cmpp20" {
		t.Fatal("The string presentation of var of Type is incorrect")
	}
}

func TestCommandIdString(t *testing.T) {
	id1, id2 := CMPP_CONNECT, CMPP_CONNECT_RESP

	if id1 != 0x00000001 {
		t.Fatalf("The value of CMPP_CONNECT is %d, not equal to 0x00000001\n", id1)
	}

	if id1.String() != "CMPP_CONNECT" {
		t.Fatalf("The string presentation of command id - CMPP_CONNECT is %s, not equal to %s\n",
			id1.String(),
			"CMPP_CONNECT")
	}

	if id2 != 0x80000001 {
		t.Fatalf("The value of CMPP_CONNECT is %s, not equal to 0x00000001\n", id2)
	}
	if id2.String() != "CMPP_CONNECT_RESP" {
		t.Fatalf("The string presentation of command id - CMPP_CONNECT_RESP is %s, not equal to %s\n",
			id2.String(),
			"CMPP_CONNECT_RESP")
	}

	id3, id4 := CMPP_ACTIVE_TEST, CMPP_ACTIVE_TEST_RESP

	if id3 != 0x00000008 {
		t.Fatalf("The value of CMPP_ACTIVE_TEST is %d, not equal to 0x00000008\n", id3)
	}

	if id3.String() != "CMPP_ACTIVE_TEST" {
		t.Fatalf("The string presentation of command id - CMPP_ACTIVE_TEST is %s, not equal to %s\n",
			id3.String(),
			"CMPP_ACTIVE_TEST")
	}

	if id4 != 0x80000008 {
		t.Fatalf("The value of CMPP_ACTIVE_TEST_RESP is %d, not equal to 0x80000008\n", id4)
	}

	if id4.String() != "CMPP_ACTIVE_TEST_RESP" {
		t.Fatalf("The string presentation of command id - CMPP_ACTIVE_TEST_RESP is %s, not equal to %s\n",
			id4.String(),
			"CMPP_ACTIVE_TEST_RESP")
	}

	id5, id6 := CMPP_GET_MO_ROUTE, CMPP_GET_MO_ROUTE_RESP
	if id5 != 0x00000017 {
		t.Fatalf("The value of CMPP_GET_MO_ROUTE is %d, not equal to 0x00000017\n", id5)
	}

	if id5.String() != "CMPP_GET_MO_ROUTE" {
		t.Fatalf("The string presentation of command id - CMPP_GET_MO_ROUTE is %s, not equal to %s\n",
			id5.String(),
			"CMPP_GET_MO_ROUTE")
	}

	if id6 != 0x80000017 {
		t.Fatalf("The value of CMPP_GET_MO_ROUTE_RESP is %d, not equal to 0x80000017\n", id6)
	}

	if id6.String() != "CMPP_GET_MO_ROUTE_RESP" {
		t.Fatalf("The string presentation of command id - CMPP_GET_MO_ROUTE_RESP is %s, not equal to %s\n",
			id6.String(),
			"CMPP_GET_MO_ROUTE_RESP")
	}

	id7, id8 := CommandId(0), CommandId(10)
	if id7.String() != "unknown" {
		t.Fatalf("The value of CommandId(0) String is %s, not equal to %s\n", id7.String(), "unknown")
	}

	if id8.String() != "unknown" {
		t.Fatalf("The value of CommandId(0) String is %s, not equal to %s\n", id8.String(), "unknown")
	}

	id9, id10 := CMPP_FWD_RESP, CMPP_MT_ROUTE_RESP
	if id9.String() != "CMPP_FWD_RESP" {
		t.Fatalf("The string presentation of command id - CMPP_FWD_RESP is %s, not equal to %s\n",
			id9.String(),
			"CMPP_FWD_RESP")
	}

	if id10.String() != "CMPP_MT_ROUTE_RESP" {
		t.Fatalf("The string presentation of command id - CMPP_MT_ROUTE_RESP is %s, not equal to %s\n",
			id10.String(),
			"CMPP_MT_ROUTE_RESP")
	}
}

func TestOpError(t *testing.T) {
	op := "do foo things"
	var e error = errors.New("error example for test OpError")
	err := NewOpError(e, op)

	if err.Cause() != e {
		t.Fatalf("OpError's cause: actual [%#v], wanted[%#v]\n", err.Error(), e)
	}

	if err.Op() != op {
		t.Fatalf("OpError's op : actual [%#v], wanted[%#v]\n", err.Op(), op)
	}
}

func TestPacketWriter(t *testing.T) {
	//test WriteString
	w1 := newPacketWriter(11)

	w1.WriteString("hello")
	w1.WriteString(" golang")

	s1, e := w1.Bytes()
	if e != nil {
		t.Fatalf("packetWriter's err : actual [%#v], wanted[nil]\n", e)
	}

	if string(s1) != "hello golang" {
		t.Fatalf("packetWriter's err : actual [%s], wanted[%s]\n", string(s1), "hello golang")
	}

	// test WriteInt
	var i uint16 = 0x1234
	w2 := newPacketWriter(10)
	w2.WriteInt(binary.BigEndian, i)

	s2, e := w2.Bytes()
	if e != nil {
		t.Fatalf("packetWriter's err : actual [%#v], wanted[nil]\n", e)
	}

	if s2[0] != 0x12 || s2[1] != 0x34 {
		t.Fatalf("packetWriter's err : actual [%#v], wanted[%#v]\n", s2, []byte{0x12, 0x34})
	}

	// test WriteFixedSizeString
	w3 := newPacketWriter(10)
	w3.WriteFixedSizeString("hello", 9)
	s3, e := w3.Bytes()
	if e != nil {
		t.Fatalf("packetWriter's err : actual [%#v], wanted[nil]\n", e)
	}

	if len(s3) != 9 {
		t.Fatalf("packetWriter's err : actual [%d], wanted[%d]\n", len(s3), 9)
	}

	if string(s3[:5]) != "hello" {
		t.Fatalf("packetWriter's err : actual [%s], wanted[%s]\n", string(s3[:5]), "hello")
	}

	// test WriteByte
	w4 := newPacketWriter(10)
	w4.WriteByte('h')
	w4.WriteByte('e')
	s4, e := w4.Bytes()
	if e != nil {
		t.Fatalf("packetWriter's err : actual [%#v], wanted[nil]\n", e)
	}

	if len(s4) != 2 {
		t.Fatalf("packetWriter's err : actual [%d], wanted[%d]\n", len(s4), 2)
	}

	if string(s4[:]) != "he" {
		t.Fatalf("packetWriter's err : actual [%s], wanted[%s]\n", string(s4[:]), "he")
	}
}

func TestPacketReader(t *testing.T) {
	// test ReadBytes
	s1 := []byte{'h', 'e', 'l', 'l', 'o'}
	r1 := newPacketReader(s1)

	d1 := make([]byte, 3)
	r1.ReadBytes(d1)
	if r1.Error() != nil {
		t.Fatalf("packetReader's err : actual [%#v], wanted[nil]\n", r1.Error())
	}

	if d1[0] != 'h' || d1[1] != 'e' || d1[2] != 'l' {
		t.Fatalf("packetReader 's err : actual [%#v], wanted[%#v]\n", d1, []byte{'h', 'e', 'l'})
	}

	d1 = make([]byte, 3)
	r1.ReadBytes(d1)
	if r1.Error() == nil {
		t.Fatal("packetReader's err : actual nil, wanted non-nil")
	}

	// test ReadInt
	s2 := []byte{0x12, 0x34}
	var i uint16
	r2 := newPacketReader(s2)
	r2.ReadInt(binary.BigEndian, &i)
	if r2.Error() != nil {
		t.Fatalf("packetReader's err : actual [%#v], wanted[nil]\n", r1.Error())
	}
	if i != 0x1234 {
		t.Fatalf("packetReader's err : actual [%d], wanted[%d]\n", i, 0x1234)
	}

	// test ReadByte
	s3 := []byte{'h', 'e', 'l', 'l', 'o'}
	r3 := newPacketReader(s3)
	c := r3.ReadByte()
	if c != 'h' {
		t.Fatalf("packetReader's err : actual [%c], wanted[%c]\n", c, 'h')
	}
	r3.ReadByte()
	r3.ReadByte()
	c = r3.ReadByte()
	if c != 'l' {
		t.Fatalf("packetReader's err : actual [%c], wanted[%c]\n", c, 'l')
	}
	c = r3.ReadByte()
	if c != 'o' {
		t.Fatalf("packetReader's err : actual [%c], wanted[%c]\n", c, 'o')
	}

	c = r3.ReadByte()
	if c != 0 {
		t.Fatalf("packetReader's err : actual [%x], wanted[%d]\n", c, 0)
	}

	e := r3.Error()
	if e == nil {
		t.Fatal("packetReader's err : actual nil, wanted non-nil")
	}

	oe, _ := e.(*OpError)
	if oe.Cause() != io.EOF {
		t.Fatalf("packetReader's err : actual [%#v], wanted io.EOF\n", oe)
	}

	// test ReadCString
	s4 := []byte{'h', 'e', 'l', 'l', 'o', 0, 0, 0, 0}
	r4 := newPacketReader(s4)
	d4 := r4.ReadCString(9)

	if len(d4) != 5 {
		t.Fatalf("packetReader's err : actual [%d], wanted [%d]\n", len(d4), 5)
	}
	if string(d4) != "hello" {
		t.Fatalf("packetReader's err : actual [%s], wanted [%s]\n", string(d4), "hello")
	}
}