mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2026-04-28 11:40:32 +00:00
70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
"""Tests for task_run status write-through sync."""
|
|
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
from skyvern.forge.sdk.db.agent_db import AgentDB
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_session():
|
|
session = AsyncMock()
|
|
session.__aenter__ = AsyncMock(return_value=session)
|
|
session.__aexit__ = AsyncMock(return_value=False)
|
|
return session
|
|
|
|
|
|
@pytest.fixture
|
|
def agent_db(mock_session):
|
|
db = AgentDB.__new__(AgentDB)
|
|
db.Session = MagicMock(return_value=mock_session)
|
|
# Set up tasks repository (sync_task_run_status delegates to self.tasks)
|
|
from skyvern.forge.sdk.db.repositories.tasks import TasksRepository
|
|
|
|
tasks = TasksRepository.__new__(TasksRepository)
|
|
tasks.Session = MagicMock(return_value=mock_session)
|
|
tasks.debug_enabled = False
|
|
tasks._is_retryable_error_fn = None
|
|
db.tasks = tasks
|
|
return db
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sync_task_run_status_updates_matching_row(agent_db, mock_session):
|
|
"""sync_task_run_status should UPDATE task_runs where run_id matches."""
|
|
await agent_db.sync_task_run_status(
|
|
organization_id="org_1",
|
|
run_id="wr_123",
|
|
status="failed",
|
|
)
|
|
mock_session.execute.assert_called_once()
|
|
call_args = mock_session.execute.call_args
|
|
# The SQL should be an UPDATE on task_runs
|
|
sql_text = str(call_args[0][0])
|
|
assert "task_runs" in sql_text
|
|
assert "status" in sql_text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sync_task_run_status_no_raise_on_error(agent_db, mock_session):
|
|
"""sync_task_run_status should swallow exceptions (best-effort)."""
|
|
mock_session.execute.side_effect = Exception("DB error")
|
|
# Should NOT raise
|
|
await agent_db.sync_task_run_status(
|
|
organization_id="org_1",
|
|
run_id="nonexistent",
|
|
status="failed",
|
|
)
|
|
|
|
|
|
def test_terminal_statuses_match_run_status():
|
|
"""Guard: TERMINAL_STATUSES and RunStatus.is_final() must agree.
|
|
|
|
If this fails, a new terminal status was added to one but not the other.
|
|
Update TERMINAL_STATUSES in skyvern/schemas/runs.py (the single source of truth).
|
|
"""
|
|
from skyvern.forge.sdk.schemas.runs import TERMINAL_STATUSES
|
|
from skyvern.schemas.runs import RunStatus
|
|
|
|
assert set(TERMINAL_STATUSES) == {s.value for s in RunStatus if s.is_final()}
|