tests: fixup paths to dramatiq.threading

RevenueCat · Apr 2, 2023 · c251564 · c251564
1 parent 1d10f28
commit c251564
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/common.py b/tests/common.py
@@ -6,7 +6,7 @@
 import pytest
 
 from dramatiq import Worker
-from dramatiq.middleware.threading import is_gevent_active
+from dramatiq.threading import is_gevent_active
 
 
 @contextmanager

diff --git a/tests/middleware/test_threading.py b/tests/middleware/test_threading.py
@@ -4,7 +4,7 @@
 
 import pytest
 
-from dramatiq.middleware import threading
+from dramatiq import threading
 
 not_supported = threading.current_platform not in threading.supported_platforms
 
@@ -46,7 +46,7 @@ def test_raise_thread_exception_on_nonexistent_thread(caplog):
     # I expect a 'failed to set exception' critical message to be logged
     expected_message = "Failed to set exception (Interrupt) in thread %d." % thread_id
     assert caplog.record_tuples == [
-        ("dramatiq.middleware.threading", logging.CRITICAL, expected_message),
+        ("dramatiq.threading", logging.CRITICAL, expected_message),
     ]
 
 
@@ -59,7 +59,7 @@ def test_raise_thread_exception_unsupported_platform(caplog, monkeypatch):
 
     # I expect a 'platform not supported' critical message to be logged
     assert caplog.record_tuples == [
-        ("dramatiq.middleware.threading", logging.CRITICAL, (
+        ("dramatiq.threading", logging.CRITICAL, (
             "Setting thread exceptions (Interrupt) is not supported "
             "for your current platform ('not supported')."
         )),