20
ctx := context.Background()
21
t.Run("HappyPath", func(t *testing.T) {
23
>
TaskID: 42,
24
>
}
25
>
sourceCluster := "test-source-cluster-" + t.Name()
26
>
targetCluster := "test-target-cluster-" + t.Name()
27
>
queueType := persistence.QueueTypeHistoryDLQ
28
>
var queueKeys []persistence.QueueKey
29
>
for i := range 3 {
30
>
queueKey := persistence.QueueKey{
31
>
QueueType: queueType,
32
>
Category: inTask.GetCategory(),
33
>
SourceCluster: sourceCluster + strconv.Itoa(i),
34
>
TargetCluster: targetCluster + strconv.Itoa(i),
35
>
}
36
>
queueKeys = append(queueKeys, queueKey)
37
>
_, err := manager.CreateQueue(ctx, &persistence.CreateQueueRequest{QueueKey: queueKey})
38
>
require.NoError(t, err)
39
>
}
40
42
>
var nextPageToken []byte
43
>
for i := int32(1); ; i++ {
44
>
res, err := listqueues.Invoke(
45
>
context.Background(),
46
>
manager,
47
>
&historyservice.ListQueuesRequest{
48
>
QueueType: int32(queueType),
49
>
PageSize: i,
50
>
NextPageToken: nextPageToken,
51
>
},
52
>
)
53
>
require.NoError(t, err)
54
>
for _, queue := range res.Queues {
55
>
listedQueueNames = append(listedQueueNames, queue.QueueName)
56
>
57
>
}
58
>
if len(res.NextPageToken) == 0 {
59
>
break
60
}
62
}
64
>
require.Contains(t, listedQueueNames, queueKey.GetQueueName())
65
>
}
66
})
67
t.Run("InvalidPageSize", func(t *testing.T) {