public void Publish_Where_Channel_Publication_Fails_Results_In_A_Non_Completed_Publication_Task_Which_Is_Placed_Back_On_The_Publication_Queue()
{
var _connection = Substitute.For<IConnection>();
var _channel = Substitute.For<IModel>();
var _publicationQueue = new BlockingCollection<Publication>();
_connection.CreateModel().Returns(_channel);
_channel
.When(channel => channel.BasicPublish(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<IBasicProperties>(), Arg.Any<byte[]>()))
.Do(callInfo => { throw new ApplicationException("Bang !"); });
var _messageDelivery = new MessageDelivery("EXCHANGE", typeof(MyEvent).Name, MessageDeliveryMode.Persistent, message => "ARoutingKey");
var _myEvent = new MyEvent(Guid.NewGuid(), "CorrlationId_1", "Detail", 1);
var _taskCompletionSource = new TaskCompletionSource<PublicationResult>();
var _publication = new Publication(_messageDelivery, _myEvent, _taskCompletionSource);
var _SUT = new Publisher(_connection, _publicationQueue, CancellationToken.None);
var _publisherTask = _SUT.Start();
_publicationQueue.Add(_publication);
try { _publisherTask.Wait(); } catch { }
Assert.IsFalse(_publication.ResultTask.IsCompleted);
Assert.AreSame(_publication, _publicationQueue.First());
}
public void Run_Where_We_Batch_Publish_Messages_Waiting_For_Batch_Completion_Each_Time()
{
this.InstantiateAndStartPublisher();
do
{
var _tasks = new List<Task>();
var _messageDelivery = new MessageDelivery(Configuration.ExchangeName1, "H", MessageDeliveryMode.Persistent, m => "Ted");
for (var _index = 1; _index <= 100; _index++)
{
var _myEvent = new MyEvent(Guid.NewGuid(), "", "R1", _index, "09:00", "DDD....");
var _taskCompletionSource = new TaskCompletionSource<PublicationResult>();
var _publication = new Publication(_messageDelivery, _myEvent, _taskCompletionSource);
this.c_publicationQueue.Add(_publication);
_tasks.Add(_publication.ResultTask);
}
Task.WaitAll(_tasks.ToArray());
Console.WriteLine("Hit enter to publish more messages, x to exit");
} while (Console.ReadLine() != "x");
Console.WriteLine("Hit enter to cancel");
Console.ReadLine();
this.c_cancellationTokenSource.Cancel();
Console.WriteLine("Hit enter to close connection (Channel should already be closed - check the dashboard)");
Console.ReadLine();
this.c_connection.Close();
Console.WriteLine("Hit enter to exit");
Console.ReadLine();
}
public Publication(
MessageDelivery configuration,
Message message,
TaskCompletionSource<PublicationResult> result)
{
this.c_configuration = configuration;
this.c_message = message;
this.c_result = result;
}
public PublicationResult(
MessageDelivery configuration,
Message message,
PublicationResultStatus status,
string statusContext = null)
{
this.Configuration = configuration;
this.Message = message;
this.Status = status;
this.StatusContext = statusContext;
}
public void Ctor_Cater_For_Direct_Exchange_Delivery()
{
var _SUT = new PMCG.Messaging.Client.Configuration.MessageDelivery(
string.Empty,
"TypeHeader",
MessageDeliveryMode.Persistent,
message => "DirectQueue");
Assert.AreEqual(string.Empty, _SUT.ExchangeName);
Assert.AreEqual("TypeHeader", _SUT.TypeHeader);
Assert.AreEqual(MessageDeliveryMode.Persistent, _SUT.DeliveryMode);
Assert.AreEqual("DirectQueue", _SUT.RoutingKeyFunc(null));
}
public void Ctor_Cater_For_Direct_Exchange_Delivery()
{
var _SUT = new MessageDelivery(
string.Empty,
"TypeHeader",
MessageDeliveryMode.Persistent,
message => "DirectQueue");
Assert.AreEqual(string.Empty, _SUT.ExchangeName);
Assert.AreEqual("TypeHeader", _SUT.TypeHeader);
Assert.AreEqual(MessageDeliveryMode.Persistent, _SUT.DeliveryMode);
Assert.AreEqual("DirectQueue", _SUT.RoutingKeyFunc(null));
}
public void Publish_Where_Channel_Is_Closed_Results_In_Faulted_Publisher_Task()
{
var _connection = Substitute.For<IConnection>();
var _channel = Substitute.For<IModel>();
var _publicationQueue = new BlockingCollection<Publication>();
_connection.CreateModel().Returns(_channel);
_channel.CreateBasicProperties().Returns(callInfo => { throw new Exception("Channel not open !"); });
var _messageDelivery = new MessageDelivery("test_publisher_confirms", typeof(MyEvent).Name, MessageDeliveryMode.Persistent, message => "ARoutingKey");
var _myEvent = new MyEvent(Guid.NewGuid(), "CorrlationId_1", "Detail", 1);
var _taskCompletionSource = new TaskCompletionSource<PublicationResult>();
var _publication = new Publication(_messageDelivery, _myEvent, _taskCompletionSource);
var _SUT = new Publisher(_connection, _publicationQueue, CancellationToken.None);
var _publisherTask = _SUT.Start();
_publicationQueue.Add(_publication);
bool _isFaultedPublisherTask = false;
try { _publisherTask.Wait(); } catch { _isFaultedPublisherTask = true; }
Assert.IsTrue(_isFaultedPublisherTask);
}
public void Publish_Where_Single_Publication_Published_And_Acked_Results_In_Task_Completion()
{
var _connection = Substitute.For<IConnection>();
var _channel = Substitute.For<IModel>();
var _publicationQueue = new BlockingCollection<Publication>();
var _messageProperties = Substitute.For<IBasicProperties>();
var _waitHandle = new AutoResetEvent(false);
_connection.CreateModel().Returns(_channel);
_channel.CreateBasicProperties().Returns(_messageProperties);
_channel.NextPublishSeqNo.Returns(1UL);
_channel
.When(channel => channel.BasicPublish(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<IBasicProperties>(), Arg.Any<byte[]>()))
.Do(callInfo => _waitHandle.Set());
var _SUT = new Publisher(_connection, _publicationQueue, CancellationToken.None);
_SUT.Start(); // Can't capture result due to compiler treating warnings as errors - var is not used
var _messageDelivery = new MessageDelivery("test_publisher_confirms", typeof(MyEvent).Name, MessageDeliveryMode.Persistent, message => "ARoutingKey");
var _myEvent = new MyEvent(Guid.NewGuid(), "CorrlationId_1", "Detail", 1);
var _taskCompletionSource = new TaskCompletionSource<PublicationResult>();
var _publication = new Publication(_messageDelivery, _myEvent, _taskCompletionSource);
_publicationQueue.Add(_publication);
_waitHandle.WaitOne(); // Allow publication to complete
_channel.BasicAcks += Raise.Event<BasicAckEventHandler>(_channel, new BasicAckEventArgs { Multiple = false, DeliveryTag = 1 });
Assert.IsTrue(_publication.ResultTask.IsCompleted);
_messageProperties.Received().ContentType = "application/json";
_messageProperties.Received().DeliveryMode = (byte)_messageDelivery.DeliveryMode;
_messageProperties.Received().Type = _messageDelivery.TypeHeader;
_messageProperties.Received().MessageId = _myEvent.Id.ToString();
_messageProperties.Received().CorrelationId = _myEvent.CorrelationId;
_channel.Received().BasicPublish(_messageDelivery.ExchangeName, _messageDelivery.RoutingKeyFunc(_myEvent), _messageProperties, Arg.Any<byte[]>());
}
public void Publish_Where_Exchange_Does_Not_Exist_Results_In_Channel_Shutdown_And_A_Channel_Shutdown_Task_Result()
{
var _connection = Substitute.For<IConnection>();
var _channel = Substitute.For<IModel>();
var _publicationQueue = new BlockingCollection<Publication>();
var _waitHandle = new AutoResetEvent(false);
_connection.CreateModel().Returns(_channel);
_channel.NextPublishSeqNo.Returns(1Ul);
_channel
.When(channel => channel.BasicPublish(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<IBasicProperties>(), Arg.Any<byte[]>()))
.Do(callInfo => _waitHandle.Set());
var _SUT = new Publisher(_connection, _publicationQueue, CancellationToken.None);
_SUT.Start();
var _messageDelivery = new MessageDelivery("NON_EXISTENT_EXCHANGE", typeof(MyEvent).Name, MessageDeliveryMode.Persistent, message => "ARoutingKey");
var _myEvent = new MyEvent(Guid.NewGuid(), "CorrlationId_1", "Detail", 100);
var _taskCompletionSource = new TaskCompletionSource<PublicationResult>();
var _publication = new Publication(_messageDelivery, _myEvent, _taskCompletionSource);
_publicationQueue.Add(_publication);
_waitHandle.WaitOne(); // Allow channel publication to complete
_channel.ModelShutdown += Raise.Event<ModelShutdownEventHandler>(_channel, new ShutdownEventArgs(ShutdownInitiator.Peer, 1, "404 Exchange does not exist !"));
// Since all running on the same thread we do not need to wait - this is also not relaistic as we know the channel shutdown event will happen on a different thread
Assert.IsTrue(_publication.ResultTask.IsCompleted);
Assert.AreEqual(PublicationResultStatus.ChannelShutdown, _publication.ResultTask.Result.Status);
Assert.IsTrue(_publication.ResultTask.Result.StatusContext.Contains("404 Exchange does not exist !"));
}
public void Publish_Where_Two_Messages_Being_Published_But_Channel_Is_Closed_Before_Acks_Received_Results_In_Two_Channel_Shutdown_Tasks()
{
var _connection = Substitute.For<IConnection>();
var _channel = Substitute.For<IModel>();
var _publicationQueue = new BlockingCollection<Publication>();
var _waitHandle = new CountdownEvent(2);
_connection.CreateModel().Returns(_channel);
_channel.NextPublishSeqNo.Returns(1Ul, 2UL);
_channel
.When(channel => channel.BasicPublish(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<IBasicProperties>(), Arg.Any<byte[]>()))
.Do(callInfo => _waitHandle.Signal());
var _SUT = new Publisher(_connection, _publicationQueue, CancellationToken.None);
_SUT.Start();
var _messageDelivery = new MessageDelivery("test_publisher_confirms", typeof(MyEvent).Name, MessageDeliveryMode.Persistent, message => "ARoutingKey");
var _myEvent = new MyEvent(Guid.NewGuid(), "CorrlationId_1", "Detail", 100);
var _taskCompletionSource1 = new TaskCompletionSource<PublicationResult>();
var _publication1 = new Publication(_messageDelivery, _myEvent, _taskCompletionSource1);
var _taskCompletionSource2 = new TaskCompletionSource<PublicationResult>();
var _publication2 = new Publication(_messageDelivery, _myEvent, _taskCompletionSource2);
_publicationQueue.Add(_publication1);
_publicationQueue.Add(_publication2);
_waitHandle.Wait(); // Allow channel publications to complete
_channel.ModelShutdown += Raise.Event<ModelShutdownEventHandler>(_channel, new ShutdownEventArgs(ShutdownInitiator.Peer, 1, "Bang!"));
// Since all running on the same thread we do not need to wait - this is also not relaistic as we know the channel shutdown event will happen on a different thread
Assert.IsTrue(_publication1.ResultTask.IsCompleted);
Assert.AreEqual(PublicationResultStatus.ChannelShutdown, _publication1.ResultTask.Result.Status);
Assert.IsTrue(_publication1.ResultTask.Result.StatusContext.Contains("Bang!"));
Assert.IsTrue(_publication2.ResultTask.IsCompleted);
Assert.AreEqual(PublicationResultStatus.ChannelShutdown, _publication2.ResultTask.Result.Status);
Assert.IsTrue(_publication2.ResultTask.Result.StatusContext.Contains("Bang!"));
}
public void Publish_Where_Single_Message_Published_And_Nacked_Results_In_Nacked_Task_Result()
{
var _connection = Substitute.For<IConnection>();
var _channel = Substitute.For<IModel>();
var _publicationQueue = new BlockingCollection<Publication>();
var _waitHandle = new AutoResetEvent(false);
_connection.CreateModel().Returns(_channel);
_channel.NextPublishSeqNo.Returns(1Ul);
_channel
.When(channel => channel.BasicPublish(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<IBasicProperties>(), Arg.Any<byte[]>()))
.Do(callInfo => _waitHandle.Set());
var _SUT = new Publisher(_connection, _publicationQueue, CancellationToken.None);
_SUT.Start();
var _messageDelivery = new MessageDelivery("test_publisher_confirms", typeof(MyEvent).Name, MessageDeliveryMode.Persistent, message => "ARoutingKey");
var _myEvent = new MyEvent(Guid.NewGuid(), "CorrlationId_1", "Detail", 100);
var _taskCompletionSource = new TaskCompletionSource<PublicationResult>();
var _publication = new Publication(_messageDelivery, _myEvent, _taskCompletionSource);
_publicationQueue.Add(_publication);
_waitHandle.WaitOne(); // Allow channel publication to complete
_channel.BasicNacks += Raise.Event<BasicNackEventHandler>(_channel, new BasicNackEventArgs { Multiple = true, DeliveryTag = 1UL });
Assert.IsTrue(_publication.ResultTask.IsCompleted);
Assert.AreEqual(PublicationResultStatus.Nacked, _publication.ResultTask.Result.Status);
Assert.IsNull(_publication.ResultTask.Result.StatusContext);
}
public void Publish_Where_100_Messages_Published_And_A_Single_Multi_Ack_For_Some_Messages_Results_In_Some_Completed_Tasks_And_Some_Still_Pending()
{
var _connection = Substitute.For<IConnection>();
var _channel = Substitute.For<IModel>();
var _publicationQueue = new BlockingCollection<Publication>();
var _waitHandle = new CountdownEvent(100);
_connection.CreateModel().Returns(_channel);
var _nextPublishSeqNo = 1UL;
_channel.NextPublishSeqNo.Returns(callInfo => _nextPublishSeqNo++); // Would not work when I used .Returns(1Ul, 2UL, 3UL); Not sure why this works !
_channel
.When(channel => channel.BasicPublish(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<IBasicProperties>(), Arg.Any<byte[]>()))
.Do(callInfo => _waitHandle.Signal());
var _SUT = new Publisher(_connection, _publicationQueue, CancellationToken.None);
_SUT.Start();
var _publications = new List<Publication>();
for (var _index = 1; _index <= 100; _index++)
{
var _messageDelivery = new MessageDelivery("test_publisher_confirms", typeof(MyEvent).Name, MessageDeliveryMode.Persistent, message => "ARoutingKey");
var _myEvent = new MyEvent(Guid.NewGuid(), "CorrlationId_1", "Detail", _index);
var _taskCompletionSource = new TaskCompletionSource<PublicationResult>();
var _publication = new Publication(_messageDelivery, _myEvent, _taskCompletionSource);
_publicationQueue.Add(_publication);
_publications.Add(_publication);
}
_waitHandle.Wait(); // Allow channel publications to complete
foreach(var _publication in _publications) { Assert.IsFalse(_publication.ResultTask.IsCompleted); }
var _deliveryTagToAcknowledge = 73;
_channel.BasicAcks += Raise.Event<BasicAckEventHandler>(_channel, new BasicAckEventArgs { Multiple = true, DeliveryTag = (ulong)_deliveryTagToAcknowledge });
Assert.AreEqual(_deliveryTagToAcknowledge, _publications.Count(publication => publication.ResultTask.IsCompleted), "A1");
Assert.AreEqual(_deliveryTagToAcknowledge, _publications.Count(publication => publication.ResultTask.IsCompleted && publication.ResultTask.Result.Status == PublicationResultStatus.Acked), "A2");
Assert.AreEqual(100 - _deliveryTagToAcknowledge, _publications.Count(publication => !publication.ResultTask.IsCompleted), "A3");
}
