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
|
package kafka
import (
"context"
"testing"
"time"
ktesting "github.com/segmentio/kafka-go/testing"
"github.com/stretchr/testify/assert"
)
func TestClientAlterClientQuotas(t *testing.T) {
// Added in Version 2.6.0 https://issues.apache.org/jira/browse/KAFKA-7740
if !ktesting.KafkaIsAtLeast("2.6.0") {
return
}
const (
entityType = "client-id"
entityName = "my-client-id"
key = "producer_byte_rate"
value = 500000.0
)
client, shutdown := newLocalClient()
defer shutdown()
alterResp, err := client.AlterClientQuotas(context.Background(), &AlterClientQuotasRequest{
Entries: []AlterClientQuotaEntry{
{
Entities: []AlterClientQuotaEntity{
{
EntityType: entityType,
EntityName: entityName,
},
},
Ops: []AlterClientQuotaOps{
{
Key: key,
Value: value,
Remove: false,
},
},
},
},
})
if err != nil {
t.Fatal(err)
}
expectedAlterResp := AlterClientQuotasResponse{
Throttle: 0,
Entries: []AlterClientQuotaResponseQuotas{
{
Error: makeError(0, ""),
Entities: []AlterClientQuotaEntity{
{
EntityName: entityName,
EntityType: entityType,
},
},
},
},
}
assert.Equal(t, expectedAlterResp, *alterResp)
// kraft mode is slow
if ktesting.KafkaIsAtLeast("3.7.0") {
time.Sleep(3 * time.Second)
}
describeResp, err := client.DescribeClientQuotas(context.Background(), &DescribeClientQuotasRequest{
Components: []DescribeClientQuotasRequestComponent{
{
EntityType: entityType,
MatchType: 0,
Match: entityName,
},
},
})
if err != nil {
t.Fatal(err)
}
expectedDescribeResp := DescribeClientQuotasResponse{
Throttle: 0,
Error: makeError(0, ""),
Entries: []DescribeClientQuotasResponseQuotas{
{
Entities: []DescribeClientQuotasEntity{
{
EntityType: entityType,
EntityName: entityName,
},
},
Values: []DescribeClientQuotasValue{
{
Key: key,
Value: value,
},
},
},
},
}
assert.Equal(t, expectedDescribeResp, *describeResp)
}
|
