Fix non-generic rate-limit tests to be genuinely non-generic

App-vNext · Jul 16, 2019 · c916be5 · c916be5
1 parent daa42cb
commit c916be5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/src/Polly.Specs/RateLimit/AsyncRateLimitPolicySpecs.cs b/src/Polly.Specs/RateLimit/AsyncRateLimitPolicySpecs.cs
@@ -18,17 +18,17 @@ public void Dispose()
 
         protected override IRateLimitPolicy GetPolicyViaSyntax(int numberOfExecutions, TimeSpan perTimeSpan)
         {
-            return Policy.RateLimitAsync<ResultClassWithRetryAfter>(numberOfExecutions, perTimeSpan);
+            return Policy.RateLimitAsync(numberOfExecutions, perTimeSpan);
         }
 
         protected override IRateLimitPolicy GetPolicyViaSyntax(int numberOfExecutions, TimeSpan perTimeSpan, int maxBurst)
         {
-            return Policy.RateLimitAsync<ResultClassWithRetryAfter>(numberOfExecutions, perTimeSpan, maxBurst);
+            return Policy.RateLimitAsync(numberOfExecutions, perTimeSpan, maxBurst);
         }
 
         protected override (bool, TimeSpan) TryExecuteThroughPolicy(IRateLimitPolicy policy)
         {
-            if (policy is AsyncRateLimitPolicy<ResultClassWithRetryAfter> typedPolicy)
+            if (policy is AsyncRateLimitPolicy typedPolicy)
             {
                 try
                 {

diff --git a/src/Polly.Specs/RateLimit/RateLimitPolicySpecs.cs b/src/Polly.Specs/RateLimit/RateLimitPolicySpecs.cs
@@ -17,17 +17,17 @@ public void Dispose()
 
         protected override IRateLimitPolicy GetPolicyViaSyntax(int numberOfExecutions, TimeSpan perTimeSpan)
         {
-            return Policy.RateLimit<ResultClassWithRetryAfter>(numberOfExecutions, perTimeSpan);
+            return Policy.RateLimit(numberOfExecutions, perTimeSpan);
         }
 
         protected override IRateLimitPolicy GetPolicyViaSyntax(int numberOfExecutions, TimeSpan perTimeSpan, int maxBurst)
         {
-            return Policy.RateLimit<ResultClassWithRetryAfter>(numberOfExecutions, perTimeSpan, maxBurst);
+            return Policy.RateLimit(numberOfExecutions, perTimeSpan, maxBurst);
         }
 
         protected override (bool, TimeSpan) TryExecuteThroughPolicy(IRateLimitPolicy policy)
         {
-            if (policy is RateLimitPolicy<ResultClassWithRetryAfter> typedPolicy)
+            if (policy is RateLimitPolicy typedPolicy)
             {
                 try
                 {

diff --git a/src/Polly.Specs/RateLimit/RateLimitPolicyTResultSpecs.cs b/src/Polly.Specs/RateLimit/RateLimitPolicyTResultSpecs.cs
@@ -1,5 +1,4 @@
 using System;
-using System.Threading.Tasks;
 using Polly.RateLimit;
 using Polly.Specs.Helpers;
 using Polly.Specs.Helpers.RateLimit;