Populate originator property on saga creation (#457) (#465)

* add failing tests

* Set originator value when creating new saga

Author: timbussmann

src/NServiceBus.Testing.Tests/Sagas/ReplyToOriginator.cs

@@ -0,0 +1,84 @@
+namespace NServiceBus.Testing.Tests.Sagas
+{
+    using System;
+    using System.Collections.Generic;
+    using System.Linq;
+    using System.Threading.Tasks;
+    using NUnit.Framework;
+
+    [TestFixture]
+    public class ReplyToOriginator
+    {
+        [Test]
+        public async Task ReplyToOriginatorShouldReplyToInitialOriginator()
+        {
+            const string originatorAddress = "expectedReplyAddress";
+
+            var saga = new TestableSaga<ReplyingSaga, ReplyingSagaData>();
+            // the Originator value is populated by the header value, not the context property
+            await saga.Handle(new StartSagaMessage() { CorrelationProperty = Guid.NewGuid() }, messageHeaders: new Dictionary<string, string>()
+            {
+                {Headers.ReplyToAddress, originatorAddress}
+            });
+            var result = await saga.HandleQueuedMessage();
+
+            var reply = result.Context.RepliedMessages.SingleOrDefault();
+            Assert.NotNull(reply);
+            Assert.AreEqual(originatorAddress, reply.Options.GetDestination());
+            Assert.AreEqual(originatorAddress, reply.Message<ReplyMessage>().OriginatorAddress);
+        }
+
+        [Test]
+        public async Task OriginatorShouldBeSetByDefault()
+        {
+            // ensure the testing API also works without explicitly defining a replyTo header value
+            var saga = new TestableSaga<ReplyingSaga, ReplyingSagaData>();
+
+            await saga.Handle(new StartSagaMessage() { CorrelationProperty = Guid.NewGuid() });
+            var result = await saga.HandleQueuedMessage();
+
+            var reply = result.Context.RepliedMessages.SingleOrDefault();
+            Assert.NotNull(reply);
+            string replyAddress = reply.Options.GetDestination();
+            Assert.NotNull(replyAddress);
+            Assert.AreEqual(replyAddress, reply.Message<ReplyMessage>().OriginatorAddress);
+        }
+
+        class ReplyingSaga : NServiceBus.Saga<ReplyingSagaData>, IAmStartedByMessages<StartSagaMessage>, IHandleMessages<SendReplyMessage>
+        {
+            protected override void ConfigureHowToFindSaga(SagaPropertyMapper<ReplyingSagaData> mapper) => mapper
+                .ConfigureMapping<StartSagaMessage>(m => m.CorrelationProperty)
+                .ToSaga(s => s.CorrelationProperty);
+
+            public Task Handle(StartSagaMessage message, IMessageHandlerContext context)
+            {
+                return context.SendLocal(new SendReplyMessage());
+            }
+
+            public Task Handle(SendReplyMessage message, IMessageHandlerContext context)
+            {
+                return ReplyToOriginator(context, new ReplyMessage { OriginatorAddress = Data.Originator });
+            }
+        }
+
+        class ReplyingSagaData : ContainSagaData
+        {
+            public Guid CorrelationProperty { get; set; }
+        }
+
+        class StartSagaMessage : IMessage
+        {
+            public Guid CorrelationProperty { get; set; }
+        }
+
+        class SendReplyMessage : ICommand
+        {
+            public Guid CorrelationProperty { get; set; }
+        }
+
+        class ReplyMessage : IMessage
+        {
+            public string OriginatorAddress { get; set; }
+        }
+    }
+}

src/NServiceBus.Testing/Sagas/TestableSaga.cs

@@ -210,13 +210,11 @@ async Task<HandleResult> InnerHandle(QueuedSagaMessage message, string handleMet
 
             using (var session = new NonDurableSynchronizedStorageSession())
             {
-                var contextBag = new ContextBag();
-
-                var loadResult = await LoadSagaData(message, session, contextBag).ConfigureAwait(false);
+                var loadResult = await LoadSagaData(message, session, context).ConfigureAwait(false);
                 saga.Entity = loadResult.Item1;
 
                 await sagaMapper.InvokeHandlerMethod(saga, handleMethodName, message, context).ConfigureAwait(false);
-                await SaveSagaData(saga, loadResult.Item2, loadResult.Item3, session, contextBag).ConfigureAwait(false);
+                await SaveSagaData(saga, loadResult.Item2, loadResult.Item3, session, context.Extensions).ConfigureAwait(false);
                 await session.CompleteAsync().ConfigureAwait(false);
             }
 
@@ -270,16 +268,14 @@ async Task<HandleResult> HandleTimeout(OutgoingMessage<object, SendOptions> time
             return await InnerHandle(queueMessage, methodName, context).ConfigureAwait(false);
         }
 
-        async Task<Tuple<TSagaData, bool, object>> LoadSagaData(QueuedSagaMessage message, SynchronizedStorageSession session, ContextBag contextBag)
+        async Task<Tuple<TSagaData, bool, object>> LoadSagaData(QueuedSagaMessage message, SynchronizedStorageSession session, TestableMessageHandlerContext context)
         {
             var messageMetadata = sagaMapper.GetMessageMetadata(message.Type);
             TSagaData sagaData;
 
-            if (message.Headers != null &&
-                message.Headers.TryGetValue(Headers.SagaId, out var sagaIdString) &&
-                Guid.TryParse(sagaIdString, out Guid sagaId))
+            if (message.Headers.TryGetValue(Headers.SagaId, out var sagaIdString) && Guid.TryParse(sagaIdString, out Guid sagaId))
             {
-                sagaData = await persister.Get<TSagaData>(sagaId, session, contextBag).ConfigureAwait(false);
+                sagaData = await persister.Get<TSagaData>(sagaId, session, context.Extensions).ConfigureAwait(false);
                 if (sagaData != null)
                 {
                     return new Tuple<TSagaData, bool, object>(sagaData, false, null);
@@ -288,7 +284,7 @@ async Task<Tuple<TSagaData, bool, object>> LoadSagaData(QueuedSagaMessage messag
 
             var messageMappedValue = sagaMapper.GetMessageMappedValue(message);
 
-            sagaData = await persister.Get<TSagaData>(sagaMapper.CorrelationPropertyName, messageMappedValue, session, contextBag).ConfigureAwait(false);
+            sagaData = await persister.Get<TSagaData>(sagaMapper.CorrelationPropertyName, messageMappedValue, session, context.Extensions).ConfigureAwait(false);
 
             if (sagaData != null)
             {
@@ -297,7 +293,10 @@ async Task<Tuple<TSagaData, bool, object>> LoadSagaData(QueuedSagaMessage messag
 
             if (messageMetadata.IsAllowedToStartSaga)
             {
-                sagaData = new TSagaData { Id = Guid.NewGuid() };
+                var originatorAddress = message.Headers.TryGetValue(Headers.ReplyToAddress, out var replyAddress)
+                    ? replyAddress
+                    : context.ReplyToAddress; // This property has a default value set even when the header isn't set to require less setup for testing
+                sagaData = new TSagaData { Id = Guid.NewGuid(), Originator = originatorAddress };
                 sagaMapper.SetCorrelationPropertyValue(sagaData, messageMappedValue);
                 return new Tuple<TSagaData, bool, object>(sagaData, true, messageMappedValue);
             }