Support updating worklist item status via relay connection

Author: steventux

src/relay_listener.py

@@ -21,6 +21,7 @@
 
 from environment import Environment
 from services.mwl.create_worklist_item import CreateWorklistItem
+from services.mwl.update_worklist_item_status import UpdateWorklistItemStatus
 from services.storage import MWLStorage
 from telemetry import configure_telemetry
 
@@ -93,6 +94,8 @@ def process_action(self, payload: dict):
             return {"status": "echo", "payload": payload}
         elif action_name == "worklist.create_item":
             return CreateWorklistItem(self.storage).call(payload)
+        elif action_name == "worklist.update_item_status":
+            return UpdateWorklistItemStatus(self.storage).call(payload)
         else:
             raise ValueError(f"Unsupported action: {action_name}")
 

tests/integration/test_relay_listener_processes_actions.py

@@ -2,11 +2,20 @@
 
 import pytest
 
+from models import WorklistItem
 from relay_listener import RelayListener
 from services.storage import MWLStorage
 
 
 class TestRelayListenerProcessesActions:
+    @pytest.fixture
+    def update_payload(self):
+        return {
+            "action_id": "action-12345",
+            "action_type": "worklist.update_item_status",
+            "parameters": {"worklist_item": {"accession_number": "ACC999999", "status": "in progress"}},
+        }
+
     @pytest.mark.asyncio
     async def test_relay_listener_creates_worklist_items(self, listener_payload, tmp_dir, fake_relay):
         storage = MWLStorage(f"{tmp_dir}/test_worklist.db")
@@ -22,3 +31,34 @@ async def test_relay_listener_creates_worklist_items(self, listener_payload, tmp
         item = stored_items[0]
         assert item.accession_number == "ACC999999"
         assert item.patient_id == "999123456"
+
+    @pytest.mark.asyncio
+    async def test_relay_listener_updates_worklist_item_status(self, update_payload, tmp_dir, fake_relay):
+        storage = MWLStorage(f"{tmp_dir}/test_worklist.db")
+        listener = RelayListener(storage)
+        relay_message = json.dumps({"accept": {"address": "wss://accept-url"}})
+
+        storage.store_worklist_item(
+            WorklistItem(**{
+                "accession_number": "ACC999999",
+                "patient_id": "999123456",
+                "patient_name": "Test^Patient",
+                "patient_birth_date": "19900101",
+                "patient_sex": "F",
+                "scheduled_date": "20240101",
+                "scheduled_time": "090000",
+                "modality": "MG",
+                "study_description": "Mammogram",
+                "source_message_id": "action-12345",
+            })
+        )
+
+        with fake_relay(relay_message, json.dumps(update_payload)) as ws_client:
+            await listener.listen()
+
+        ws_client.send.assert_called_once_with(json.dumps({"accession_number": "ACC999999", "status": "IN PROGRESS"}))
+        stored_items = storage.find_worklist_items()
+        assert len(stored_items) == 1
+        item = stored_items[0]
+        assert item.accession_number == "ACC999999"
+        assert item.patient_id == "999123456"

tests/test_relay_listener.py

@@ -72,7 +72,7 @@ async def test_relay_listener_listen(self, storage_instance, listener_payload, f
             )
         )
 
-    def test_process_action(self, storage_instance, listener_payload):
+    def test_process_create_item_action(self, storage_instance, listener_payload):
         subject = RelayListener(storage_instance)
 
         response = subject.process_action(listener_payload)
@@ -93,6 +93,33 @@ def test_process_action(self, storage_instance, listener_payload):
             )
         )
 
+    def test_process_update_item_status_action(self, storage_instance, listener_payload):
+        subject = RelayListener(storage_instance)
+
+        subject.process_action(listener_payload)
+
+        update_payload = {
+            "action_id": "action-12345",
+            "action_type": "worklist.update_item_status",
+            "parameters": {"worklist_item": {"accession_number": "ACC999999", "status": "IN PROGRESS"}},
+        }
+
+        response = subject.process_action(update_payload)
+        assert response == {"accession_number": "ACC999999", "status": "IN PROGRESS"}
+
+        storage_instance.update_status.assert_called_once_with("ACC999999", "IN PROGRESS")
+
+    def test_process_action_missing_keys(self, storage_instance, listener_payload):
+        subject = RelayListener(storage_instance)
+
+        del listener_payload["parameters"]["worklist_item"]["accession_number"]
+
+        response = subject.process_action(listener_payload)
+        assert response["status"] == "error"
+        assert "Missing key" in response["message"]
+
+        storage_instance.store_worklist_item.assert_not_called()
+
     def test_process_action_invalid_type(self, storage_instance, listener_payload):
         subject = RelayListener(storage_instance)