fix: refactoring throttling test

vostok · Dec 22, 2023 · ec4071a · ec4071a
1 parent c34c51e
commit ec4071a
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/Vostok.ClusterClient.Core.Tests/Modules/AdaptiveThrottlingModule_Tests.cs b/Vostok.ClusterClient.Core.Tests/Modules/AdaptiveThrottlingModule_Tests.cs
@@ -61,7 +61,7 @@ public void Should_create_adaptive_throttling_module_with_settings_for_all_prior
                 module.PerPriorityOptions(priority).Should().NotBeNull();
             }
         }
-        
+
         [Test]
         public void Should_correctly_compute_rejection_probability_for_different_priority()
         {
@@ -307,13 +307,13 @@ public void Should_not_account_according_to_priority_for_requests_still_in_progr
             Console.Out.WriteLine(module.RejectionProbability(priority));
         }
 
-        public static object[] PriorityCase =
+        public static IEnumerable<RequestPriority?> PriorityCase()
         {
-            new object[] {null},
-            new object[] {RequestPriority.Critical},
-            new object[] {RequestPriority.Ordinary},
-            new object[] {RequestPriority.Sheddable}
-        };
+            yield return null;
+            yield return RequestPriority.Critical;
+            yield return RequestPriority.Ordinary;
+            yield return RequestPriority.Sheddable;
+        }
 
         private void Accept(int count, RequestPriority? priority = null)
         {