[NO-TICKET] Repository Design Pattern refactor (#9056) follow-up — missing parameter bugs (#5291)

tests/unit/test_browser_session_update_started_at.py

@@ -0,0 +1,58 @@
+"""Test that BrowserSessionsRepository.update_persistent_browser_session() accepts started_at."""
+
+from datetime import datetime, timezone
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from skyvern.forge.sdk.db.repositories.browser_sessions import BrowserSessionsRepository
+from tests.unit.conftest import MockAsyncSessionCtx, make_mock_session
+
+
+def _make_browser_repo(mock_pbs: MagicMock) -> BrowserSessionsRepository:
+    mock_session = make_mock_session(mock_pbs)
+    return BrowserSessionsRepository(session_factory=lambda: MockAsyncSessionCtx(mock_session))
+
+
+@pytest.mark.asyncio
+async def test_update_persistent_browser_session_accepts_started_at() -> None:
+    """update_persistent_browser_session() must accept started_at without raising TypeError."""
+    mock_pbs = MagicMock()
+    mock_pbs.status = "running"
+    mock_pbs.started_at = None
+    repo = _make_browser_repo(mock_pbs)
+
+    now = datetime.now(timezone.utc)
+    with patch(
+        "skyvern.forge.sdk.schemas.persistent_browser_sessions.PersistentBrowserSession.model_validate",
+        return_value=MagicMock(),
+    ):
+        await repo.update_persistent_browser_session(
+            "pbs_123",
+            organization_id="org_123",
+            started_at=now,
+        )
+
+    assert mock_pbs.started_at == now
+
+
+@pytest.mark.asyncio
+async def test_update_persistent_browser_session_without_started_at() -> None:
+    """When started_at is not passed, the field should not be touched."""
+    mock_pbs = MagicMock()
+    mock_pbs.status = "created"
+    original_started_at = datetime(2026, 1, 1, tzinfo=timezone.utc)
+    mock_pbs.started_at = original_started_at
+    repo = _make_browser_repo(mock_pbs)
+
+    with patch(
+        "skyvern.forge.sdk.schemas.persistent_browser_sessions.PersistentBrowserSession.model_validate",
+        return_value=MagicMock(),
+    ):
+        await repo.update_persistent_browser_session(
+            "pbs_123",
+            organization_id="org_123",
+            status="running",
+        )
+
+    assert mock_pbs.started_at == original_started_at