sarama/protocol/request_test.go

package protocol

import enc "sarama/encoding"
import (
	"bytes"
	"testing"
)

var (
	requestSimple = []byte{
		0x00, 0x00, 0x00, 0x17, // msglen
		0x06, 0x66,
		0x00, 0xD2,
		0x00, 0x00, 0x12, 0x34,
		0x00, 0x08, 'm', 'y', 'C', 'l', 'i', 'e', 'n', 't',
		0x00, 0x03, 'a', 'b', 'c'}
)

type testRequestBody struct {
}

func (s *testRequestBody) key() int16 {
	return 0x666
}

func (s *testRequestBody) version() int16 {
	return 0xD2
}

func (s *testRequestBody) Encode(pe enc.PacketEncoder) error {
	return pe.PutString("abc")
}

func TestRequest(t *testing.T) {
	request := request{correlation_id: 0x1234, id: "myClient", body: new(testRequestBody)}
	testEncodable(t, "simple", &request, requestSimple)
}

// not specific to request tests, just helper functions for testing structures that
// implement the encoder or decoder interfaces that needed somewhere to live

func testEncodable(t *testing.T, name string, in enc.Encoder, expect []byte) {
	packet, err := enc.Encode(in)
	if err != nil {
		t.Error(err)
	} else if !bytes.Equal(packet, expect) {
		t.Error("Encoding", name, "failed\ngot ", packet, "\nwant", expect)
	}
}

func testDecodable(t *testing.T, name string, out enc.Decoder, in []byte) {
	err := enc.Decode(in, out)
	if err != nil {
		t.Error("Decoding", name, "failed:", err)
	}
}