public void TestReplyToTwoDeep()
{
var mockConnectionFactory = new Mock <ConnectionFactory>();
var mockConnection = new Mock <IConnection>();
var mockChannel = new Mock <IModel>();
mockConnectionFactory.Setup(m => m.CreateConnection()).Returns(mockConnection.Object);
mockConnection.Setup(m => m.IsOpen).Returns(true);
mockConnection.Setup(m => m.CreateModel()).Returns(mockChannel.Object);
mockChannel.Setup(m => m.CreateBasicProperties()).Returns(() => new BasicProperties());
var template = new RabbitTemplate(new SingleConnectionFactory(mockConnectionFactory.Object));
var replyQueue = new Queue("new.replyTo");
template.ReplyQueue = replyQueue;
var messageProperties = new MessageProperties();
messageProperties.ReplyTo = "replyTo2";
messageProperties.SetHeader(RabbitTemplate.STACKED_REPLY_TO_HEADER, "replyTo1");
messageProperties.SetHeader(RabbitTemplate.STACKED_CORRELATION_HEADER, "a");
var message = new Message(Encoding.UTF8.GetBytes("Hello, world!"), messageProperties);
var props = new List <IBasicProperties>();
mockChannel.Setup(m => m.BasicPublish(It.IsAny <string>(), It.IsAny <string>(), It.IsAny <bool>(), It.IsAny <bool>(), It.IsAny <IBasicProperties>(), It.IsAny <byte[]>())).Callback <string, string, bool, bool, IBasicProperties, byte[]>
(
(a1, a2, a3, a4, a5, a6) =>
{
var basicProps = a5;
props.Add(basicProps);
var springProps = new DefaultMessagePropertiesConverter().ToMessageProperties(basicProps, null, "UTF-8");
var replyMessage = new Message(Encoding.UTF8.GetBytes("!dlrow olleH"), springProps);
template.OnMessage(replyMessage);
});
var reply = template.SendAndReceive(message);
Assert.AreEqual(1, props.Count);
var basicProperties = props[0];
Assert.AreEqual("new.replyTo", basicProperties.ReplyTo);
Assert.AreEqual("replyTo2:replyTo1", basicProperties.Headers[RabbitTemplate.STACKED_REPLY_TO_HEADER]);
Assert.IsTrue(((string)basicProperties.Headers[RabbitTemplate.STACKED_CORRELATION_HEADER]).EndsWith(":a"));
Assert.AreEqual("replyTo1", reply.MessageProperties.Headers[RabbitTemplate.STACKED_REPLY_TO_HEADER]);
Assert.AreEqual("a", reply.MessageProperties.Headers[RabbitTemplate.STACKED_CORRELATION_HEADER]);
}
public void TestReplyToThreeDeep()
{
var mockConnectionFactory = new Mock<ConnectionFactory>();
var mockConnection = new Mock<IConnection>();
var mockChannel = new Mock<IModel>();
mockConnectionFactory.Setup(m => m.CreateConnection()).Returns(mockConnection.Object);
mockConnection.Setup(m => m.IsOpen).Returns(true);
mockConnection.Setup(m => m.CreateModel()).Returns(mockChannel.Object);
mockChannel.Setup(m => m.CreateBasicProperties()).Returns(() => new BasicProperties());
var template = new RabbitTemplate(new SingleConnectionFactory(mockConnectionFactory.Object));
var replyQueue = new Queue("new.replyTo");
template.ReplyQueue = replyQueue;
var messageProperties = new MessageProperties();
messageProperties.ReplyTo = "replyTo2";
messageProperties.SetHeader(RabbitTemplate.STACKED_REPLY_TO_HEADER, "replyTo1");
messageProperties.SetHeader(RabbitTemplate.STACKED_CORRELATION_HEADER, "a");
var message = new Message(Encoding.UTF8.GetBytes("Hello, world!"), messageProperties);
var props = new List<IBasicProperties>();
var count = new AtomicInteger();
var nestedReplyTo = new List<string>();
var nestedReplyStack = new List<string>();
var nestedCorrelation = new List<string>();
mockChannel.Setup(m => m.BasicPublish(It.IsAny<string>(), It.IsAny<string>(), It.IsAny<bool>(), It.IsAny<bool>(), It.IsAny<IBasicProperties>(), It.IsAny<byte[]>())).Callback<string, string, bool, bool, IBasicProperties, byte[]>
(
(a1, a2, a3, a4, a5, a6) =>
{
var basicProps = a5;
props.Add(basicProps);
var springProps = new DefaultMessagePropertiesConverter().ToMessageProperties(basicProps, null, "UTF-8");
var replyMessage = new Message(Encoding.UTF8.GetBytes("!dlrow olleH"), springProps);
if (count.IncrementValueAndReturn() < 2)
{
var anotherMessage = new Message(Encoding.UTF8.GetBytes("Second"), springProps);
replyMessage = template.SendAndReceive(anotherMessage);
nestedReplyTo.Add(replyMessage.MessageProperties.ReplyTo);
nestedReplyStack.Add((string)replyMessage.MessageProperties.Headers[RabbitTemplate.STACKED_REPLY_TO_HEADER]);
nestedCorrelation.Add((string)replyMessage.MessageProperties.Headers[RabbitTemplate.STACKED_CORRELATION_HEADER]);
}
template.OnMessage(replyMessage);
});
var reply = template.SendAndReceive(message);
Assert.IsNotNull(reply);
Assert.AreEqual(2, props.Count);
var basicProperties = props[0];
Assert.AreEqual("new.replyTo", basicProperties.ReplyTo);
Assert.AreEqual("replyTo2:replyTo1", basicProperties.Headers[RabbitTemplate.STACKED_REPLY_TO_HEADER]);
Assert.IsTrue(((string)basicProperties.Headers[RabbitTemplate.STACKED_CORRELATION_HEADER]).EndsWith(":a"));
basicProperties = props[1];
Assert.AreEqual("new.replyTo", basicProperties.ReplyTo);
Assert.AreEqual("new.replyTo:replyTo2:replyTo1", basicProperties.Headers[RabbitTemplate.STACKED_REPLY_TO_HEADER]);
Assert.IsTrue(((string)basicProperties.Headers[RabbitTemplate.STACKED_CORRELATION_HEADER]).EndsWith(":a"));
Assert.AreEqual("replyTo1", reply.MessageProperties.Headers[RabbitTemplate.STACKED_REPLY_TO_HEADER]);
Assert.AreEqual("a", reply.MessageProperties.Headers[RabbitTemplate.STACKED_CORRELATION_HEADER]);
Assert.AreEqual(1, nestedReplyTo.Count);
Assert.AreEqual(1, nestedReplyStack.Count);
Assert.AreEqual(1, nestedCorrelation.Count);
Assert.AreEqual("replyTo2:replyTo1", nestedReplyStack[0]);
Assert.IsTrue(nestedCorrelation[0].EndsWith(":a"));
}
public MessageProperties ToMessageProperties(IBasicProperties source, Envelope envelope, Encoding charset)
{
var target = new MessageProperties();
var headers = source.Headers;
if (headers.Count > 0)
{
foreach (var entry in headers)
{
var key = entry.Key;
if (MessageProperties.X_DELAY.Equals(key))
{
var value = entry.Value;
if (value is int)
{
target.ReceivedDelay = (int)value;
}
}
else
{
target.SetHeader(key, entry.Value);
}
}
}
target.Timestamp = DateTimeOffset.FromUnixTimeMilliseconds(source.Timestamp.UnixTime).DateTime;
target.MessageId = source.MessageId;
target.ReceivedUserId = source.UserId;
target.AppId = source.AppId;
target.ClusterId = source.ClusterId;
target.Type = source.Type;
var deliveryMode = (int)source.DeliveryMode;
target.ReceivedDeliveryMode = (MessageDeliveryMode)Enum.ToObject(typeof(MessageDeliveryMode), deliveryMode);
target.DeliveryMode = null;
target.Expiration = source.Expiration;
target.Priority = source.Priority;
target.ContentType = source.ContentType;
target.ContentEncoding = source.ContentEncoding;
var correlationId = source.CorrelationId;
if (!string.IsNullOrEmpty(correlationId))
{
target.CorrelationId = correlationId;
}
var replyTo = source.ReplyTo;
if (replyTo != null)
{
target.ReplyTo = replyTo;
}
if (envelope != null)
{
target.ReceivedExchange = envelope.Exchange;
target.ReceivedRoutingKey = envelope.RoutingKey;
target.Redelivered = envelope.Redeliver;
target.DeliveryTag = envelope.DeliveryTag;
}
return(target);
}