ref #9: [feat] add tests
This commit is contained in:
Azamat
parent
3a7973accd
commit
fb974fff1e
5 changed files with 899 additions and 2 deletions
284
test/test_sandbox_usecase.py
Normal file
284
test/test_sandbox_usecase.py
Normal file
|
|
@ -0,0 +1,284 @@
|
|||
from datetime import UTC, datetime, timedelta
|
||||
|
||||
from domain.sandbox import SandboxSession, SandboxStatus
|
||||
from repository.sandbox_session import InMemorySandboxSessionRepository
|
||||
from usecase.sandbox import CleanupExpiredSandboxes, CreateSandbox, CreateSandboxCommand
|
||||
|
||||
|
||||
class FakeClock:
|
||||
def __init__(self, now: datetime) -> None:
|
||||
self._now = now
|
||||
|
||||
def now(self) -> datetime:
|
||||
return self._now
|
||||
|
||||
|
||||
class FakeLogger:
|
||||
def __init__(self) -> None:
|
||||
self.messages: list[
|
||||
tuple[str, str, dict[str, str | int | float | bool] | None]
|
||||
] = []
|
||||
|
||||
def debug(self, message: str, attrs=None) -> None:
|
||||
self.messages.append(('debug', message, attrs))
|
||||
|
||||
def info(self, message: str, attrs=None) -> None:
|
||||
self.messages.append(('info', message, attrs))
|
||||
|
||||
def warning(self, message: str, attrs=None) -> None:
|
||||
self.messages.append(('warning', message, attrs))
|
||||
|
||||
def error(self, message: str, attrs=None) -> None:
|
||||
self.messages.append(('error', message, attrs))
|
||||
|
||||
|
||||
class FakeRuntime:
|
||||
def __init__(self) -> None:
|
||||
self.create_calls: list[dict[str, object]] = []
|
||||
self.stop_calls: list[str] = []
|
||||
|
||||
def create(
|
||||
self,
|
||||
*,
|
||||
session_id: str,
|
||||
chat_id: str,
|
||||
created_at: datetime,
|
||||
expires_at: datetime,
|
||||
) -> SandboxSession:
|
||||
self.create_calls.append(
|
||||
{
|
||||
'session_id': session_id,
|
||||
'chat_id': chat_id,
|
||||
'created_at': created_at,
|
||||
'expires_at': expires_at,
|
||||
}
|
||||
)
|
||||
return SandboxSession(
|
||||
session_id=session_id,
|
||||
chat_id=chat_id,
|
||||
container_id=f'container-{session_id}',
|
||||
status=SandboxStatus.RUNNING,
|
||||
created_at=created_at,
|
||||
expires_at=expires_at,
|
||||
)
|
||||
|
||||
def stop(self, container_id: str) -> None:
|
||||
self.stop_calls.append(container_id)
|
||||
|
||||
|
||||
def test_create_sandbox_reuses_active_session_when_not_expired() -> None:
|
||||
now = datetime(2026, 4, 2, 12, 0, tzinfo=UTC)
|
||||
session = SandboxSession(
|
||||
session_id='session-1',
|
||||
chat_id='chat-1',
|
||||
container_id='container-1',
|
||||
status=SandboxStatus.RUNNING,
|
||||
created_at=now - timedelta(minutes=1),
|
||||
expires_at=now + timedelta(minutes=4),
|
||||
)
|
||||
repository = InMemorySandboxSessionRepository()
|
||||
repository.save(session)
|
||||
runtime = FakeRuntime()
|
||||
logger = FakeLogger()
|
||||
usecase = CreateSandbox(
|
||||
repository=repository,
|
||||
runtime=runtime,
|
||||
clock=FakeClock(now),
|
||||
logger=logger,
|
||||
ttl=timedelta(minutes=5),
|
||||
)
|
||||
|
||||
result = usecase.execute(CreateSandboxCommand(chat_id='chat-1'))
|
||||
|
||||
assert result == session
|
||||
assert runtime.create_calls == []
|
||||
assert runtime.stop_calls == []
|
||||
assert repository.get_active_by_chat_id('chat-1') == session
|
||||
assert logger.messages == [
|
||||
(
|
||||
'info',
|
||||
'sandbox_reused',
|
||||
{
|
||||
'chat_id': 'chat-1',
|
||||
'session_id': 'session-1',
|
||||
'container_id': 'container-1',
|
||||
},
|
||||
)
|
||||
]
|
||||
|
||||
|
||||
def test_create_sandbox_replaces_expired_session_and_creates_new_one(
|
||||
monkeypatch,
|
||||
) -> None:
|
||||
now = datetime(2026, 4, 2, 12, 0, tzinfo=UTC)
|
||||
expired_session = SandboxSession(
|
||||
session_id='session-old',
|
||||
chat_id='chat-1',
|
||||
container_id='container-old',
|
||||
status=SandboxStatus.RUNNING,
|
||||
created_at=now - timedelta(minutes=10),
|
||||
expires_at=now,
|
||||
)
|
||||
repository = InMemorySandboxSessionRepository()
|
||||
repository.save(expired_session)
|
||||
runtime = FakeRuntime()
|
||||
logger = FakeLogger()
|
||||
usecase = CreateSandbox(
|
||||
repository=repository,
|
||||
runtime=runtime,
|
||||
clock=FakeClock(now),
|
||||
logger=logger,
|
||||
ttl=timedelta(minutes=5),
|
||||
)
|
||||
monkeypatch.setattr('usecase.sandbox._new_session_id', lambda: 'session-new')
|
||||
|
||||
result = usecase.execute(CreateSandboxCommand(chat_id='chat-1'))
|
||||
|
||||
assert runtime.stop_calls == ['container-old']
|
||||
assert runtime.create_calls == [
|
||||
{
|
||||
'session_id': 'session-new',
|
||||
'chat_id': 'chat-1',
|
||||
'created_at': now,
|
||||
'expires_at': now + timedelta(minutes=5),
|
||||
}
|
||||
]
|
||||
assert result == SandboxSession(
|
||||
session_id='session-new',
|
||||
chat_id='chat-1',
|
||||
container_id='container-session-new',
|
||||
status=SandboxStatus.RUNNING,
|
||||
created_at=now,
|
||||
expires_at=now + timedelta(minutes=5),
|
||||
)
|
||||
assert repository.get_active_by_chat_id('chat-1') == result
|
||||
assert logger.messages == [
|
||||
(
|
||||
'info',
|
||||
'sandbox_replaced',
|
||||
{
|
||||
'chat_id': 'chat-1',
|
||||
'session_id': 'session-old',
|
||||
'container_id': 'container-old',
|
||||
},
|
||||
),
|
||||
(
|
||||
'info',
|
||||
'sandbox_created',
|
||||
{
|
||||
'chat_id': 'chat-1',
|
||||
'session_id': 'session-new',
|
||||
'container_id': 'container-session-new',
|
||||
},
|
||||
),
|
||||
]
|
||||
|
||||
|
||||
def test_create_sandbox_creates_new_session_when_none_exists() -> None:
|
||||
now = datetime(2026, 4, 2, 12, 0, tzinfo=UTC)
|
||||
repository = InMemorySandboxSessionRepository()
|
||||
runtime = FakeRuntime()
|
||||
logger = FakeLogger()
|
||||
usecase = CreateSandbox(
|
||||
repository=repository,
|
||||
runtime=runtime,
|
||||
clock=FakeClock(now),
|
||||
logger=logger,
|
||||
ttl=timedelta(minutes=5),
|
||||
)
|
||||
|
||||
result = usecase.execute(CreateSandboxCommand(chat_id='chat-1'))
|
||||
|
||||
assert result.chat_id == 'chat-1'
|
||||
assert result.container_id == f'container-{result.session_id}'
|
||||
assert result.status is SandboxStatus.RUNNING
|
||||
assert result.created_at == now
|
||||
assert result.expires_at == now + timedelta(minutes=5)
|
||||
assert len(runtime.create_calls) == 1
|
||||
assert runtime.create_calls[0] == {
|
||||
'session_id': result.session_id,
|
||||
'chat_id': 'chat-1',
|
||||
'created_at': now,
|
||||
'expires_at': now + timedelta(minutes=5),
|
||||
}
|
||||
assert runtime.stop_calls == []
|
||||
assert repository.get_active_by_chat_id('chat-1') == result
|
||||
assert logger.messages == [
|
||||
(
|
||||
'info',
|
||||
'sandbox_created',
|
||||
{
|
||||
'chat_id': 'chat-1',
|
||||
'session_id': result.session_id,
|
||||
'container_id': result.container_id,
|
||||
},
|
||||
)
|
||||
]
|
||||
|
||||
|
||||
def test_cleanup_expired_sandboxes_stops_and_deletes_only_expired_sessions() -> None:
|
||||
now = datetime(2026, 4, 2, 12, 0, tzinfo=UTC)
|
||||
expired_session = SandboxSession(
|
||||
session_id='session-expired',
|
||||
chat_id='chat-expired',
|
||||
container_id='container-expired',
|
||||
status=SandboxStatus.RUNNING,
|
||||
created_at=now - timedelta(minutes=10),
|
||||
expires_at=now - timedelta(seconds=1),
|
||||
)
|
||||
boundary_session = SandboxSession(
|
||||
session_id='session-boundary',
|
||||
chat_id='chat-boundary',
|
||||
container_id='container-boundary',
|
||||
status=SandboxStatus.RUNNING,
|
||||
created_at=now - timedelta(minutes=5),
|
||||
expires_at=now,
|
||||
)
|
||||
active_session = SandboxSession(
|
||||
session_id='session-active',
|
||||
chat_id='chat-active',
|
||||
container_id='container-active',
|
||||
status=SandboxStatus.RUNNING,
|
||||
created_at=now - timedelta(minutes=1),
|
||||
expires_at=now + timedelta(minutes=5),
|
||||
)
|
||||
repository = InMemorySandboxSessionRepository()
|
||||
repository.save(expired_session)
|
||||
repository.save(boundary_session)
|
||||
repository.save(active_session)
|
||||
runtime = FakeRuntime()
|
||||
logger = FakeLogger()
|
||||
usecase = CleanupExpiredSandboxes(
|
||||
repository=repository,
|
||||
runtime=runtime,
|
||||
clock=FakeClock(now),
|
||||
logger=logger,
|
||||
)
|
||||
|
||||
result = usecase.execute()
|
||||
|
||||
assert result == [expired_session, boundary_session]
|
||||
assert runtime.stop_calls == ['container-expired', 'container-boundary']
|
||||
assert repository.get_active_by_chat_id('chat-expired') is None
|
||||
assert repository.get_active_by_chat_id('chat-boundary') is None
|
||||
assert repository.get_active_by_chat_id('chat-active') == active_session
|
||||
assert logger.messages == [
|
||||
(
|
||||
'info',
|
||||
'sandbox_cleaned',
|
||||
{
|
||||
'chat_id': 'chat-expired',
|
||||
'session_id': 'session-expired',
|
||||
'container_id': 'container-expired',
|
||||
},
|
||||
),
|
||||
(
|
||||
'info',
|
||||
'sandbox_cleaned',
|
||||
{
|
||||
'chat_id': 'chat-boundary',
|
||||
'session_id': 'session-boundary',
|
||||
'container_id': 'container-boundary',
|
||||
},
|
||||
),
|
||||
]
|
||||
Loading…
Add table
Add a link
Reference in a new issue