112
sourceCluster string,
113
targetCluster string,
115
>
t.Helper()
116
>
117
>
var nextPageToken []byte
118
>
119
>
// We want to run a test where the client makes multiple requests to the server because the client is stateful. In
120
>
// particular, the first request here should establish a connection, and the next one should reuse that connection.
121
>
for i := range numTasks {
122
>
res, err := client.GetDLQTasks(context.Background(), &historyservice.GetDLQTasksRequest{
123
>
DlqKey: &commonspb.HistoryDLQKey{
124
>
TaskCategory: int32(tasks.CategoryTransfer.ID()),
125
>
SourceCluster: sourceCluster,
126
>
TargetCluster: targetCluster,
127
>
},
128
>
PageSize: 1,
129
>
NextPageToken: nextPageToken,
130
>
})
131
>
require.NoError(t, err)
132
>
assert.Equal(t, 1, len(res.DlqTasks))
133
>
assert.Equal(t, int64(persistence.FirstQueueMessageID+i), res.DlqTasks[0].Metadata.MessageId)
134
>
nextPageToken = res.NextPageToken
135
>
}
136
}
137