Begin writing tests for worker

src/blueapi/worker/multithread.py

@@ -45,4 +45,4 @@ def _run_worker_thread(worker: Worker[T]) -> None:
 
     LOGGER.info("Worker starting")
     # TODO: Use API methods only!
-    worker._run_forever()
+    worker.run()

src/blueapi/worker/reworker.py

@@ -83,7 +83,7 @@ def stop(self) -> None:
         self._task_queue.put(KillSignal())
         self._stopped.wait(timeout=30.0)
 
-    def _run_forever(self) -> None:
+    def run(self) -> None:
         LOGGER.info("Worker starting")
         self._ctx.run_engine.state_hook = self._on_state_change
         self._ctx.run_engine.subscribe(self._on_document)

src/blueapi/worker/worker.py

@@ -37,15 +37,15 @@ def start(self) -> None:
         """
 
     @abstractmethod
-    def stop(self) -> None:
+    def run(self) -> None:
         """
-        Stop worker
+        Run worker, blocks
         """
 
     @abstractmethod
-    def block(self) -> None:
+    def stop(self) -> None:
         """
-        Block while worker is running
+        Stop worker
         """
 
     @property

tests/worker/test_reworker.py

@@ -0,0 +1,43 @@
+from concurrent.futures import Future
+from threading import Event
+
+import pytest
+
+from blueapi.core import BlueskyContext
+from blueapi.worker import RunEngineWorker, RunPlan, Worker, WorkerEvent, WorkerState
+
+
+@pytest.fixture
+def context() -> BlueskyContext:
+    ctx = BlueskyContext()
+    ctx.with_startup_script("blueapi.startup.example")
+    return ctx
+
+
+@pytest.fixture
+def worker(context: BlueskyContext) -> Worker:
+    worker = RunEngineWorker(context)
+    yield worker
+    worker.stop()
+
+
+def test_stop(worker: Worker) -> None:
+    worker.start()
+
+
+def test_submit(worker: Worker) -> None:
+    worker.start()
+    started = Future()
+
+    def process_event(event: WorkerEvent, task_id: str) -> None:
+        started.set_result(event)
+
+    worker.worker_events.subscribe(process_event)
+    worker.submit_task(
+        "test",
+        RunPlan(
+            name="sleep",
+            params={"time": 0.1},
+        ),
+    )
+    assert started.result(timeout=5.0).status.state is WorkerState.RUNNING