Ocelot/test/Ocelot.UnitTests/Middleware/OcelotPipelineExtensionsTests.cs

namespace Ocelot.UnitTests.Middleware
{
    using Microsoft.AspNetCore.Builder;
    using Microsoft.AspNetCore.Http;
    using Microsoft.Extensions.Configuration;
    using Microsoft.Extensions.DependencyInjection;
    using Ocelot.DependencyInjection;
    using Ocelot.DownstreamRouteFinder.Middleware;
    using Ocelot.DownstreamUrlCreator.Middleware;
    using Ocelot.LoadBalancer.Middleware;
    using Ocelot.Middleware;
    using Ocelot.Request.Middleware;
    using Ocelot.WebSockets.Middleware;
    using Shouldly;
    using TestStack.BDDfy;
    using Xunit;

    public class OcelotPipelineExtensionsTests
    {
        private ApplicationBuilder _builder;
        private RequestDelegate _handlers;

        [Fact]
        public void should_set_up_pipeline()
        {
            this.Given(_ => GivenTheDepedenciesAreSetUp())
                 .When(_ => WhenIBuild())
                 .Then(_ => ThenThePipelineIsBuilt())
                 .BDDfy();
        }

        [Fact]
        public void should_expand_pipeline()
        {
            this.Given(_ => GivenTheDepedenciesAreSetUp())
                 .When(_ => WhenIExpandBuild())
                 .Then(_ => ThenThePipelineIsBuilt())
                 .BDDfy();
        }

        private void ThenThePipelineIsBuilt()
        {
            _handlers.ShouldNotBeNull();
        }

        private void WhenIBuild()
        {
            _handlers = _builder.BuildOcelotPipeline(new OcelotPipelineConfiguration());
        }

        private void WhenIExpandBuild()
        {
            OcelotPipelineConfiguration configuration = new OcelotPipelineConfiguration();
            //Func<HttpContext,  bool>, Action<IApplicationBuilder>
            configuration.MapWhenOcelotPipeline.Add((httpContext) => httpContext.WebSockets.IsWebSocketRequest, app =>
            {
                app.UseDownstreamRouteFinderMiddleware();
                app.UseDownstreamRequestInitialiser();
                app.UseLoadBalancingMiddleware();
                app.UseDownstreamUrlCreatorMiddleware();
                app.UseWebSocketsProxyMiddleware();
            });
            _handlers = _builder.BuildOcelotPipeline(new OcelotPipelineConfiguration());
        }

        private void GivenTheDepedenciesAreSetUp()
        {
            IConfigurationBuilder test = new ConfigurationBuilder();
            var root = test.Build();
            var services = new ServiceCollection();
            services.AddSingleton<IConfiguration>(root);
            services.AddOcelot();
            var provider = services.BuildServiceProvider();
            _builder = new ApplicationBuilder(provider);
        }
    }
}