Use monkeypatch for WorkerPool._sleep override.

derekpierre · Aug 8, 2024 · 1c22001 · 1c22001
1 parent 2c9a5e9
commit 1c22001
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/tests/test_bucket_sampling.py b/tests/test_bucket_sampling.py
@@ -24,15 +24,13 @@ def json(self):
 
 
 @pytest.fixture(autouse=True)
-def mock_worker_pool_sleep():
-
+def mock_worker_pool_sleep(monkeypatch):
     original = WorkerPool._sleep
 
     def _sleep(worker_pool, timeout):
         original(worker_pool, 0.01)
-        pass
 
-    WorkerPool._sleep = _sleep
+    monkeypatch.setattr(WorkerPool, "_sleep", _sleep)
 
 
 def test_bucket_sampling_schema(get_random_checksum_address):

diff --git a/tests/test_get_ursulas.py b/tests/test_get_ursulas.py
@@ -10,15 +10,13 @@
 
 
 @pytest.fixture(autouse=True)
-def mock_worker_pool_sleep():
-
+def mock_worker_pool_sleep(monkeypatch):
     original = WorkerPool._sleep
 
     def _sleep(worker_pool, timeout):
         original(worker_pool, 0.01)
-        pass
 
-    WorkerPool._sleep = _sleep
+    monkeypatch.setattr(WorkerPool, "_sleep", _sleep)
 
 
 def test_get_ursulas_schema(get_random_checksum_address):