public async Task Should_batch_non_isolated_operations()
{
var settings = new SettingsHolder();
var mockSqsClient = new MockSqsClient();
var dispatcher = new MessageDispatcher(new TransportConfiguration(settings), null, mockSqsClient, new QueueUrlCache(mockSqsClient));
var transportOperations = new TransportOperations(
new TransportOperation(
new OutgoingMessage(Guid.NewGuid().ToString(), new Dictionary <string, string>(), Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address1"),
DispatchConsistency.Default),
new TransportOperation(
new OutgoingMessage(Guid.NewGuid().ToString(), new Dictionary <string, string>(), Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address2"),
DispatchConsistency.Default));
var transportTransaction = new TransportTransaction();
var context = new ContextBag();
await dispatcher.Dispatch(transportOperations, transportTransaction, context);
Assert.IsEmpty(mockSqsClient.RequestsSent);
Assert.AreEqual(2, mockSqsClient.BatchRequestsSent.Count);
Assert.AreEqual("address1", mockSqsClient.BatchRequestsSent.ElementAt(0).QueueUrl);
Assert.AreEqual("address2", mockSqsClient.BatchRequestsSent.ElementAt(1).QueueUrl);
}
public async Task Does_not_send_extra_properties_in_payload_by_default()
{
var settings = new SettingsHolder();
var mockSqsClient = new MockSqsClient();
var dispatcher = new MessageDispatcher(new TransportConfiguration(settings), null, mockSqsClient, new QueueUrlCache(mockSqsClient));
var transportOperations = new TransportOperations(
new TransportOperation(
new OutgoingMessage("1234", new Dictionary <string, string>
{
{ TransportHeaders.TimeToBeReceived, expectedTtbr.ToString() },
{ Headers.ReplyToAddress, expectedReplyToAddress }
}, Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address")));
var transportTransaction = new TransportTransaction();
var context = new ContextBag();
await dispatcher.Dispatch(transportOperations, transportTransaction, context);
Assert.IsNotEmpty(mockSqsClient.RequestsSent, "No requests sent");
var request = mockSqsClient.RequestsSent.First();
IDictionary <string, JToken> bodyJson = JObject.Parse(request.MessageBody);
Assert.IsFalse(bodyJson.ContainsKey("TimeToBeReceived"), "TimeToBeReceived serialized");
Assert.IsFalse(bodyJson.ContainsKey("ReplyToAddress"), "ReplyToAddress serialized");
}
public void Should_raise_queue_does_not_exists_for_delayed_delivery_for_isolated_dispatch()
{
var settings = new SettingsHolder();
var transportExtensions = new TransportExtensions <SqsTransport>(settings);
transportExtensions.UnrestrictedDurationDelayedDelivery();
var mockSqsClient = new MockSqsClient();
mockSqsClient.RequestResponse = req => throw new QueueDoesNotExistException("Queue does not exist");
var dispatcher = new MessageDispatcher(new TransportConfiguration(settings), null, mockSqsClient, new QueueUrlCache(mockSqsClient));
var transportOperations = new TransportOperations(
new TransportOperation(
new OutgoingMessage(Guid.NewGuid().ToString(), new Dictionary <string, string>(), Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address1"),
DispatchConsistency.Isolated,
new List <DeliveryConstraint>
{
new DelayDeliveryWith(TimeSpan.FromMinutes(30))
}));
var transportTransaction = new TransportTransaction();
var context = new ContextBag();
var exception = Assert.ThrowsAsync <QueueDoesNotExistException>(async() => await dispatcher.Dispatch(transportOperations, transportTransaction, context));
StringAssert.StartsWith("Destination 'address1' doesn't support delayed messages longer than", exception.Message);
}
public async Task Includes_message_id_in_message_attributes(List <DeliveryConstraint> constraints)
{
var settings = new SettingsHolder();
var transportExtensions = new TransportExtensions <SqsTransport>(settings);
transportExtensions.UnrestrictedDurationDelayedDelivery();
var mockSqsClient = new MockSqsClient();
var dispatcher = new MessageDispatcher(new TransportConfiguration(settings), null, mockSqsClient, new QueueUrlCache(mockSqsClient));
var expectedId = "1234";
var transportOperations = new TransportOperations(
new TransportOperation(
new OutgoingMessage(expectedId, new Dictionary <string, string>(), Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address"),
DispatchConsistency.Isolated,
constraints));
var transportTransaction = new TransportTransaction();
var context = new ContextBag();
await dispatcher.Dispatch(transportOperations, transportTransaction, context);
var sentMessage = mockSqsClient.RequestsSent.First();
Assert.AreEqual(expectedId, sentMessage.MessageAttributes[Headers.MessageId].StringValue);
}
public async Task Should_upload_large_isolated_operations_request_to_s3()
{
var settings = new SettingsHolder();
var transportExtensions = new TransportExtensions <SqsTransport>(settings);
transportExtensions.S3("someBucket", "somePrefix");
var mockS3Client = new MockS3Client();
var mockSqsClient = new MockSqsClient();
var dispatcher = new MessageDispatcher(new TransportConfiguration(settings), mockS3Client, mockSqsClient, new QueueUrlCache(new MockSqsClient()));
var transportOperations = new TransportOperations(
new TransportOperation(
new OutgoingMessage(Guid.NewGuid().ToString(), new Dictionary <string, string>(), Encoding.Default.GetBytes(new string('x', 256 * 1024))),
new UnicastAddressTag("address1"),
DispatchConsistency.Isolated),
new TransportOperation(
new OutgoingMessage(Guid.NewGuid().ToString(), new Dictionary <string, string>(), Encoding.Default.GetBytes(new string('x', 256 * 1024))),
new UnicastAddressTag("address2"),
DispatchConsistency.Isolated),
new TransportOperation( /* Crazy long message id will cause the message to go over limits because attributes count as well */
new OutgoingMessage(new string('x', 256 * 1024), new Dictionary <string, string>(), Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address2"),
DispatchConsistency.Isolated));
var transportTransaction = new TransportTransaction();
var context = new ContextBag();
await dispatcher.Dispatch(transportOperations, transportTransaction, context);
Assert.AreEqual(3, mockSqsClient.RequestsSent.Count);
Assert.AreEqual(3, mockS3Client.PutObjectRequestsSent.Count);
var firstUpload = mockS3Client.PutObjectRequestsSent.ElementAt(0);
var secondUpload = mockS3Client.PutObjectRequestsSent.ElementAt(1);
var thirdUpload = mockS3Client.PutObjectRequestsSent.ElementAt(2);
Assert.AreEqual("someBucket", firstUpload.BucketName);
Assert.AreEqual("someBucket", secondUpload.BucketName);
Assert.AreEqual("someBucket", thirdUpload.BucketName);
StringAssert.Contains($@"""Body"":"""",""S3BodyKey"":""{firstUpload.Key}", mockSqsClient.RequestsSent.ElementAt(0).MessageBody);
StringAssert.Contains($@"""Body"":"""",""S3BodyKey"":""{secondUpload.Key}", mockSqsClient.RequestsSent.ElementAt(1).MessageBody);
StringAssert.Contains($@"""Body"":"""",""S3BodyKey"":""{thirdUpload.Key}", mockSqsClient.RequestsSent.ElementAt(2).MessageBody);
}
public async Task Sends_V1_compatible_payload_when_configured()
{
var settings = new SettingsHolder();
settings.Set(SettingsKeys.V1CompatibilityMode, true);
var mockSqsClient = new MockSqsClient();
var dispatcher = new MessageDispatcher(new TransportConfiguration(settings), null, mockSqsClient, new QueueUrlCache(mockSqsClient));
var transportOperations = new TransportOperations(
new TransportOperation(
new OutgoingMessage("1234", new Dictionary <string, string>
{
{ TransportHeaders.TimeToBeReceived, expectedTtbr.ToString() },
{ Headers.ReplyToAddress, expectedReplyToAddress }
}, Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address"),
DispatchConsistency.Isolated));
var transportTransaction = new TransportTransaction();
var context = new ContextBag();
await dispatcher.Dispatch(transportOperations, transportTransaction, context);
Assert.IsNotEmpty(mockSqsClient.RequestsSent, "No requests sent");
var request = mockSqsClient.RequestsSent.First();
var bodyJson = JObject.Parse(request.MessageBody);
Assert.IsTrue(bodyJson.ContainsKey("TimeToBeReceived"), "TimeToBeReceived not serialized");
Assert.AreEqual(expectedTtbr.ToString(), bodyJson["TimeToBeReceived"].Value <string>(), "Expected TTBR mismatch");
Assert.IsTrue(bodyJson.ContainsKey("ReplyToAddress"), "ReplyToAddress not serialized");
Assert.IsTrue(bodyJson["ReplyToAddress"].HasValues, "ReplyToAddress is not an object");
Assert.IsTrue(bodyJson["ReplyToAddress"].Value <JObject>().ContainsKey("Queue"), "ReplyToAddress does not have a Queue value");
Assert.AreEqual(expectedReplyToAddress, bodyJson["ReplyToAddress"].Value <JObject>()["Queue"].Value <string>(), "Expected ReplyToAddress mismatch");
}
public async Task Should_dispatch_non_batched_all_messages_that_failed_in_batch()
{
var settings = new SettingsHolder();
var mockSqsClient = new MockSqsClient();
var dispatcher = new MessageDispatcher(new TransportConfiguration(settings), null, mockSqsClient, new QueueUrlCache(mockSqsClient));
var firstMessageIdThatWillFail = Guid.NewGuid().ToString();
var secondMessageIdThatWillFail = Guid.NewGuid().ToString();
mockSqsClient.BatchRequestResponse = req =>
{
var firstMessageMatch = req.Entries.SingleOrDefault(x => x.MessageAttributes[Headers.MessageId].StringValue == firstMessageIdThatWillFail);
if (firstMessageMatch != null)
{
return(new SendMessageBatchResponse
{
Failed = new List <BatchResultErrorEntry>
{
new BatchResultErrorEntry
{
Id = firstMessageMatch.Id,
Message = "You know why"
}
}
});
}
var secondMessageMatch = req.Entries.SingleOrDefault(x => x.MessageAttributes[Headers.MessageId].StringValue == secondMessageIdThatWillFail);
if (secondMessageMatch != null)
{
return(new SendMessageBatchResponse
{
Failed = new List <BatchResultErrorEntry>
{
new BatchResultErrorEntry
{
Id = secondMessageMatch.Id,
Message = "You know why"
}
}
});
}
return(new SendMessageBatchResponse());
};
var firstMessageThatWillBeSuccessful = Guid.NewGuid().ToString();
var secondMessageThatWillBeSuccessful = Guid.NewGuid().ToString();
var transportOperations = new TransportOperations(
new TransportOperation(
new OutgoingMessage(firstMessageIdThatWillFail, new Dictionary <string, string>(), Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address1"),
DispatchConsistency.Default),
new TransportOperation(
new OutgoingMessage(firstMessageThatWillBeSuccessful, new Dictionary <string, string>(), Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address1"),
DispatchConsistency.Default),
new TransportOperation(
new OutgoingMessage(secondMessageThatWillBeSuccessful, new Dictionary <string, string>(), Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address2"),
DispatchConsistency.Default),
new TransportOperation(
new OutgoingMessage(secondMessageIdThatWillFail, new Dictionary <string, string>(), Encoding.Default.GetBytes("{}")),
new UnicastAddressTag("address2"),
DispatchConsistency.Default));
var transportTransaction = new TransportTransaction();
var context = new ContextBag();
await dispatcher.Dispatch(transportOperations, transportTransaction, context);
Assert.AreEqual(2, mockSqsClient.RequestsSent.Count);
Assert.AreEqual(firstMessageIdThatWillFail, mockSqsClient.RequestsSent.ElementAt(0).MessageAttributes[Headers.MessageId].StringValue);
Assert.AreEqual(secondMessageIdThatWillFail, mockSqsClient.RequestsSent.ElementAt(1).MessageAttributes[Headers.MessageId].StringValue);
Assert.AreEqual(2, mockSqsClient.BatchRequestsSent.Count);
CollectionAssert.AreEquivalent(new[] { firstMessageIdThatWillFail, firstMessageThatWillBeSuccessful }, mockSqsClient.BatchRequestsSent.ElementAt(0).Entries.Select(x => x.MessageAttributes[Headers.MessageId].StringValue));
CollectionAssert.AreEquivalent(new[] { secondMessageIdThatWillFail, secondMessageThatWillBeSuccessful }, mockSqsClient.BatchRequestsSent.ElementAt(1).Entries.Select(x => x.MessageAttributes[Headers.MessageId].StringValue));
}
