1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
package kafka
import (
"bufio"
"bytes"
"context"
"fmt"
"reflect"
"testing"
"time"
)
func TestListGroupsResponseV1(t *testing.T) {
item := listGroupsResponseV1{
ErrorCode: 2,
Groups: []listGroupsResponseGroupV1{
{
GroupID: "a",
ProtocolType: "b",
},
},
}
b := bytes.NewBuffer(nil)
w := &writeBuffer{w: b}
item.writeTo(w)
var found listGroupsResponseV1
remain, err := (&found).readFrom(bufio.NewReader(b), b.Len())
if err != nil {
t.Error(err)
t.FailNow()
}
if remain != 0 {
t.Errorf("expected 0 remain, got %v", remain)
t.FailNow()
}
if !reflect.DeepEqual(item, found) {
t.Error("expected item and found to be the same")
t.FailNow()
}
}
func TestClientListGroups(t *testing.T) {
client, shutdown := newLocalClient()
defer shutdown()
topic := makeTopic()
gid := fmt.Sprintf("%s-test-group", topic)
createTopic(t, topic, 1)
defer deleteTopic(t, topic)
w := newTestWriter(WriterConfig{
Topic: topic,
})
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
err := w.WriteMessages(
ctx,
Message{
Key: []byte("key"),
Value: []byte("value"),
},
)
if err != nil {
t.Fatal(err)
}
r := NewReader(ReaderConfig{
Brokers: []string{"localhost:9092"},
Topic: topic,
GroupID: gid,
MinBytes: 10,
MaxBytes: 1000,
})
_, err = r.ReadMessage(ctx)
if err != nil {
t.Fatal(err)
}
resp, err := client.ListGroups(
ctx,
&ListGroupsRequest{},
)
if err != nil {
t.Fatal(err)
}
if resp.Error != nil {
t.Error(
"Unexpected error in response",
"expected", nil,
"got", resp.Error,
)
}
hasGroup := false
hasProtocol := false
for _, group := range resp.Groups {
if group.GroupID == gid {
hasGroup = true
if group.ProtocolType == "consumer" {
hasProtocol = true
}
break
}
}
if !hasGroup {
t.Error("Group not found in list")
}
if !hasProtocol {
t.Error("Group does not have expected protocol type")
}
}
|
