fix(distribution): add more tests for expected behaviour

Part of #81
form3tech-oss · Jun 10, 2024 · 817300c · 817300c
1 parent 48f6dbc
commit 817300c
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/internal/trigger/api/iteration_distribution_test.go b/internal/trigger/api/iteration_distribution_test.go
@@ -31,6 +31,22 @@ func TestRegularRateDistribution(t *testing.T) {
 			rate:                     10,
 			expectedDistributedRates: []int{5, 5},
 		},
+		{
+			iterationDuration:        215 * time.Millisecond,
+			rate:                     1,
+			expectedDistributedRates: []int{0, 1},
+		},
+		{
+			iterationDuration:        299 * time.Millisecond,
+			rate:                     1,
+			expectedDistributedRates: []int{0, 1},
+		},
+
+		{
+			iterationDuration:        300 * time.Millisecond,
+			rate:                     1,
+			expectedDistributedRates: []int{0, 0, 1},
+		},
 		{
 			iterationDuration:        900 * time.Millisecond,
 			rate:                     7,