public async Task PeekBatchOfMessagesLogsEvents(bool specifySeqNumber)
{
var mockLogger = new Mock <ServiceBusEventSource>();
var maxMessages = 4;
var mockTransportReceiver = new Mock <TransportReceiver>();
var mockConnection = GetMockConnection(mockTransportReceiver);
mockTransportReceiver.Setup(
transportReceiver => transportReceiver.PeekBatchAtAsync(
It.IsAny <long?>(),
maxMessages,
It.IsAny <CancellationToken>()))
.Returns(
Task.FromResult((IList <ServiceBusReceivedMessage>)
new List <ServiceBusReceivedMessage> {
new ServiceBusReceivedMessage(),
new ServiceBusReceivedMessage()
}));
var receiver = new ServiceBusReceiver(mockConnection.Object, "queueName", false, new ServiceBusReceiverOptions())
{
Logger = mockLogger.Object
};
var seqNumber = 5;
IList <ServiceBusReceivedMessage> msgs;
if (specifySeqNumber)
{
msgs = await receiver.PeekBatchAtAsync(seqNumber, maxMessages);
}
else
{
msgs = await receiver.PeekBatchAsync(maxMessages : maxMessages);
}
mockLogger
.Verify(
log => log.PeekMessageStart(
receiver.Identifier,
specifySeqNumber ? (long?)seqNumber : null,
// the amount requested
maxMessages),
Times.Once);
mockLogger
.Verify(
log => log.PeekMessageComplete(
receiver.Identifier,
// the amount we actually received
msgs.Count),
Times.Once);
}
public async Task RoundRobinSessions()
{
await using (var scope = await ServiceBusScope.CreateWithQueue(enablePartitioning: false, enableSession: true))
{
await using var client = new ServiceBusClient(TestEnvironment.ServiceBusConnectionString);
ServiceBusSender sender = client.GetSender(scope.QueueName);
var messageCt = 10;
HashSet <string> sessions = new HashSet <string>()
{
"1", "2", "3"
};
// send the messages
foreach (string session in sessions)
{
using ServiceBusMessageBatch batch = await sender.CreateBatchAsync();
ServiceBusMessageBatch messageBatch = AddMessages(batch, messageCt, session);
await sender.SendBatchAsync(messageBatch);
}
// create receiver not scoped to a specific session
for (int i = 0; i < 10; i++)
{
ServiceBusReceiver receiver = await client.GetSessionReceiverAsync(scope.QueueName);
var session = receiver.GetSessionManager();
foreach (ServiceBusReceivedMessage peekedMessage in await receiver.PeekBatchAtAsync(
sequenceNumber: 1,
maxMessages: 10))
{
var sessionId = session.SessionId;
Assert.AreEqual(sessionId, peekedMessage.SessionId);
}
// Close the receiver client when we are done with it. Since the sessionClient doesn't own the underlying connection, the connection remains open, but the session link will be closed.
await receiver.DisposeAsync();
}
}
}
