diff --git a/src/Polly.Specs/Helpers/RateLimit/IRateLimiterExtensions.cs b/src/Polly.Specs/Helpers/RateLimit/IRateLimiterExtensions.cs index 4d48dc376ac..64da0965670 100644 --- a/src/Polly.Specs/Helpers/RateLimit/IRateLimiterExtensions.cs +++ b/src/Polly.Specs/Helpers/RateLimit/IRateLimiterExtensions.cs @@ -4,7 +4,7 @@ namespace Polly.Specs.Helpers.RateLimit { - public static class IRateLimiterExtensions + internal static class IRateLimiterExtensions { public static void ShouldPermitAnExecution(this IRateLimiter rateLimiter) { diff --git a/src/Polly.Specs/RateLimit/LockBasedTokenBucketRateLimiterTests.cs b/src/Polly.Specs/RateLimit/LockBasedTokenBucketRateLimiterTests.cs index 23be20161bc..775f47361f6 100644 --- a/src/Polly.Specs/RateLimit/LockBasedTokenBucketRateLimiterTests.cs +++ b/src/Polly.Specs/RateLimit/LockBasedTokenBucketRateLimiterTests.cs @@ -5,7 +5,7 @@ namespace Polly.Specs.RateLimit { public class LockBasedTokenBucketRateLimiterTests : TokenBucketRateLimiterTestsBase { - public override IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity) + internal override IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity) => new LockBasedTokenBucketRateLimiter(onePer, bucketCapacity); } } diff --git a/src/Polly.Specs/RateLimit/LockFreeTokenBucketRateLimiterTests.cs b/src/Polly.Specs/RateLimit/LockFreeTokenBucketRateLimiterTests.cs index c9af7fa837e..31376594f63 100644 --- a/src/Polly.Specs/RateLimit/LockFreeTokenBucketRateLimiterTests.cs +++ b/src/Polly.Specs/RateLimit/LockFreeTokenBucketRateLimiterTests.cs @@ -5,7 +5,7 @@ namespace Polly.Specs.RateLimit { public class LockFreeTokenBucketRateLimiterTests : TokenBucketRateLimiterTestsBase { - public override IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity) + internal override IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity) => new LockFreeTokenBucketRateLimiter(onePer, bucketCapacity); } } diff --git a/src/Polly.Specs/RateLimit/TokenBucketRateLimiterTestsBase.cs b/src/Polly.Specs/RateLimit/TokenBucketRateLimiterTestsBase.cs index 0eefc960fa6..c296f0d5bf7 100644 --- a/src/Polly.Specs/RateLimit/TokenBucketRateLimiterTestsBase.cs +++ b/src/Polly.Specs/RateLimit/TokenBucketRateLimiterTestsBase.cs @@ -13,7 +13,7 @@ namespace Polly.Specs.RateLimit [Collection(Polly.Specs.Helpers.Constants.SystemClockDependentTestCollection)] public abstract class TokenBucketRateLimiterTestsBase : RateLimitSpecsBase, IDisposable { - public abstract IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity); + internal abstract IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity); public void Dispose() { diff --git a/src/Polly/RateLimit/IRateLimiter.cs b/src/Polly/RateLimit/IRateLimiter.cs index 9136bbcc201..79e724f3acb 100644 --- a/src/Polly/RateLimit/IRateLimiter.cs +++ b/src/Polly/RateLimit/IRateLimiter.cs @@ -5,7 +5,7 @@ namespace Polly.RateLimit /// /// Defines methods to be provided by a rate-limiter used in a Polly /// - public interface IRateLimiter + internal interface IRateLimiter { /// /// Returns whether the execution is permitted; if not, returns what should be waited before retrying. diff --git a/src/Polly/RateLimit/LockBasedTokenBucketRateLimiter.cs b/src/Polly/RateLimit/LockBasedTokenBucketRateLimiter.cs index 6ea0dfa6ef9..26feb3d05cd 100644 --- a/src/Polly/RateLimit/LockBasedTokenBucketRateLimiter.cs +++ b/src/Polly/RateLimit/LockBasedTokenBucketRateLimiter.cs @@ -6,7 +6,7 @@ namespace Polly.RateLimit /// /// A lock-based token-bucket rate-limiter for a Polly . /// - public class LockBasedTokenBucketRateLimiter : IRateLimiter + internal class LockBasedTokenBucketRateLimiter : IRateLimiter { private readonly long addTokenTickInterval; private readonly long bucketCapacity; diff --git a/src/Polly/RateLimit/LockFreeTokenBucketRateLimiter.cs b/src/Polly/RateLimit/LockFreeTokenBucketRateLimiter.cs index 152fb60b433..42571d735be 100644 --- a/src/Polly/RateLimit/LockFreeTokenBucketRateLimiter.cs +++ b/src/Polly/RateLimit/LockFreeTokenBucketRateLimiter.cs @@ -7,7 +7,7 @@ namespace Polly.RateLimit /// /// A lock-free token-bucket rate-limiter for a Polly . /// - public class LockFreeTokenBucketRateLimiter : IRateLimiter + internal class LockFreeTokenBucketRateLimiter : IRateLimiter { private readonly long addTokenTickInterval; private readonly long bucketCapacity;