mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2026-04-28 11:40:32 +00:00
AgentDB Phase 7: Migrate remaining 8 domains to typed repos (#5366)
This commit is contained in:
parent
58fed69496
commit
26b8f4d73e
84 changed files with 811 additions and 8804 deletions
|
|
@ -13,7 +13,7 @@ async def test_initialize_task_v2_populates_task_run_url_when_user_url_is_known(
|
|||
organization = SimpleNamespace(organization_id="org_123")
|
||||
user_url = "https://example.com"
|
||||
|
||||
app.DATABASE.create_task_v2.return_value = SimpleNamespace(
|
||||
app.DATABASE.observer.create_task_v2.return_value = SimpleNamespace(
|
||||
observer_cruise_id="tsk_123",
|
||||
workflow_run_id=None,
|
||||
url=user_url,
|
||||
|
|
@ -24,14 +24,14 @@ async def test_initialize_task_v2_populates_task_run_url_when_user_url_is_known(
|
|||
title=DEFAULT_WORKFLOW_TITLE,
|
||||
)
|
||||
app.WORKFLOW_SERVICE.setup_workflow_run.return_value = SimpleNamespace(workflow_run_id="wr_123")
|
||||
app.DATABASE.update_task_v2.return_value = SimpleNamespace(
|
||||
app.DATABASE.observer.update_task_v2.return_value = SimpleNamespace(
|
||||
observer_cruise_id="tsk_123",
|
||||
workflow_run_id="wr_123",
|
||||
workflow_id="wf_123",
|
||||
workflow_permanent_id="wpid_123",
|
||||
url=user_url,
|
||||
)
|
||||
app.DATABASE.create_task_run.return_value = SimpleNamespace(run_id="tsk_123")
|
||||
app.DATABASE.tasks.create_task_run.return_value = SimpleNamespace(run_id="tsk_123")
|
||||
|
||||
await initialize_task_v2(
|
||||
organization=organization,
|
||||
|
|
@ -40,7 +40,7 @@ async def test_initialize_task_v2_populates_task_run_url_when_user_url_is_known(
|
|||
create_task_run=True,
|
||||
)
|
||||
|
||||
app.DATABASE.create_task_run.assert_awaited_once_with(
|
||||
app.DATABASE.tasks.create_task_run.assert_awaited_once_with(
|
||||
task_run_type=RunType.task_v2,
|
||||
organization_id="org_123",
|
||||
run_id="tsk_123",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue