Cover DelegateInvokingLoadBalancerCreator by unit tests.

2025-11-04 09:55:28 +08:00 · 2019-07-10 21:11:34 +02:00
parent 22f304cecf
commit f9ce987cc5
1 changed files with 102 additions and 0 deletions
--- a/test/Ocelot.UnitTests/LoadBalancer/DelegateInvokingLoadBalancerCreatorTests.cs
+++ b/test/Ocelot.UnitTests/LoadBalancer/DelegateInvokingLoadBalancerCreatorTests.cs
@@ -0,0 +1,102 @@
+using System;
+using System.Threading.Tasks;
+using Moq;
+using Ocelot.Configuration;
+using Ocelot.Configuration.Builder;
+using Ocelot.LoadBalancer.LoadBalancers;
+using Ocelot.Middleware;
+using Ocelot.Responses;
+using Ocelot.ServiceDiscovery.Providers;
+using Ocelot.Values;
+using Shouldly;
+using TestStack.BDDfy;
+using Xunit;
+
+namespace Ocelot.UnitTests.LoadBalancer
+{
+    public class DelegateInvokingLoadBalancerCreatorTests
+    {
+        private readonly DelegateInvokingLoadBalancerCreator<FakeLoadBalancer> _creator;
+        private readonly Func<DownstreamReRoute, IServiceDiscoveryProvider, ILoadBalancer> _creatorFunc;
+        private readonly Mock<IServiceDiscoveryProvider> _serviceProvider;
+        private DownstreamReRoute _reRoute;
+        private ILoadBalancer _loadBalancer;
+        private string _typeName;
+
+        public DelegateInvokingLoadBalancerCreatorTests()
+        {
+            _creatorFunc = (reRoute, serviceDiscoveryProvider) =>
+                new FakeLoadBalancer(reRoute, serviceDiscoveryProvider);
+            _creator = new DelegateInvokingLoadBalancerCreator<FakeLoadBalancer>(_creatorFunc);
+            _serviceProvider = new Mock<IServiceDiscoveryProvider>();
+        }
+
+        [Fact]
+        public void should_return_expected_name()
+        {
+            this.When(x => x.WhenIGetTheLoadBalancerTypeName())
+                .Then(x => x.ThenTheLoadBalancerTypeIs("FakeLoadBalancer"))
+                .BDDfy();
+        }
+
+        [Fact]
+        public void should_return_result_of_specified_creator_func()
+        {
+            var reRoute = new DownstreamReRouteBuilder()
+                .Build();
+
+            this.Given(x => x.GivenAReRoute(reRoute))
+                .When(x => x.WhenIGetTheLoadBalancer())
+                .Then(x => x.ThenTheLoadBalancerIsReturned<FakeLoadBalancer>())
+                .BDDfy();
+        }
+
+        private void GivenAReRoute(DownstreamReRoute reRoute)
+        {
+            _reRoute = reRoute;
+        }
+
+        private void WhenIGetTheLoadBalancer()
+        {
+            _loadBalancer = _creator.Create(_reRoute, _serviceProvider.Object);
+        }
+
+        private void WhenIGetTheLoadBalancerTypeName()
+        {
+            _typeName = _creator.Type;
+        }
+
+        private void ThenTheLoadBalancerIsReturned<T>()
+            where T : ILoadBalancer
+        {
+            _loadBalancer.ShouldBeOfType<T>();
+        }
+
+        private void ThenTheLoadBalancerTypeIs(string type)
+        {
+            _typeName.ShouldBe(type);
+        }
+
+        private class FakeLoadBalancer : ILoadBalancer
+        {
+            public FakeLoadBalancer(DownstreamReRoute reRoute, IServiceDiscoveryProvider serviceDiscoveryProvider)
+            {
+                ReRoute = reRoute;
+                ServiceDiscoveryProvider = serviceDiscoveryProvider;
+            }
+
+            public DownstreamReRoute ReRoute { get; }
+            public IServiceDiscoveryProvider ServiceDiscoveryProvider { get; }
+
+            public Task<Response<ServiceHostAndPort>> Lease(DownstreamContext context)
+            {
+                throw new NotImplementedException();
+            }
+
+            public void Release(ServiceHostAndPort hostAndPort)
+            {
+                throw new NotImplementedException();
+            }
+        }
+    }
+}