another client test

kafka/client_test.go

@@ -1,7 +1,10 @@
 package kafka
 
-import "testing"
-import "sarama/mock"
+import (
+	"encoding/binary"
+	"sarama/mock"
+	"testing"
+)
 
 func TestSimpleClient(t *testing.T) {
 	responses := make(chan []byte, 1)
@@ -17,3 +20,27 @@ func TestSimpleClient(t *testing.T) {
 	}
 	client.Close()
 }
+
+func TestExtraBroker(t *testing.T) {
+	responses := make(chan []byte, 1)
+	mockBroker := mock.NewBroker(t, responses)
+	mockExtra := mock.NewBroker(t, make(chan []byte))
+	defer mockBroker.Close()
+	defer mockExtra.Close()
+
+	// return the extra mock as another available broker
+	response := []byte{
+		0x00, 0x00, 0x00, 0x01,
+		0x00, 0x00, 0x00, 0x01,
+		0x00, 0x09, 'l', 'o', 'c', 'a', 'l', 'h', 'o', 's', 't',
+		0x00, 0x00, 0x00, 0x00,
+		0x00, 0x00, 0x00, 0x00}
+	binary.BigEndian.PutUint32(response[19:], uint32(mockExtra.Port()))
+	responses <- response
+
+	client, err := NewClient("clientID", "localhost", mockBroker.Port())
+	if err != nil {
+		t.Fatal(err)
+	}
+	client.Close()
+}