tests/MassTransit.Azure.ServiceBus.Core.Tests/AzureServiceBusTestFixture.cs - develop - vshmidt/masstransit

tests/MassTransit.Azure.ServiceBus.Core.Tests/AzureServiceBusTestFixture.cs

86 строк3 KB

Chris Patterson

Working on an inbox/outbox implementation for Entity Framework Core, InMemory, and ultimately additional transaction-based storage engines

17 июн 2022, 02:49

17 июн 2022, 02:4977e1f4b

namespace MassTransit.Azure.ServiceBus.Core.Tests
{
    using System;
    using System.Threading;
    using System.Threading.Tasks;
    using NUnit.Framework;
    using NUnit.Framework.Internal;
    using TestFramework;
    using Testing;


    [TestFixture]
    public abstract class AzureServiceBusTestFixture :
        BusTestFixture
    {
        TestExecutionContext _fixtureContext;
        protected AzureServiceBusTestHarness AzureServiceBusTestHarness { get; }

        public AzureServiceBusTestFixture(string inputQueueName = null, Uri serviceUri = null, ServiceBusTokenProviderSettings settings = null)
            : this(new AzureServiceBusTestHarness(
                serviceUri ?? AzureServiceBusEndpointUriCreator.Create(Configuration.ServiceNamespace, "MassTransit.Azure.ServiceBus.Core.Tests"),
                settings?.NamedKeyCredential ?? ((ServiceBusTokenProviderSettings)new TestAzureServiceBusAccountSettings()).NamedKeyCredential,
                inputQueueName))
        {
        }

        protected AzureServiceBusTestFixture(AzureServiceBusTestHarness harness)
            : base(harness)
        {
            AzureServiceBusTestHarness = harness;

            AzureServiceBusTestHarness.OnConfigureServiceBusBus += ConfigureServiceBusBus;
            AzureServiceBusTestHarness.OnConfigureServiceBusReceiveEndpoint += ConfigureServiceBusReceiveEndpoint;
        }

        protected string InputQueueName => AzureServiceBusTestHarness.InputQueueName;

        /// <summary>
        /// The sending endpoint for the InputQueue
        /// </summary>
        protected ISendEndpoint InputQueueSendEndpoint => AzureServiceBusTestHarness.InputQueueSendEndpoint;

        /// <summary>
        /// The sending endpoint for the Bus
        /// </summary>
        protected ISendEndpoint BusSendEndpoint => AzureServiceBusTestHarness.BusSendEndpoint;

        protected ISentMessageList Sent => AzureServiceBusTestHarness.Sent;

        protected Uri BusAddress => AzureServiceBusTestHarness.BusAddress;

        protected Uri HostAddress => AzureServiceBusTestHarness.HostAddress;

        protected Uri InputQueueAddress => AzureServiceBusTestHarness.InputQueueAddress;

        [OneTimeSetUp]
        public async Task SetupAzureServiceBusTestFixture()
        {
            _fixtureContext = TestExecutionContext.CurrentContext;

            LoggerFactory.Current = _fixtureContext;

            using var source = new CancellationTokenSource(TimeSpan.FromSeconds(20));

            await AzureServiceBusTestHarness.Start(source.Token);
        }

        [OneTimeTearDown]
        public async Task TearDownInMemoryTestFixture()
        {
            LoggerFactory.Current = _fixtureContext;

            await AzureServiceBusTestHarness.Stop();

            AzureServiceBusTestHarness.Dispose();
        }

        protected virtual void ConfigureServiceBusBus(IServiceBusBusFactoryConfigurator configurator)
        {
        }

        protected virtual void ConfigureServiceBusReceiveEndpoint(IServiceBusReceiveEndpointConfigurator configurator)
        {
        }
    }
}