From 3fceea7a7fe4d6e35d04e80095ce8e6191add0fd Mon Sep 17 00:00:00 2001
From: Jason Munro <jmunro@lbl.gov>
Date: Wed, 26 Jan 2022 13:12:07 -0800
Subject: [PATCH] Test worker exit signal

---
 src/maggma/cli/__init__.py    | 2 +-
 tests/cli/test_distributed.py | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/maggma/cli/__init__.py b/src/maggma/cli/__init__.py
index e8fb9bf59..d12c5afcd 100644
--- 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
index 0506aa1d0..53c3ddd1b 100644
--- 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():