Automatically mark all WorkerState tests

crusaderky · crusaderky · commit 97332b408e83 · 2022-07-09T23:44:44.000+01:00
diff --git a/conftest.py b/conftest.py
@@ -35,5 +35,8 @@ def pytest_collection_modifyitems(config, items):
         if "slow" in item.keywords:
             item.add_marker(skip_slow)
 
+        if "ws" in item.fixturenames:
+            item.add_marker(pytest.mark.workerstate)
+
 
 pytest_plugins = ["distributed.pytest_resourceleaks"]
diff --git a/setup.cfg b/setup.cfg
@@ -96,6 +96,7 @@ markers =
     ipython: marks tests as exercising IPython
     gpu: marks tests we want to run on GPUs
     leaking: ignore leaked resources; see pytest_resourceleaks.py for usage
+    workerstate: deterministic test for the worker state machine. Automatically applied to all tests that use the 'ws' fixture.
 
 # pytest-timeout settings
 # 'thread' kills off the whole test suite. 'signal' only kills the offending test.
