From 8ced33552e33abc1f407d899b8059dbbab1f801d Mon Sep 17 00:00:00 2001 From: Victor Giers Date: Mon, 15 Jun 2026 21:04:41 +0200 Subject: [PATCH] Use direct workflow ID when testing workflow runs --- backend/tests/test_workflow_api.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/backend/tests/test_workflow_api.py b/backend/tests/test_workflow_api.py index f3326ab..92501b0 100644 --- a/backend/tests/test_workflow_api.py +++ b/backend/tests/test_workflow_api.py @@ -46,6 +46,7 @@ class WorkflowApiTests(unittest.IsolatedAsyncioTestCase): db.commit() self.user_message_id = user.message_id self.assistant_message_id = assistant.message_id + self.direct_workflow_id = direct.id self.direct_revision_id = direct.current_revision_id db.close() @@ -78,17 +79,13 @@ class WorkflowApiTests(unittest.IsolatedAsyncioTestCase): db.close() async def test_run_resolves_previous_assistant_and_message_url_targets(self): - db = self.Session() - remember = db.query(WorkflowDefinition).filter_by(slug="remember-this").one() - remember_id = remember.id - db.close() fake_runtime = SimpleNamespace(start=lambda _run_id: None) request = WorkflowRunRequest( session_id="regenerate-chat", message="Save https://example.com/source for later.", model="test-model", selection_mode="explicit", - workflow_id=remember_id, + workflow_id=self.direct_workflow_id, ) with patch.object(api, "SessionLocal", self.Session), patch.object(api, "runtime", fake_runtime): response = await api.create_workflow_run(request)