src/MassTransit/Testing/ConsumerTestHarnessExtensions.cs - develop - vshmidt/masstransit

src/MassTransit/Testing/ConsumerTestHarnessExtensions.cs

52 строки2 KB

Chris Patterson

Integration of all external packages into MassTransit, split out into Abstractions, Middleware, and the core MassTransit assembly.

22 янв 2022, 18:24

22 янв 2022, 18:24b5b4f50

namespace MassTransit.Testing
{
    using System;
    using Consumer;


    public static class ConsumerTestHarnessExtensions
    {
        public static ConsumerTestHarness<T> Consumer<T>(this BusTestHarness harness, string queueName = null)
            where T : class, IConsumer, new()
        {
            var consumerFactory = new DefaultConstructorConsumerFactory<T>();

            return new ConsumerTestHarness<T>(harness, consumerFactory, queueName);
        }

        public static ConsumerTestHarness<T> Consumer<T>(this BusTestHarness harness, Action<IConsumerConfigurator<T>> configure,
            string queueName = null)
            where T : class, IConsumer, new()
        {
            var consumerFactory = new DefaultConstructorConsumerFactory<T>();

            return new ConsumerTestHarness<T>(harness, consumerFactory, configure, queueName);
        }

        public static ConsumerTestHarness<T> Consumer<T>(this BusTestHarness harness, IConsumerFactory<T> consumerFactory, string queueName = null)
            where T : class, IConsumer, new()
        {
            return new ConsumerTestHarness<T>(harness, consumerFactory, queueName);
        }

        public static ConsumerTestHarness<T> Consumer<T>(this BusTestHarness harness, IConsumerFactory<T> consumerFactory,
            Action<IConsumerConfigurator<T>> configure, string queueName = null)
            where T : class, IConsumer, new()
        {
            return new ConsumerTestHarness<T>(harness, consumerFactory, configure, queueName);
        }

        public static ConsumerTestHarness<T> Consumer<T>(this BusTestHarness harness, Func<T> consumerFactoryMethod, string queueName = null)
            where T : class, IConsumer
        {
            return new ConsumerTestHarness<T>(harness, new DelegateConsumerFactory<T>(consumerFactoryMethod), queueName);
        }

        public static ConsumerTestHarness<T> Consumer<T>(this BusTestHarness harness, Func<T> consumerFactoryMethod,
            Action<IConsumerConfigurator<T>> configure, string queueName = null)
            where T : class, IConsumer
        {
            return new ConsumerTestHarness<T>(harness, new DelegateConsumerFactory<T>(consumerFactoryMethod), configure, queueName);
        }
    }
}