Function TestReader_unreadHead

tars/protocol/codec/codec_test.go:544–605

(t *testing.T)

}

func TestReader_unreadHead(t *testing.T) {
	writer := NewBuffer()
	err := writer.WriteString("hello", 0)
	if err != nil {
		t.Errorf("Write buffer failed.err:%v\n", err)
	}
	err = writer.WriteUint8(1, 1)
	if err != nil {
		t.Errorf("Write buffer failed.err:%v\n", err)
	}
	err = writer.WriteFloat32(1.2, 2)
	if err != nil {
		t.Errorf("Write buffer failed.err:%v\n", err)
	}

	// string type read head
	reader := r(writer)
	wantType, wantTag := STRING1, byte(0)
	gotType, gotTag, err := reader.readHead()
	if err != nil {
		t.Errorf("Read buffer failed.err:%v\n", err)
	}
	if gotType != wantType || gotTag != wantTag {
		t.Errorf("Failed to readHead. wantType:%v, wantTag:%v, gotType:%v, gotTag:%v\n",
			wantType, wantTag, gotType, gotTag)
	}

	// string type unread head
	reader.unreadHead(gotTag)
	gotType, gotTag, err = reader.readHead()
	if err != nil {
		t.Errorf("Read head failed.err:%v\n", err)
	}
	// skip next 6 byte. 1 byte for string length, 5 byte for string itself.
	reader.Skip(6)
	if gotType != wantType || gotTag != wantTag {
		t.Errorf("Failed to readHead. wantType:%v, wantTag:%v, gotType:%v, gotTag:%v\n",
			wantType, wantTag, gotType, gotTag)
	}

	// uint8 read head
	wantType, wantTag = BYTE, byte(1)
	gotType, gotTag, err = reader.readHead()
	if err != nil {
		t.Errorf("Read buffer failed.err:%v\n", err)
	}
	if gotType != wantType || gotTag != wantTag {
		t.Errorf("Failed to readHead. wantType:%v, wantTag:%v, gotType:%v, gotTag:%v\n",
			wantType, wantTag, gotType, gotTag)
	}

	// uint8 unread head
	reader.unreadHead(gotTag)
	gotType, gotTag, err = reader.readHead()
	if err != nil {
		t.Errorf("Read head failed.err:%v\n", err)
	}
	if gotType != wantType || gotTag != wantTag {