Test worker exit signal

materialsproject · Jan 26, 2022 · 3fceea7 · 3fceea7
1 parent 6c4f5fd
commit 3fceea7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/maggma/cli/__init__.py b/src/maggma/cli/__init__.py
@@ -65,7 +65,7 @@
     help="Number of chunks to distribute to workers",
 )
 @click.option(
-    "-d",
+    "-w",
     "--num-workers",
     "num_workers",
     default=0,

diff --git a/tests/cli/test_distributed.py b/tests/cli/test_distributed.py
@@ -90,6 +90,11 @@ async def test_manager_give_out_chunks(manager_server, log_to_stdout):
         assert work["@module"] == "tests.cli.test_distributed"
         assert work["val"] == i
 
+    for i in range(0, 10):
+        await socket.send(b"Ready")
+        message = await socket.recv()
+        assert message == b'"EXIT"'
+
 
 @pytest.mark.asyncio
 async def test_worker():