Skip to content

Commit ead6582

Browse files
committed
test: fix dev task relay usage audit mock setup
1 parent c3b563d commit ead6582

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

server/tests/test_gateway/test_dev_task_relay.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,9 @@ def _session_factory(session: AsyncMock | None = None):
1616
if session is None:
1717
session = AsyncMock()
1818
session.add = MagicMock()
19+
execute_result = MagicMock()
20+
execute_result.scalar_one_or_none.return_value = None
21+
session.execute = AsyncMock(return_value=execute_result)
1922

2023
@asynccontextmanager
2124
async def factory():

0 commit comments

Comments
 (0)