mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-08-09 04:40:57 +00:00
Junyan Qin
95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from langbot.pkg.api.http.context import ExecutionContext
|
|
from langbot.pkg.platform.sources.http_bot import HttpBotAdapter
|
|
from langbot.pkg.platform.sources import http_bot as http_bot_module
|
|
|
|
|
|
def _session(key):
|
|
session = SimpleNamespace()
|
|
session._langbot_session_key = key
|
|
return session
|
|
|
|
|
|
def _adapter(app, execution_context) -> HttpBotAdapter:
|
|
adapter = HttpBotAdapter.model_construct(
|
|
config={'signature_required': False},
|
|
logger=SimpleNamespace(execution_context=execution_context),
|
|
bot_uuid='bot-a',
|
|
outbound_states={},
|
|
idempotency_cache={},
|
|
sync_waiters={},
|
|
inbound_tasks=set(),
|
|
)
|
|
object.__setattr__(adapter, 'ap', app)
|
|
return adapter
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_http_bot_reset_removes_only_exact_execution_scope():
|
|
context = ExecutionContext(
|
|
instance_uuid='instance-a',
|
|
workspace_uuid='workspace-a',
|
|
placement_generation=3,
|
|
bot_uuid='bot-a',
|
|
)
|
|
target_key = ('instance-a', 'workspace-a', 3, 'bot-a', 'person', 'shared-session')
|
|
retained_keys = [
|
|
('instance-b', 'workspace-a', 3, 'bot-a', 'person', 'shared-session'),
|
|
('instance-a', 'workspace-b', 3, 'bot-a', 'person', 'shared-session'),
|
|
('instance-a', 'workspace-a', 4, 'bot-a', 'person', 'shared-session'),
|
|
('instance-a', 'workspace-a', 3, 'bot-b', 'person', 'shared-session'),
|
|
('instance-a', 'workspace-a', 3, 'bot-a', 'group', 'shared-session'),
|
|
('instance-a', 'workspace-a', 3, 'bot-a', 'person', 'other-session'),
|
|
]
|
|
sessions = [_session(target_key), *[_session(key) for key in retained_keys], SimpleNamespace()]
|
|
app = SimpleNamespace(sess_mgr=SimpleNamespace(session_list=sessions))
|
|
adapter = _adapter(app, context)
|
|
|
|
removed = await adapter._reset_session('person', 'shared-session')
|
|
|
|
assert removed is True
|
|
assert [getattr(session, '_langbot_session_key', None) for session in app.sess_mgr.session_list] == [
|
|
*retained_keys,
|
|
None,
|
|
]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_http_bot_reset_fails_closed_without_trusted_scope():
|
|
app = SimpleNamespace(sess_mgr=SimpleNamespace(session_list=[]))
|
|
adapter = _adapter(app, None)
|
|
|
|
with pytest.raises(RuntimeError, match='trusted execution scope'):
|
|
await adapter._reset_session('person', 'shared-session')
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_http_bot_bounds_inbound_listener_tasks(monkeypatch):
|
|
monkeypatch.setattr(http_bot_module, '_INBOUND_TASK_MAX', 1)
|
|
adapter = _adapter(SimpleNamespace(), None)
|
|
started = asyncio.Event()
|
|
release = asyncio.Event()
|
|
|
|
async def blocking_listener():
|
|
started.set()
|
|
await release.wait()
|
|
|
|
first = adapter._start_inbound_task(blocking_listener())
|
|
await started.wait()
|
|
rejected = adapter._start_inbound_task(blocking_listener())
|
|
|
|
assert first is not None
|
|
assert rejected is None
|
|
assert len(adapter.inbound_tasks) == 1
|
|
|
|
release.set()
|
|
await first
|
|
await asyncio.sleep(0)
|
|
assert adapter.inbound_tasks == set()
|