From 9bdeb1d7cbc5e3fc0e02f5e9e9405dd1caf29fd8 Mon Sep 17 00:00:00 2001 From: yzqzss Date: Tue, 6 Aug 2024 16:43:04 +0800 Subject: [PATCH] fix test --- internal/pkg/queue/enqueue_test.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/internal/pkg/queue/enqueue_test.go b/internal/pkg/queue/enqueue_test.go index 9b360788..47708081 100644 --- 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"]) }