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/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)