From 1f4c67beb41b933fe3914bdb225f169d9b97af6a Mon Sep 17 00:00:00 2001 From: zhihao11ui Date: Wed, 16 Oct 2024 03:50:46 -0500 Subject: [PATCH] Fix Flaky Tests --- tests/test_redisqueue.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/test_redisqueue.py b/tests/test_redisqueue.py index 5b1f520..d7ebc04 100644 --- a/tests/test_redisqueue.py +++ b/tests/test_redisqueue.py @@ -59,6 +59,8 @@ def test_mock_queue_connection(): def test_mock_queue_put_get(): + mock_queue.connect() + mock_queue.clear() assert mock_queue.qsize() == 0 task = MockTask() @@ -75,6 +77,8 @@ def test_mock_queue_put_get(): def test_mock_queue_unique(): + mock_queue.connect() + mock_queue.clear() assert mock_queue.qsize() == 0 task = MockTask() @@ -98,6 +102,7 @@ def test_mock_queue_unique(): def test_mock_queue_get_put_same_task(): + mock_queue.connect() mock_queue.clear() task = MockTask()