#529 removed consul deps and introduced delegate to find service discovery provider (#538)

test/Ocelot.UnitTests/Configuration/ConsulFileConfigurationRepositoryTests.cs

@@ -1,263 +0,0 @@
-namespace Ocelot.UnitTests.Configuration
-{
-    using Xunit;
-    using TestStack.BDDfy;
-    using Shouldly;
-    using Ocelot.Configuration.Repository;
-    using Moq;
-    using Ocelot.Infrastructure.Consul;
-    using Ocelot.Logging;
-    using Ocelot.Configuration.File;
-    using Ocelot.Cache;
-    using System;
-    using System.Collections.Generic;
-    using Ocelot.Responses;
-    using System.Threading.Tasks;
-    using Ocelot.Configuration;
-    using Ocelot.Configuration.Builder;
-    using Ocelot.ServiceDiscovery.Configuration;
-    using Consul;
-    using Newtonsoft.Json;
-    using System.Text;
-    using System.Threading;
-    using System.Linq;
-
-    public class ConsulFileConfigurationRepositoryTests
-    {
-        private ConsulFileConfigurationRepository _repo;
-        private Mock<IOcelotCache<FileConfiguration>> _cache;
-        private Mock<IInternalConfigurationRepository> _internalRepo;
-        private Mock<IConsulClientFactory> _factory;
-        private Mock<IOcelotLoggerFactory> _loggerFactory;
-        private Mock<IConsulClient> _client;
-        private Mock<IKVEndpoint> _kvEndpoint;
-        private FileConfiguration _fileConfiguration;
-        private Response _setResult;
-        private Response<FileConfiguration> _getResult;
-
-        public ConsulFileConfigurationRepositoryTests()
-        {
-            _cache = new Mock<IOcelotCache<FileConfiguration>>();
-            _internalRepo = new Mock<IInternalConfigurationRepository>();
-            _loggerFactory = new Mock<IOcelotLoggerFactory>();
-
-            _factory = new Mock<IConsulClientFactory>();
-            _client = new Mock<IConsulClient>();
-            _kvEndpoint = new Mock<IKVEndpoint>();
-
-            _client
-                .Setup(x => x.KV)
-                .Returns(_kvEndpoint.Object);
-
-            _factory
-                .Setup(x => x.Get(It.IsAny<ConsulRegistryConfiguration>()))
-                .Returns(_client.Object);
-
-            _internalRepo
-                .Setup(x => x.Get())
-                .Returns(new OkResponse<IInternalConfiguration>(new InternalConfiguration(new List<ReRoute>(), "", new ServiceProviderConfigurationBuilder().Build(), "", It.IsAny<LoadBalancerOptions>(), It.IsAny<string>(), It.IsAny<QoSOptions>(), It.IsAny<HttpHandlerOptions>())));
-            
-            _repo = new ConsulFileConfigurationRepository(_cache.Object, _internalRepo.Object, _factory.Object, _loggerFactory.Object);
-        }
-
-        [Fact]
-        public void should_set_config()
-        {
-            var config = FakeFileConfiguration();
-
-            this.Given(_ => GivenIHaveAConfiguration(config))
-                .And(_ => GivenWritingToConsulSucceeds())
-                .When(_ => WhenISetTheConfiguration())
-                .Then(_ => ThenTheConfigurationIsStoredAs(config))
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_get_config()
-        {
-            var config = FakeFileConfiguration();
-
-             this.Given(_ => GivenIHaveAConfiguration(config))
-                .And(_ => GivenFetchFromConsulSucceeds())
-                .When(_ => WhenIGetTheConfiguration())
-                .Then(_ => ThenTheConfigurationIs(config))
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_get_null_config()
-        {
-             this.Given(_ => GivenFetchFromConsulReturnsNull())
-                .When(_ => WhenIGetTheConfiguration())
-                .Then(_ => ThenTheConfigurationIsNull())
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_get_config_from_cache()
-        {
-            var config = FakeFileConfiguration();
-
-             this.Given(_ => GivenIHaveAConfiguration(config))
-                .And(_ => GivenFetchFromCacheSucceeds())
-                .When(_ => WhenIGetTheConfiguration())
-                .Then(_ => ThenTheConfigurationIs(config))
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_set_config_key()
-        {
-            var config = FakeFileConfiguration();
-
-            this.Given(_ => GivenIHaveAConfiguration(config))
-                .And(_ => GivenTheConfigKeyComesFromFileConfig("Tom"))
-                .And(_ => GivenFetchFromConsulSucceeds())
-                .When(_ => WhenIGetTheConfiguration())
-                .And(_ => ThenTheConfigKeyIs("Tom"))
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_set_default_config_key()
-        {
-            var config = FakeFileConfiguration();
-
-            this.Given(_ => GivenIHaveAConfiguration(config))
-                .And(_ => GivenFetchFromConsulSucceeds())
-                .When(_ => WhenIGetTheConfiguration())
-                .And(_ => ThenTheConfigKeyIs("InternalConfiguration"))
-                .BDDfy();
-        }
-
-        private void ThenTheConfigKeyIs(string expected)
-        {
-            _kvEndpoint
-                .Verify(x => x.Get(expected, It.IsAny<CancellationToken>()), Times.Once);
-        }
-
-        private void GivenTheConfigKeyComesFromFileConfig(string key)
-        {
-            _internalRepo
-                .Setup(x => x.Get())
-                .Returns(new OkResponse<IInternalConfiguration>(new InternalConfiguration(new List<ReRoute>(), "",
-                    new ServiceProviderConfigurationBuilder().WithConfigurationKey(key).Build(), "",
-                    new LoadBalancerOptionsBuilder().Build(), "", new QoSOptionsBuilder().Build(),
-                    new HttpHandlerOptionsBuilder().Build())));
-            
-            _repo = new ConsulFileConfigurationRepository(_cache.Object, _internalRepo.Object, _factory.Object, _loggerFactory.Object);
-        }
-
-        private void ThenTheConfigurationIsNull()
-        {
-            _getResult.Data.ShouldBeNull();
-        }
-
-        private void ThenTheConfigurationIs(FileConfiguration config)
-        {
-            var expected = JsonConvert.SerializeObject(config, Formatting.Indented);
-            var result = JsonConvert.SerializeObject(_getResult.Data, Formatting.Indented);
-            result.ShouldBe(expected);
-        }
-
-        private async Task WhenIGetTheConfiguration()
-        {
-            _getResult = await _repo.Get();
-        }
-
-        private void GivenWritingToConsulSucceeds()
-        {
-            var response = new WriteResult<bool>();
-            response.Response = true;
-
-            _kvEndpoint
-                .Setup(x => x.Put(It.IsAny<KVPair>(), It.IsAny<CancellationToken>())).ReturnsAsync(response);
-        }
-
-         private void GivenFetchFromCacheSucceeds()
-        {
-            _cache.Setup(x => x.Get(It.IsAny<string>(), It.IsAny<string>())).Returns(_fileConfiguration);
-        }
-
-        private void GivenFetchFromConsulReturnsNull()
-        {
-            QueryResult<KVPair> result = new QueryResult<KVPair>();
-
-            _kvEndpoint
-                .Setup(x => x.Get(It.IsAny<string>(), It.IsAny<CancellationToken>()))
-                .ReturnsAsync(result);
-        }
-
-        private void GivenFetchFromConsulSucceeds()
-        {
-            var json = JsonConvert.SerializeObject(_fileConfiguration, Formatting.Indented);
-
-            var bytes = Encoding.UTF8.GetBytes(json);
-
-            var kvp = new KVPair("OcelotConfiguration");
-            kvp.Value = bytes;
-
-            var query = new QueryResult<KVPair>();
-            query.Response = kvp;
-
-            _kvEndpoint
-                .Setup(x => x.Get(It.IsAny<string>(), It.IsAny<CancellationToken>()))
-                .ReturnsAsync(query);
-        }
-
-        private void ThenTheConfigurationIsStoredAs(FileConfiguration config)
-        {
-            var json = JsonConvert.SerializeObject(config, Formatting.Indented);
-
-            var bytes = Encoding.UTF8.GetBytes(json);
-
-            _kvEndpoint
-                .Verify(x => x.Put(It.Is<KVPair>(k => k.Value.SequenceEqual(bytes)), It.IsAny<CancellationToken>()), Times.Once);
-        }
-
-        private async Task WhenISetTheConfiguration()
-        {
-            _setResult = await _repo.Set(_fileConfiguration);
-        }
-
-        private void GivenIHaveAConfiguration(FileConfiguration config)
-        {
-            _fileConfiguration = config;
-        }
-        
-        private FileConfiguration FakeFileConfiguration()
-        {
-            var reRoutes = new List<FileReRoute>
-            {
-                new FileReRoute
-                {
-                    DownstreamHostAndPorts = new List<FileHostAndPort>
-                    {
-                        new FileHostAndPort
-                        {
-                            Host = "123.12.12.12",
-                            Port = 80,
-                        }
-                    },
-                    DownstreamScheme = "https",
-                    DownstreamPathTemplate = "/asdfs/test/{test}"
-                }
-            };
-
-            var globalConfiguration = new FileGlobalConfiguration
-            {
-                ServiceDiscoveryProvider = new FileServiceDiscoveryProvider
-                {
-                    Port = 198,
-                    Host = "blah"
-                }
-            };
-
-            return new FileConfiguration
-            {
-                GlobalConfiguration = globalConfiguration,
-                ReRoutes = reRoutes
-            };
-        }
-    }
-}

test/Ocelot.UnitTests/DependencyInjection/OcelotBuilderTests.cs

@@ -1,28 +1,23 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Net.Http;
-using Microsoft.AspNetCore.Hosting;
-using Microsoft.AspNetCore.Hosting.Internal;
-using Microsoft.Extensions.Configuration;
-using Microsoft.Extensions.DependencyInjection;
-using Ocelot.Cache;
-using Ocelot.Configuration;
-using Ocelot.Configuration.File;
-using Ocelot.Configuration.Setter;
-using Ocelot.DependencyInjection;
-using Ocelot.Requester;
-using Ocelot.UnitTests.Requester;
-using Shouldly;
-using IdentityServer4.AccessTokenValidation;
-using TestStack.BDDfy;
-using Xunit;
-using static Ocelot.UnitTests.Middleware.UserDefinedResponseAggregatorTests;
-using Ocelot.Middleware.Multiplexer;
-
 namespace Ocelot.UnitTests.DependencyInjection
 {
-    using Butterfly;
+    using System;
+    using System.Collections.Generic;
+    using System.Linq;
+    using System.Net.Http;
+    using Microsoft.AspNetCore.Hosting;
+    using Microsoft.AspNetCore.Hosting.Internal;
+    using Microsoft.Extensions.Configuration;
+    using Microsoft.Extensions.DependencyInjection;
+    using Ocelot.Configuration.Setter;
+    using Ocelot.DependencyInjection;
+    using Ocelot.Requester;
+    using Ocelot.UnitTests.Requester;
+    using Shouldly;
+    using IdentityServer4.AccessTokenValidation;
+    using TestStack.BDDfy;
+    using Xunit;
+    using static Ocelot.UnitTests.Middleware.UserDefinedResponseAggregatorTests;
+    using Ocelot.Middleware.Multiplexer;
 
     public class OcelotBuilderTests
     {
@@ -80,15 +75,6 @@ namespace Ocelot.UnitTests.DependencyInjection
                 .BDDfy();
         }
 
-        [Fact]
-        public void should_set_up_consul()
-        {            
-            this.Given(x => WhenISetUpOcelotServices())
-                .When(x => WhenISetUpConsul())
-                .Then(x => ThenAnExceptionIsntThrown())
-                .BDDfy();
-        }
-
         [Fact]
         public void should_set_up_rafty()
         {            
@@ -271,18 +257,6 @@ namespace Ocelot.UnitTests.DependencyInjection
             first.ShouldBe(second);
         }
 
-        private void WhenISetUpConsul()
-        {
-            try
-            {
-                _ocelotBuilder.AddStoreOcelotConfigurationInConsul();
-            }
-            catch (Exception e)
-            {
-                _ex = e;
-            }       
-        }
-
         private void WhenISetUpRafty()
         {
             try

test/Ocelot.UnitTests/Infrastructure/InMemoryBusTests.cs

@@ -22,7 +22,7 @@ namespace Ocelot.UnitTests.Infrastructure
                 called = true;
             });
             _bus.Publish(new object(), 1);
-            await Task.Delay(10);
+            await Task.Delay(100);
             called.ShouldBeTrue();
         }
 

test/Ocelot.UnitTests/ServiceDiscovery/ConsulServiceDiscoveryProviderTests.cs

@@ -1,297 +0,0 @@
-using System;
-using System.Collections.Generic;
-using System.IO;
-using System.Linq;
-using Consul;
-using Microsoft.AspNetCore.Builder;
-using Microsoft.AspNetCore.Hosting;
-using Microsoft.AspNetCore.Http;
-using Moq;
-using Ocelot.Infrastructure.Consul;
-using Ocelot.Logging;
-using Ocelot.ServiceDiscovery.Configuration;
-using Ocelot.ServiceDiscovery.Providers;
-using Ocelot.Values;
-using Xunit;
-using TestStack.BDDfy;
-using Shouldly;
-
-namespace Ocelot.UnitTests.ServiceDiscovery
-{
-    public class ConsulServiceDiscoveryProviderTests : IDisposable
-    {
-        private IWebHost _fakeConsulBuilder;
-        private readonly List<ServiceEntry> _serviceEntries;
-        private ConsulServiceDiscoveryProvider _provider;
-        private readonly string _serviceName;
-        private readonly int _port;
-        private readonly string _consulHost;
-        private readonly string _fakeConsulServiceDiscoveryUrl;
-        private List<Service> _services;
-        private readonly Mock<IOcelotLoggerFactory> _factory;
-        private readonly Mock<IOcelotLogger> _logger;
-        private string _receivedToken;
-        private readonly IConsulClientFactory _clientFactory;
-
-        public ConsulServiceDiscoveryProviderTests()
-        {
-            _serviceName = "test";
-            _port = 8500;
-            _consulHost = "localhost";
-            _fakeConsulServiceDiscoveryUrl = $"http://{_consulHost}:{_port}";
-            _serviceEntries = new List<ServiceEntry>();
-
-            _factory = new Mock<IOcelotLoggerFactory>();
-            _clientFactory = new ConsulClientFactory();
-            _logger = new Mock<IOcelotLogger>();
-            _factory.Setup(x => x.CreateLogger<ConsulServiceDiscoveryProvider>()).Returns(_logger.Object);
-
-            var config = new ConsulRegistryConfiguration(_consulHost, _port, _serviceName, null);
-            _provider = new ConsulServiceDiscoveryProvider(config, _factory.Object, _clientFactory);
-        }
-
-        [Fact]
-        public void should_return_service_from_consul()
-        {
-            var serviceEntryOne = new ServiceEntry()
-            {
-                Service = new AgentService()
-                {
-                    Service = _serviceName,
-                    Address = "localhost",
-                    Port = 50881,
-                    ID = Guid.NewGuid().ToString(),
-                    Tags = new string[0]
-                },
-            };
-
-            this.Given(x =>GivenThereIsAFakeConsulServiceDiscoveryProvider(_fakeConsulServiceDiscoveryUrl, _serviceName))
-                .And(x => GivenTheServicesAreRegisteredWithConsul(serviceEntryOne))
-                .When(x => WhenIGetTheServices())
-                .Then(x => ThenTheCountIs(1))
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_use_token()
-        {
-            var token = "test token";
-            var config = new ConsulRegistryConfiguration(_consulHost, _port, _serviceName, token);
-            _provider = new ConsulServiceDiscoveryProvider(config, _factory.Object, _clientFactory);
-
-            var serviceEntryOne = new ServiceEntry()
-            {
-                Service = new AgentService()
-                {
-                    Service = _serviceName,
-                    Address = "localhost",
-                    Port = 50881,
-                    ID = Guid.NewGuid().ToString(),
-                    Tags = new string[0]
-                },
-            };
-
-            this.Given(_ => GivenThereIsAFakeConsulServiceDiscoveryProvider(_fakeConsulServiceDiscoveryUrl, _serviceName))
-                .And(_ => GivenTheServicesAreRegisteredWithConsul(serviceEntryOne))
-                .When(_ => WhenIGetTheServices())
-                .Then(_ => ThenTheCountIs(1))
-                .And(_ => _receivedToken.ShouldBe(token))
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_not_return_services_with_invalid_address()
-        {
-            var serviceEntryOne = new ServiceEntry()
-            {
-                Service = new AgentService()
-                {
-                    Service = _serviceName,
-                    Address = "http://localhost",
-                    Port = 50881,
-                    ID = Guid.NewGuid().ToString(),
-                    Tags = new string[0]
-                },
-            };
-
-            var serviceEntryTwo = new ServiceEntry()
-            {
-                Service = new AgentService()
-                {
-                    Service = _serviceName,
-                    Address = "http://localhost",
-                    Port = 50888,
-                    ID = Guid.NewGuid().ToString(),
-                    Tags = new string[0]
-                },
-            };
-
-            this.Given(x => GivenThereIsAFakeConsulServiceDiscoveryProvider(_fakeConsulServiceDiscoveryUrl, _serviceName))
-                .And(x => GivenTheServicesAreRegisteredWithConsul(serviceEntryOne, serviceEntryTwo))
-                .When(x => WhenIGetTheServices())
-                .Then(x => ThenTheCountIs(0))
-                .And(x => ThenTheLoggerHasBeenCalledCorrectlyForInvalidAddress())
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_not_return_services_with_empty_address()
-        {
-            var serviceEntryOne = new ServiceEntry()
-            {
-                Service = new AgentService()
-                {
-                    Service = _serviceName,
-                    Address = "",
-                    Port = 50881,
-                    ID = Guid.NewGuid().ToString(),
-                    Tags = new string[0]
-                },
-            };
-
-            var serviceEntryTwo = new ServiceEntry()
-            {
-                Service = new AgentService()
-                {
-                    Service = _serviceName,
-                    Address = null,
-                    Port = 50888,
-                    ID = Guid.NewGuid().ToString(),
-                    Tags = new string[0]
-                },
-            };
-
-            this.Given(x => GivenThereIsAFakeConsulServiceDiscoveryProvider(_fakeConsulServiceDiscoveryUrl, _serviceName))
-                .And(x => GivenTheServicesAreRegisteredWithConsul(serviceEntryOne, serviceEntryTwo))
-                .When(x => WhenIGetTheServices())
-                .Then(x => ThenTheCountIs(0))
-                .And(x => ThenTheLoggerHasBeenCalledCorrectlyForEmptyAddress())
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_not_return_services_with_invalid_port()
-        {
-            var serviceEntryOne = new ServiceEntry()
-            {
-                Service = new AgentService()
-                {
-                    Service = _serviceName,
-                    Address = "localhost",
-                    Port = -1,
-                    ID = Guid.NewGuid().ToString(),
-                    Tags = new string[0]
-                },
-            };
-
-            var serviceEntryTwo = new ServiceEntry()
-            {
-                Service = new AgentService()
-                {
-                    Service = _serviceName,
-                    Address = "localhost",
-                    Port = 0,
-                    ID = Guid.NewGuid().ToString(),
-                    Tags = new string[0]
-                },
-            };
-
-            this.Given(x => GivenThereIsAFakeConsulServiceDiscoveryProvider(_fakeConsulServiceDiscoveryUrl, _serviceName))
-                .And(x => GivenTheServicesAreRegisteredWithConsul(serviceEntryOne, serviceEntryTwo))
-                .When(x => WhenIGetTheServices())
-                .Then(x => ThenTheCountIs(0))
-                .And(x => ThenTheLoggerHasBeenCalledCorrectlyForInvalidPorts())
-                .BDDfy();
-        }
-
-        private void ThenTheLoggerHasBeenCalledCorrectlyForInvalidAddress()
-        {
-            _logger.Verify(
-                x => x.LogWarning(
-                    "Unable to use service Address: http://localhost and Port: 50881 as it is invalid. Address must contain host only e.g. localhost and port must be greater than 0"),
-                Times.Once);
-
-            _logger.Verify(
-                x => x.LogWarning(
-                    "Unable to use service Address: http://localhost and Port: 50888 as it is invalid. Address must contain host only e.g. localhost and port must be greater than 0"),
-                Times.Once);
-        }
-
-        private void ThenTheLoggerHasBeenCalledCorrectlyForEmptyAddress()
-        {
-            _logger.Verify(
-                x => x.LogWarning(
-                    "Unable to use service Address:  and Port: 50881 as it is invalid. Address must contain host only e.g. localhost and port must be greater than 0"),
-                Times.Once);
-
-            _logger.Verify(
-                x => x.LogWarning(
-                    "Unable to use service Address:  and Port: 50888 as it is invalid. Address must contain host only e.g. localhost and port must be greater than 0"),
-                Times.Once);
-        }
-
-        private void ThenTheLoggerHasBeenCalledCorrectlyForInvalidPorts()
-        {
-            _logger.Verify(
-                x => x.LogWarning(
-                    "Unable to use service Address: localhost and Port: -1 as it is invalid. Address must contain host only e.g. localhost and port must be greater than 0"),
-                Times.Once);
-
-            _logger.Verify(
-                x => x.LogWarning(
-                    "Unable to use service Address: localhost and Port: 0 as it is invalid. Address must contain host only e.g. localhost and port must be greater than 0"),
-                Times.Once);
-        }
-
-        private void ThenTheCountIs(int count)
-        {
-            _services.Count.ShouldBe(count);
-        }
-
-        private void WhenIGetTheServices()
-        {
-            _services = _provider.Get().GetAwaiter().GetResult();
-        }
-
-        private void GivenTheServicesAreRegisteredWithConsul(params ServiceEntry[] serviceEntries)
-        {
-            foreach (var serviceEntry in serviceEntries)
-            {
-                _serviceEntries.Add(serviceEntry);
-            }
-        }
-
-        private void GivenThereIsAFakeConsulServiceDiscoveryProvider(string url, string serviceName)
-        {
-            _fakeConsulBuilder = new WebHostBuilder()
-                .UseUrls(url)
-                .UseKestrel()
-                .UseContentRoot(Directory.GetCurrentDirectory())
-                .UseIISIntegration()
-                .UseUrls(url)
-                .Configure(app =>
-                {
-                    app.Run(async context =>
-                    {
-                        if (context.Request.Path.Value == $"/v1/health/service/{serviceName}")
-                        {
-                            if (context.Request.Headers.TryGetValue("X-Consul-Token", out var values))
-                            {
-                                _receivedToken = values.First();
-                            }
-
-                            await context.Response.WriteJsonAsync(_serviceEntries);
-                        }
-                    });
-                })
-                .Build();
-
-            _fakeConsulBuilder.Start();
-        }
-
-        public void Dispose()
-        {
-            _fakeConsulBuilder?.Dispose();
-        }
-    }
-}

test/Ocelot.UnitTests/ServiceDiscovery/PollingConsulServiceDiscoveryProviderTests.cs

@@ -1,80 +0,0 @@
-namespace Ocelot.UnitTests.ServiceDiscovery
-{
-    using System;
-    using System.Collections.Generic;
-    using Moq;
-    using Ocelot.Logging;
-    using Ocelot.ServiceDiscovery.Providers;
-    using Ocelot.Values;
-    using Xunit;
-    using TestStack.BDDfy;
-    using Shouldly;
-    using static Ocelot.Infrastructure.Wait;
-
-    public class PollingConsulServiceDiscoveryProviderTests
-    {
-        private readonly int _delay;
-        private PollingConsulServiceDiscoveryProvider _provider;
-        private readonly List<Service> _services;
-        private readonly Mock<IOcelotLoggerFactory> _factory;
-        private readonly Mock<IOcelotLogger> _logger;
-        private readonly Mock<IServiceDiscoveryProvider> _consulServiceDiscoveryProvider;
-        private List<Service> _result;
-
-        public PollingConsulServiceDiscoveryProviderTests()
-        {
-            _services = new List<Service>();
-            _delay = 1;
-            _factory = new Mock<IOcelotLoggerFactory>();
-            _logger = new Mock<IOcelotLogger>();
-            _factory.Setup(x => x.CreateLogger<PollingConsulServiceDiscoveryProvider>()).Returns(_logger.Object);
-            _consulServiceDiscoveryProvider = new Mock<IServiceDiscoveryProvider>();
-        }
-
-        [Fact]
-        public void should_return_service_from_consul()
-        {
-            var service = new Service("", new ServiceHostAndPort("", 0), "", "", new List<string>());
-
-            this.Given(x => GivenConsulReturns(service))
-                .When(x => WhenIGetTheServices(1))
-                .Then(x => ThenTheCountIs(1))
-                .BDDfy();
-        }
-
-        private void GivenConsulReturns(Service service)
-        {
-            _services.Add(service);
-            _consulServiceDiscoveryProvider.Setup(x => x.Get()).ReturnsAsync(_services);
-        }
-
-        private void ThenTheCountIs(int count)
-        {
-            _result.Count.ShouldBe(count);
-        }
-
-        private void WhenIGetTheServices(int expected)
-        {
-            _provider = new PollingConsulServiceDiscoveryProvider(_delay, "", _factory.Object, _consulServiceDiscoveryProvider.Object);
-
-            var result = WaitFor(3000).Until(() => {
-                try
-                {
-                    _result = _provider.Get().GetAwaiter().GetResult();
-                    if(_result.Count == expected)
-                    {
-                        return true;
-                    }
-
-                    return false;
-                }
-                catch(Exception)
-                {
-                    return false;
-                }
-            });
-
-            result.ShouldBeTrue();
-        }
-    }
-}

test/Ocelot.UnitTests/ServiceDiscovery/ServiceFabricServiceDiscoveryProviderTests.cs

@@ -1,23 +1,12 @@
-using Ocelot.ServiceDiscovery.Configuration;
-using Ocelot.ServiceDiscovery.Providers;
-
-namespace Ocelot.UnitTests.ServiceDiscovery
+namespace Ocelot.UnitTests.ServiceDiscovery
 {
-    using System;
     using System.Collections.Generic;
-    using System.IO;
-    using System.Text;
-    using Consul;
-    using Microsoft.AspNetCore.Builder;
-    using Microsoft.AspNetCore.Hosting;
-    using Microsoft.AspNetCore.Http;
-    using Moq;
-    using Ocelot.Logging;
-    using Ocelot.ServiceDiscovery;
     using Ocelot.Values;
     using Xunit;
     using TestStack.BDDfy;
     using Shouldly;
+    using Ocelot.ServiceDiscovery.Configuration;
+    using Ocelot.ServiceDiscovery.Providers;
 
     public class ServiceFabricServiceDiscoveryProviderTests
     {

test/Ocelot.UnitTests/ServiceDiscovery/ServiceProviderFactoryTests.cs

@@ -1,39 +1,41 @@
-using System;
-using System.Collections.Generic;
-using Moq;
-using Ocelot.Configuration;
-using Ocelot.Configuration.Builder;
-using Ocelot.Infrastructure.Consul;
-using Ocelot.Logging;
-using Ocelot.ServiceDiscovery;
-using Ocelot.ServiceDiscovery.Providers;
-using Shouldly;
-using TestStack.BDDfy;
-using Xunit;
-
 namespace Ocelot.UnitTests.ServiceDiscovery
 {
-    using Pivotal.Discovery.Client;
+    using System;
+    using System.Threading.Tasks;
+    using Microsoft.Extensions.DependencyInjection;
     using Steeltoe.Common.Discovery;
+    using Values;
+    using System.Collections.Generic;
+    using Moq;
+    using Ocelot.Configuration;
+    using Ocelot.Configuration.Builder;
+    using Ocelot.Logging;
+    using Ocelot.ServiceDiscovery;
+    using Ocelot.ServiceDiscovery.Providers;
+    using Shouldly;
+    using TestStack.BDDfy;
+    using Xunit;
 
     public class ServiceProviderFactoryTests
     {
         private ServiceProviderConfiguration _serviceConfig;
         private IServiceDiscoveryProvider _result;
-        private readonly ServiceDiscoveryProviderFactory _factory;
+        private ServiceDiscoveryProviderFactory _factory;
         private DownstreamReRoute _reRoute;
         private Mock<IOcelotLoggerFactory> _loggerFactory;
         private Mock<IDiscoveryClient> _discoveryClient;
         private Mock<IOcelotLogger> _logger;
+        private IServiceProvider _provider;
+        private IServiceCollection _collection;
 
         public ServiceProviderFactoryTests()
         {
             _loggerFactory = new Mock<IOcelotLoggerFactory>();
             _logger = new Mock<IOcelotLogger>();
-            _loggerFactory.Setup(x => x.CreateLogger<PollingConsulServiceDiscoveryProvider>()).Returns(_logger.Object);
             _discoveryClient = new Mock<IDiscoveryClient>();
-            var consulClient = new Mock<IConsulClientFactory>();
-            _factory = new ServiceDiscoveryProviderFactory(_loggerFactory.Object, consulClient.Object, _discoveryClient.Object);
+            _collection = new ServiceCollection();
+            _provider = _collection.BuildServiceProvider();
+            _factory = new ServiceDiscoveryProviderFactory(_loggerFactory.Object, _discoveryClient.Object, _provider);
         }
         
         [Fact]
@@ -72,7 +74,7 @@ namespace Ocelot.UnitTests.ServiceDiscovery
         }
 
         [Fact]
-        public void should_return_consul_service_provider()
+        public void should_call_delegate()
         {
             var reRoute = new DownstreamReRouteBuilder()
                 .WithServiceName("product")
@@ -83,27 +85,9 @@ namespace Ocelot.UnitTests.ServiceDiscovery
                 .Build();
 
             this.Given(x => x.GivenTheReRoute(serviceConfig, reRoute))
+                .And(x => GivenAFakeDelegate())
                 .When(x => x.WhenIGetTheServiceProvider())
-                .Then(x => x.ThenTheServiceProviderIs<ConsulServiceDiscoveryProvider>())
-                .BDDfy();
-        }
-
-        [Fact]
-        public void should_return_polling_consul_service_provider()
-        {
-            var reRoute = new DownstreamReRouteBuilder()
-                .WithServiceName("product")
-                .WithUseServiceDiscovery(true)
-                .Build();
-
-            var serviceConfig = new ServiceProviderConfigurationBuilder()
-                .WithType("PollConsul")
-                .WithPollingInterval(100000)
-                .Build();
-
-            this.Given(x => x.GivenTheReRoute(serviceConfig, reRoute))
-                .When(x => x.WhenIGetTheServiceProvider())
-                .Then(x => x.ThenTheServiceProviderIs<PollingConsulServiceDiscoveryProvider>())
+                .Then(x => x.ThenTheDelegateIsCalled())
                 .BDDfy();
         }
 
@@ -143,6 +127,27 @@ namespace Ocelot.UnitTests.ServiceDiscovery
                 .BDDfy();
         }
 
+        private void GivenAFakeDelegate()
+        {
+            ServiceDiscoveryFinderDelegate fake = (provider, config, name) => new Fake();
+            _collection.AddSingleton(fake);
+            _provider = _collection.BuildServiceProvider();
+            _factory = new ServiceDiscoveryProviderFactory(_loggerFactory.Object, _discoveryClient.Object, _provider);
+        }
+
+        class Fake : IServiceDiscoveryProvider
+        {
+            public Task<List<Service>> Get()
+            {
+                return null;
+            }
+        }
+
+        private void ThenTheDelegateIsCalled()
+        {
+            _result.GetType().Name.ShouldBe("Fake");
+        }
+
         private void ThenTheFollowingServicesAreReturned(List<DownstreamHostAndPort> downstreamAddresses)
         {
             var result = (ConfigurationServiceProvider)_result;