States tests passing

PrefectHQ · Jun 26, 2020 · 44081f7 · 44081f7
1 parent c47081f
commit 44081f7
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 321 deletions.
diff --git a/src/prefect_server/_api/states.py b/src/prefect_server/_api/states.py
@@ -68,7 +68,7 @@ async def set_flow_run_state(
     )
 
     if not flow_run:
-        raise ValueError(f"State update failed for flow run {flow_run_id}")
+        raise ValueError(f"State update failed for flow run ID {flow_run_id}")
 
     # --------------------------------------------------------
     # insert the new state in the database
@@ -185,7 +185,7 @@ async def set_task_run_state(
     )
 
     if not task_run:
-        raise ValueError(f"State update failed for task run {task_run_id}")
+        raise ValueError(f"State update failed for task run ID {task_run_id}")
 
     # ------------------------------------------------------
     # if the state is running, ensure the flow run is also running

diff --git a/tests/graphql/test_server.py b/tests/graphql/test_server.py