protected async Task PeekLockWithAbandonTestCase(MessageSender messageSender, MessageReceiver messageReceiver, int messageCount)
{
// Send messages
await TestUtility.SendMessagesAsync(messageSender, messageCount);
// Receive 5 messages and Abandon them
int abandonMessagesCount = 5;
var receivedMessages = await TestUtility.ReceiveMessagesAsync(messageReceiver, abandonMessagesCount);
Assert.True(receivedMessages.Count() == abandonMessagesCount);
await TestUtility.AbandonMessagesAsync(messageReceiver, receivedMessages);
// Receive all 10 messages
receivedMessages = await TestUtility.ReceiveMessagesAsync(messageReceiver, messageCount);
Assert.True(receivedMessages.Count() == messageCount);
// TODO: Some reason for partitioned entities the delivery count is incorrect. Investigate and enable
// 5 of these messages should have deliveryCount = 2
int messagesWithDeliveryCount2 = receivedMessages.Where(message => message.DeliveryCount == 2).Count();
TestUtility.Log($"Messages with Delivery Count 2: {messagesWithDeliveryCount2}");
Assert.True(messagesWithDeliveryCount2 == abandonMessagesCount);
// Complete Messages
await TestUtility.CompleteMessagesAsync(messageReceiver, receivedMessages);
}