apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-06-25 08:48:13 +02:00
Code Issues Projects Releases Packages Wiki Activity

style: format test_text_chat_session_service.py with ruff

Browse source
This commit is contained in:
Varun Nuthalapati 2026-06-07 11:51:13 -07:00
parent 5b3ea3e018
commit 577a85d72d
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
api/tests/test_text_chat_session_service.py
View file

@ -1,4 +1,4 @@
from unittest.mock import AsyncMock, patch from unittest.mock import AsyncMock
import pytest import pytest
@ -81,7 +81,10 @@ async def test_reload_text_chat_session_uses_run_id_to_resolve_organization(
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_execute_pending_turn_surfaces_original_exception_message(monkeypatch): async def test_execute_pending_turn_surfaces_original_exception_message(monkeypatch):
session = WorkflowRunTextSessionModel(workflow_run_id=42) session = WorkflowRunTextSessionModel(workflow_run_id=42)
session.session_data = {"turns": [{"id": "turn-1", "status": "pending"}], "cursor_turn_id": "turn-1"} session.session_data = {
"turns": [{"id": "turn-1", "status": "pending"}],
"cursor_turn_id": "turn-1",
}
session.checkpoint = None session.checkpoint = None
monkeypatch.setattr( monkeypatch.setattr(
@ -95,7 +98,9 @@ async def test_execute_pending_turn_surfaces_original_exception_message(monkeypa
AsyncMock(), AsyncMock(),
) )
with pytest.raises(TextChatSessionExecutionError, match="Workflow has 2 start nodes"): with pytest.raises(
TextChatSessionExecutionError, match="Workflow has 2 start nodes"
):
await execute_pending_text_chat_turn( await execute_pending_text_chat_turn(
workflow_id=1, workflow_id=1,
run_id=42, run_id=42,