fix test

internetarchive · Aug 6, 2024 · 9bdeb1d · 9bdeb1d
1 parent 95af078
commit 9bdeb1d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/internal/pkg/queue/enqueue_test.go b/internal/pkg/queue/enqueue_test.go
@@ -89,6 +89,7 @@ func TestEnqueue(t *testing.T) {
 		ElementsPerHost := q.GetElementsPerHost()
 		q.statsMutex.Lock()
 		defer q.statsMutex.Unlock()
+
 		if (*ElementsPerHost)["example.fr"] != 2 {
 			t.Fatalf("Expected ElementsPerHost[example.fr] to be 2, got %d", (*ElementsPerHost)["example.fr"])
 		}
@@ -254,6 +255,8 @@ func TestBatchEnqueue(t *testing.T) {
 
 		ElementsPerHost := q.GetElementsPerHost()
 		q.statsMutex.Lock()
+		defer q.statsMutex.Unlock()
+
 		if (*ElementsPerHost)["example.fr"] != 1 {
 			t.Fatalf("Expected ElementsPerHost[example.fr] to be 1, got %d", (*ElementsPerHost)["example.fr"])
 		}
@@ -304,6 +307,8 @@ func TestBatchEnqueue(t *testing.T) {
 
 		ElementsPerHost := q.GetElementsPerHost()
 		q.statsMutex.Lock()
+		defer q.statsMutex.Unlock()
+
 		if (*ElementsPerHost)["example.fr"] != 2 {
 			t.Fatalf("Expected ElementsPerHost[example.fr] to be 2, got %d", (*ElementsPerHost)["example.fr"])
 		}