Add tests

filinvadim · Sep 3, 2024 · 554d5ff · 554d5ff
1 parent 520b4c4
commit 554d5ff
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/limiter_test.go b/limiter_test.go
@@ -80,6 +80,10 @@ func TestLimiter_ExceedLimit(t *testing.T) {
 
 	wg.Wait()
 
+	if !limiter.IsLocked() {
+		t.Fatalf("Expected limiter to be locked")
+	}
+
 	duration := time.Since(start)
 	if duration < interval {
 		t.Fatalf("Expected at least %v wait time, but got %v", interval, duration)
@@ -135,7 +139,9 @@ func TestLimiter_ExceedWeightedLimit(t *testing.T) {
 	}
 
 	wg.Wait()
-
+	if !limiter.IsLocked() {
+		t.Fatalf("Expected limiter to be locked")
+	}
 	duration := time.Since(start)
 	if duration < interval {
 		t.Fatalf("Expected at least %v wait time, but got %v", interval, duration)