Skip to content

Commit 293996d

Browse files
authored
Update test_manager.py
1 parent 06aa2e8 commit 293996d

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

‎tests/test_manager.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,10 @@
77

88
import pytest
99

10-
from app.controller.manager import Controller
11-
from app.controller.runner import ExecResult, Runner, SandboxNotFoundError
12-
from app.db import get_session_factory
10+
from app.controllers.manager import Controller
11+
from app.controllers.runner import ExecResult, Runner, SandboxNotFoundError
1312
from app.models import Conversation, User, new_id
13+
from app.models.db import get_session_factory
1414

1515

1616
@dataclass
@@ -280,8 +280,8 @@ def test_phantom_finalize_when_row_missing() -> None:
280280
"""If the request thread never commits the Run row (crash), the
281281
worker still persists the outcome as a phantom row rather than
282282
losing the terminal state."""
283-
from app.controller import manager as mgr
284-
from app.controller.protocol import RunOutcome
283+
from app.controllers import manager as mgr
284+
from app.controllers.protocol import RunOutcome
285285

286286
class NullRunner(Runner):
287287
def create(self, user_id, conversation_id):
@@ -306,8 +306,8 @@ def reap_idle(self, ttl_seconds):
306306
controller = mgr.Controller(runner=NullRunner())
307307
# do NOT create the Run row; directly invoke _finish_run
308308
controller._finish_run("run_phantom", RunOutcome(errors=["orphaned"], exit_code=1))
309-
from app.db import get_session_factory
310309
from app.models import Run
310+
from app.models.db import get_session_factory
311311

312312
db = get_session_factory()()
313313
try:

0 commit comments

Comments
 (0)