TimmyGilissen / NServiceBus.Extensions.IntegrationTesting

NServiceBus.Extensions.IntegrationTesting

Usage

This library extends the NServiceBus.Extensions.Hosting package to provide extensions for integration testing for messaging-based applications.

Typical integration tests with NServiceBus manually execute a single handler at a time. However, an entire application might handle multiple messages in a chain.

This package provides two extensions:

• Testing-friendly configuration for NServiceBus EndpointConfiguration
• A test fixture to consume in your integration test, typically with the ASP.NET Core MVC testing package.

To use, first create a WebApplicationFactory instance as you would with ASP.NET Core:

public class TestFactory : WebApplicationFactory<Startup> 
{
}

Next, you will need to override your normal host building to provide test-specific configuration for NServiceBus:

protected override IHostBuilder CreateHostBuilder() =>
     Host.CreateDefaultBuilder()
         .UseNServiceBus(ctxt =>
         {
             var endpoint = new EndpointConfiguration("HostApplicationFactoryTests");

             // Set up NServiceBus with testing-friendly defaults
             endpoint.ConfigureTestEndpoint();

             // Set up any of your other configuration here

             return endpoint;
         })
         .ConfigureWebHostDefaults(b => b.UseStartup<Startup>());

Finally, the EndpointFixture methods are static, so you might want to import the static EndpointFixture methods. This is not strictly necessary but makes readiblity better in tests:

using static NServiceBus.Extensions.IntegrationTesting.EndpointFixture;

Typically with xUnit, this factory becomes a fixture:

public class HostApplicationFactoryTests 
    : IClassFixture<TestFactory>
{
    private readonly TestFactory _factory;

    public HostApplicationFactoryTests(TestFactory factory) => _factory = factory;

In your tests, you can call the various overloads to send a message and wait in HostingApplicationFactory:

[Fact]
public async Task Can_send_and_wait()
{
    var firstMessage = new FirstMessage {Message = "Hello World"};

    var session = _factory.Services.GetService<IMessageSession>()_

    var result = await ExecuteAndWaitForHandled<FinalMessage>(() => session.SendLocal(firstMessage)));

    result.IncomingMessageContexts.Count.ShouldBe(3);
    result.OutgoingMessageContexts.Count.ShouldBe(3);

    result.ReceivedMessages.ShouldNotBeEmpty();

    var message = result.ReceivedMessages.OfType<FinalMessage>().Single();

    message.Message.ShouldBe(firstMessage.Message);
}

In the above, the FinalMessage is the expected "final message" after sending the first message. Either the method returns with a list of all sent and received messages, or it will time out.

You can wait for specific messages to be sent or received, as well as supply a custom timeout (if startup takes a while).

Saga testing

In the example above, it shows a way to interact with a given message that has been handled.

A more complex scenario could be is having an integration test for working with Saga's.

Before you start testing the saga don't forget to add the following line of code to your CreateHostBuilder.

endpoint.UsePersistence<LearningPersistence>();

E.g: To complete the business process you need input from various systems. When all that input is captured and processed, the saga is complete and can be marked for completion.

In this case it is not possible to act on a certain type of message. In that specific case you can use ExecuteAndWaitForHandled.

[Fact]
public async Task Will_wait_for_saga_to_be_completed()
{
    var firstMessage = new FirstMessage {Message = "Hello World"};

    var session = _factory.Services.GetService<IMessageSession>();

    var result = await ExecuteAndWaitForSagaCompletion<SagaExample>(() => session.SendLocal(firstMessage));

    var saga = result.InvokedHandlers.Single(x =>
        x.MessageHandler.HandlerType == typeof(SagaExample)).GetSagaInstance();
    
    Assert.NotNull(saga);
    
    Assert.Equal(firstMessage.Message, ((SagaData)saga.Instance.Entity).Message);
}

The function GetSagaInstance() is the following extension method. It makes it possible to easily cast the IInvokeHandlerContest to an ActiveSagaInstance.

public static class InvokeHandlerContextExtension
{
    public static ActiveSagaInstance GetSagaInstance(this IInvokeHandlerContext context)
    {
        return context.Extensions.TryGet(out ActiveSagaInstance saga) ? saga : null;
    }
}

The test example also demonstrates how you can validate the SagaData that has been set during the lifecycle of the saga.

Assert.Equal(firstMessage.Message, ((SagaData)saga.Instance.Entity).Message);

Multiple endpoints

If you have multiple endpoints to test in a single integration test, you can create a single fixture that includes a WebApplicationFactory for each endpoint:

public class SystemFixture : IDisposable
{
    public WebAppFactory WebAppHost { get; }

    public WebApplicationFactory<Program> WorkerHost { get; }

    public ChildWorkerServiceFactory ChildWorkerHost { get; }

Unfortunately, the hosts aren't started until the first time a client is created so you may want to include a Start method:

public void Start()
{
    WorkerHost.CreateClient();
    WebAppHost.CreateClient();
    ChildWorkerHost.CreateClient();
}

public void Dispose()
{
    WebAppHost?.Dispose();
    WorkerHost?.Dispose();
    ChildWorkerHost?.Dispose();
}

Xunit usage

In XUnit, tests execute in parallel by default. With async messaging, this means that multiple tests sending in parallel will overwrite each other.

To get around this, configure your fixture to be a collection fixture with parallelization turned off:

[CollectionDefinition(nameof(SystemCollection), DisableParallelization = true)]
public class SystemCollection : ICollectionFixture<SystemFixture> { }

Then in your integration test, you create a constructor as normal but add the collection fixture definition:

[Collection(nameof(SystemCollection))]
public class EndToEndTests : XunitContextBase
{
    private readonly SystemFixture _fixture;

    public EndToEndTests(SystemFixture fixture, ITestOutputHelper output) : base(output)
    {
        _fixture = fixture;
        _fixture.Start();
    }

In this case I am also using the XunitContext project to output console messages to the test window so that we can see the log messages from the started hosts.