finished refactoring config cretor

2025-06-18 23:38:16 +08:00 · 2017-03-05 16:56:41 +00:00
parent 8bbd781820
commit 558a0dfdab
7 changed files with 261 additions and 20 deletions
--- a/test/Ocelot.UnitTests/Configuration/RateLimitOptionsCreatorTests.cs
+++ b/test/Ocelot.UnitTests/Configuration/RateLimitOptionsCreatorTests.cs
@ -0,0 +1,108 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using Ocelot.Configuration;
+using Ocelot.Configuration.Builder;
+using Ocelot.Configuration.Creator;
+using Ocelot.Configuration.File;
+using Shouldly;
+using TestStack.BDDfy;
+using Xunit;
+
+namespace Ocelot.UnitTests.Configuration
+{
+    public class RateLimitOptionsCreatorTests
+    {
+        private FileReRoute _fileReRoute;
+        private FileGlobalConfiguration _fileGlobalConfig;
+        private bool _enabled;
+        private RateLimitOptionsCreator _creator;
+        private RateLimitOptions _result;
+
+        public RateLimitOptionsCreatorTests()
+        {
+            _creator = new RateLimitOptionsCreator();
+        }
+
+        [Fact]
+        public void should_create_rate_limit_options()
+        {
+            var fileReRoute = new FileReRoute
+            {
+                RateLimitOptions = new FileRateLimitRule
+                {
+                    ClientWhitelist = new List<string>(),
+                    Period = "Period",
+                    Limit = 1,
+                    PeriodTimespan = 1,
+                    EnableRateLimiting = true
+                }
+            };
+            var fileGlobalConfig = new FileGlobalConfiguration
+            {
+                RateLimitOptions = new FileRateLimitOptions
+                {
+                    ClientIdHeader = "ClientIdHeader",
+                    DisableRateLimitHeaders = true,
+                    QuotaExceededMessage = "QuotaExceededMessage",
+                    RateLimitCounterPrefix = "RateLimitCounterPrefix",
+                    HttpStatusCode = 200
+                }
+            };
+            var expected = new RateLimitOptionsBuilder()
+                .WithClientIdHeader("ClientIdHeader")
+                .WithClientWhiteList(fileReRoute.RateLimitOptions.ClientWhitelist)
+                .WithDisableRateLimitHeaders(true)
+                .WithEnableRateLimiting(true)
+                .WithHttpStatusCode(200)
+                .WithQuotaExceededMessage("QuotaExceededMessage")
+                .WithRateLimitCounterPrefix("RateLimitCounterPrefix")
+                .WithRateLimitRule(new RateLimitRule(fileReRoute.RateLimitOptions.Period,
+                       TimeSpan.FromSeconds(fileReRoute.RateLimitOptions.PeriodTimespan),
+                       fileReRoute.RateLimitOptions.Limit))
+                .Build();
+
+            this.Given(x => x.GivenTheFollowingFileReRoute(fileReRoute))
+                .And(x => x.GivenTheFollowingFileGlobalConfig(fileGlobalConfig))
+                .And(x => x.GivenRateLimitingIsEnabled())
+                .When(x => x.WhenICreate())
+                .Then(x => x.ThenTheFollowingIsReturned(expected))
+                .BDDfy();
+        }
+
+        private void GivenTheFollowingFileReRoute(FileReRoute fileReRoute)
+        {
+            _fileReRoute = fileReRoute;
+        }
+
+        private void GivenTheFollowingFileGlobalConfig(FileGlobalConfiguration fileGlobalConfig)
+        {
+            _fileGlobalConfig = fileGlobalConfig;
+        }
+
+        private void GivenRateLimitingIsEnabled()
+        {
+            _enabled = true;
+        }
+
+        private void WhenICreate()
+        {
+            _result = _creator.Create(_fileReRoute, _fileGlobalConfig, _enabled);
+        }
+
+        private void ThenTheFollowingIsReturned(RateLimitOptions expected)
+        {
+            _result.ClientIdHeader.ShouldBe(expected.ClientIdHeader);
+            _result.ClientWhitelist.ShouldBe(expected.ClientWhitelist);
+            _result.DisableRateLimitHeaders.ShouldBe(expected.DisableRateLimitHeaders);
+            _result.EnableRateLimiting.ShouldBe(expected.EnableRateLimiting);
+            _result.HttpStatusCode.ShouldBe(expected.HttpStatusCode);
+            _result.QuotaExceededMessage.ShouldBe(expected.QuotaExceededMessage);
+            _result.RateLimitCounterPrefix.ShouldBe(expected.RateLimitCounterPrefix);
+            _result.RateLimitRule.Limit.ShouldBe(expected.RateLimitRule.Limit);
+            _result.RateLimitRule.Period.ShouldBe(expected.RateLimitRule.Period);
+            _result.RateLimitRule.PeriodTimespan.Ticks.ShouldBe(expected.RateLimitRule.PeriodTimespan.Ticks);
+        }
+    }
+}