use LuigiTestCase

spotify · Feb 1, 2025 · 2ec1144 · 2ec1144
1 parent 43c2856
commit 2ec1144
Show file tree

Hide file tree

Showing 3 changed files with 151 additions and 92 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -78,8 +78,8 @@ unixsocket = [
 ]
 
 common = [
-  "pytest<7.0",
-  "pytest-cov>=2.0,<3.0",
+  "pytest",
+  "pytest-cov",
   "mock<2.0",
   "moto>=1.3.10,<5.0",
   "HTTPretty==0.8.10",

diff --git a/test/worker_test.py b/test/worker_test.py
@@ -1154,7 +1154,7 @@ def test_stop_getting_new_work_assistant_keep_alive(self):
         self._test_stop_getting_new_work(keep_alive=True, assistant=True)
 
 
-class DynamicDependenciesTest(unittest.TestCase):
+class DynamicDependenciesTest(LuigiTestCase):
     n_workers = 1
     timeout = float('inf')
 
@@ -1606,7 +1606,7 @@ def complete(self):
         return False
 
 
-class MultipleWorkersTest(unittest.TestCase):
+class MultipleWorkersTest(LuigiTestCase):
 
     @unittest.skip('Always skip. There are many intermittent failures')
     def test_multiple_workers(self):