diff --git a/tests/Color-Chan.Discord.Rest.Tests/Policies/DiscordRateLimitPolicyTests.cs b/tests/Color-Chan.Discord.Rest.Tests/Policies/DiscordRateLimitPolicyTests.cs index 71e0a4b8..b43aefe9 100644 --- a/tests/Color-Chan.Discord.Rest.Tests/Policies/DiscordRateLimitPolicyTests.cs +++ b/tests/Color-Chan.Discord.Rest.Tests/Policies/DiscordRateLimitPolicyTests.cs @@ -6,10 +6,12 @@ using System.Threading.Tasks; using Color_Chan.Discord.Caching.Extensions; using Color_Chan.Discord.Caching.Services; +using Color_Chan.Discord.Rest.Configurations; using Color_Chan.Discord.Rest.Policies; using FluentAssertions; using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.Logging; +using Microsoft.Extensions.Options; using NUnit.Framework; using Polly; @@ -30,7 +32,8 @@ public async Task Should_successfully_pass_global_rateLimit_policy(string limit, .AddColorChanCache() .AddLogging() .BuildServiceProvider(); - var policy = new DiscordRateLimitPolicy(services.GetRequiredService(), services.GetRequiredService>()); + var restConfig = new OptionsWrapper(new RestConfiguration()); + var policy = new DiscordRateLimitPolicy(services.GetRequiredService(), services.GetRequiredService>(), restConfig); var context = new Context { { "endpoint", endpoint }, { "method", method } }; var message = new HttpResponseMessage(); @@ -53,7 +56,8 @@ public async Task Should_successfully_pass_rateLimit_policy(string limit, string .AddColorChanCache() .AddLogging() .BuildServiceProvider(); - var policy = new DiscordRateLimitPolicy(services.GetRequiredService(), services.GetRequiredService>()); + var restConfig = new OptionsWrapper(new RestConfiguration()); + var policy = new DiscordRateLimitPolicy(services.GetRequiredService(), services.GetRequiredService>(), restConfig); var context = new Context { { "endpoint", endpoint }, { "method", method } }; var message = new HttpResponseMessage(); @@ -83,7 +87,8 @@ public async Task Should_handle_global_rateLimit(string retryAfter, string endpo .AddColorChanCache() .AddLogging() .BuildServiceProvider(); - var policy = new DiscordRateLimitPolicy(services.GetRequiredService(), services.GetRequiredService>()); + var restConfig = new OptionsWrapper(new RestConfiguration()); + var policy = new DiscordRateLimitPolicy(services.GetRequiredService(), services.GetRequiredService>(), restConfig); var context = new Context { { "endpoint", endpoint }, { "method", "GET" } }; var message = new HttpResponseMessage(); message.StatusCode = HttpStatusCode.TooManyRequests; @@ -112,7 +117,8 @@ public async Task Should_prevent_rateLimit(string limit, string remaining, strin .AddColorChanCache() .AddLogging() .BuildServiceProvider(); - var policy = new DiscordRateLimitPolicy(services.GetRequiredService(), services.GetRequiredService>()); + var restConfig = new OptionsWrapper(new RestConfiguration()); + var policy = new DiscordRateLimitPolicy(services.GetRequiredService(), services.GetRequiredService>(), restConfig); var context = new Context { { "endpoint", endpoint }, { "method", method } }; var message = new HttpResponseMessage(); message.StatusCode = HttpStatusCode.OK;