func getTopicMetadata(t *testing.T, k *kafka.Cluster, topic string) *TopicMetadataResponse {
b := broker.New(k.AnyBroker())
defer b.Close()
respMsg, err := Metadata(topic).Fetch(b)
if err != nil {
t.Fatal(err)
}
brokers := k.Brokers()
for i := range brokers {
if respMsg.Brokers[i].Addr() != brokers[i] {
t.Fatalf("broker: expect %s but got %s", brokers[i], respMsg.Brokers[i].Addr())
}
}
if len(respMsg.TopicMetadatas) != 1 {
t.Fatalf("len(TopicMetadatas): expect 1 but got %d", len(respMsg.TopicMetadatas))
}
meta := &respMsg.TopicMetadatas[0]
if meta.ErrorCode != NoError {
t.Fatal(meta.ErrorCode)
}
if meta.TopicName != topic {
t.Fatalf("topic: expect %s but got %s", topic, meta.TopicName)
}
return respMsg
}
func getTopicMetadata(t *testing.T, k *kafka.Cluster, topic string) *TopicMetadataResponse {
conn, err := net.Dial("tcp", k.AnyBroker())
if err != nil {
t.Fatal(err)
}
defer conn.Close()
req := &Request{
CorrelationID: rand.Int31(),
RequestMessage: &TopicMetadataRequest{topic},
}
respMsg := &TopicMetadataResponse{}
resp := &Response{ResponseMessage: respMsg}
sendReceive(t, conn, req, resp)
brokers := k.Brokers()
for i := range brokers {
if respMsg.Brokers[i].Addr() != brokers[i] {
t.Fatalf("broker: expect %s but got %s", brokers[i], respMsg.Brokers[i].Addr())
}
}
if len(respMsg.TopicMetadatas) != 1 {
t.Fatalf("len(TopicMetadatas): expect 1 but got %d", len(respMsg.TopicMetadatas))
}
meta := &respMsg.TopicMetadatas[0]
if meta.ErrorCode != NoError {
t.Fatal(meta.ErrorCode)
}
if meta.TopicName != topic {
t.Fatalf("topic: expect %s but got %s", topic, meta.TopicName)
}
return respMsg
}
func getCoord(t *testing.T, k *kafka.Cluster, group string) string {
reqMsg := GroupCoordinatorRequest(group)
req := &Request{
RequestMessage: &reqMsg,
}
respMsg := &GroupCoordinatorResponse{}
resp := &Response{ResponseMessage: respMsg}
conn, err := net.Dial("tcp", k.AnyBroker())
if err != nil {
t.Fatal(err)
}
sendReceive(t, conn, req, resp)
if respMsg.HasError() {
t.Fatal(respMsg.ErrorCode)
}
return respMsg.Broker.Addr()
}
