Handle publish

src/NServiceBus.Extensions.IntegrationTesting/EndpointFixture.cs

@@ -88,7 +88,8 @@ private static async Task<ObservedMessageContexts> ExecuteAndWait<TMessageContex
                         }
 
                         break;
-                    case NsbActivityNames.OutgoingMessageActivityName:
+                    case NsbActivityNames.SendMessageActivityName:
+                    case NsbActivityNames.PublishMessageActivityName:
                         var outgoingContext = activitySource.GetTagItem("testing.outgoing.message.context") as IOutgoingLogicalMessageContext;
 
                         outgoingMessageContexts.Add(outgoingContext);

src/NServiceBus.Extensions.IntegrationTesting/NsbActivityNames.cs

@@ -4,7 +4,9 @@ internal static class NsbActivityNames
     {
         public const string IncomingMessageActivityName = "NServiceBus.Diagnostics.ReceiveMessage";
 
-        public const string OutgoingMessageActivityName = "NServiceBus.Diagnostics.SendMessage";
+        public const string SendMessageActivityName = "NServiceBus.Diagnostics.SendMessage";
+
+        public const string PublishMessageActivityName = "NServiceBus.Diagnostics.PublishMessage";
 
         public const string InvokeHandlerActivityName = "NServiceBus.Diagnostics.InvokeHandler";
     }

tests/NServiceBus.Extensions.IntegrationTesting.Tests/HostApplicationFactoryTests.cs

@@ -32,8 +32,8 @@ public async Task Can_send_and_wait()
 
             var result = await ExecuteAndWaitForHandled<FinalMessage>(() => session.SendLocal(firstMessage));
 
-            result.IncomingMessageContexts.Count.ShouldBe(3);
-            result.OutgoingMessageContexts.Count.ShouldBe(3);
+            result.IncomingMessageContexts.Count.ShouldBe(4);
+            result.OutgoingMessageContexts.Count.ShouldBe(4);
 
             result.ReceivedMessages.ShouldNotBeEmpty();
 
@@ -101,6 +101,11 @@ public class SecondMessage : ICommand
             public string Message { get; set; }
         }
 
+        public class ThirdMessage : IEvent
+        {
+            public string Message { get; set; }
+        }
+
         public class FinalMessage : ICommand
         {
             public string Message { get; set; }
@@ -120,6 +125,12 @@ public Task Handle(FirstMessage message, IMessageHandlerContext context) =>
         public class SecondHandler : IHandleMessages<SecondMessage>
         {
             public Task Handle(SecondMessage message, IMessageHandlerContext context) => 
+                context.Publish(new ThirdMessage {Message = message.Message});
+        }
+
+        public class ThirdHandler : IHandleMessages<ThirdMessage>
+        {
+            public Task Handle(ThirdMessage message, IMessageHandlerContext context) => 
                 context.SendLocal(new FinalMessage {Message = message.Message});
         }