139
queueType persistence.QueueV2Type,
140
queueName string,
142
>
response, err := queue.ReadMessages(ctx, &persistence.InternalReadMessagesRequest{
143
>
QueueType: queueType,
144
>
QueueName: queueName,
145
>
PageSize: 1,
146
>
NextPageToken: nil,
147
>
})
148
>
require.NoError(t, err)
149
>
assert.Equal(t, 0, len(response.Messages))
150
>
151
>
encodingType := enumspb.ENCODING_TYPE_JSON
152
>
_, err = persistencetest.EnqueueMessage(ctx, queue, queueType, queueName)
153
>
require.NoError(t, err)
154
>
155
>
_, err = persistencetest.EnqueueMessage(ctx, queue, queueType, queueName, func(p *persistencetest.EnqueueParams) {
156
>
p.Data = []byte("2")
157
>
})
158
>
require.NoError(t, err)
159
>
160
>
response, err = queue.ReadMessages(ctx, &persistence.InternalReadMessagesRequest{
161
>
QueueType: queueType,
162
>
QueueName: queueName,
163
>
PageSize: 1,
164
>
NextPageToken: nil,
165
>
})
166
>
require.NoError(t, err)
167
>
require.Len(t, response.Messages, 1)
168
>
assert.Equal(t, int64(persistence.FirstQueueMessageID), response.Messages[0].MetaData.ID)
169
>
assert.Equal(t, []byte("1"), response.Messages[0].Data.Data)
170
>
assert.Equal(t, encodingType, response.Messages[0].Data.EncodingType)
171
>
assert.NotNil(t, response.NextPageToken)
172
>
173
>
response, err = queue.ReadMessages(ctx, &persistence.InternalReadMessagesRequest{
174
>
QueueType: queueType,
175
>
QueueName: queueName,
176
>
PageSize: 1,
177
>
NextPageToken: response.NextPageToken,
178
>
})
179
>
require.NoError(t, err)
180
>
require.Len(t, response.Messages, 1)
181
>
assert.Equal(t, int64(persistence.FirstQueueMessageID+1), response.Messages[0].MetaData.ID)
182
>
assert.Equal(t, []byte("2"), response.Messages[0].Data.Data)
183
>
assert.Equal(t, encodingType, response.Messages[0].Data.EncodingType)
184
>
185
>
response, err = queue.ReadMessages(ctx, &persistence.InternalReadMessagesRequest{
186
>
QueueType: queueType,
187
>
QueueName: queueName,
188
>
PageSize: 1,
189
>
NextPageToken: response.NextPageToken,
190
>
})
191
>
require.NoError(t, err)
192
>
assert.Empty(t, response.Messages)
193
>
assert.Nil(t, response.NextPageToken)
194
}
195