mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-08-09 04:40:57 +00:00
huanghuoguoguo
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
import types
|
|
from importlib import import_module
|
|
from types import SimpleNamespace
|
|
from unittest.mock import ANY, AsyncMock
|
|
|
|
import pytest
|
|
import quart
|
|
|
|
core_app_module = types.ModuleType('langbot.pkg.core.app')
|
|
core_app_module.Application = object
|
|
sys.modules.setdefault('langbot.pkg.core.app', core_app_module)
|
|
|
|
|
|
pytestmark = pytest.mark.asyncio
|
|
|
|
|
|
async def _create_test_client(agent_service: SimpleNamespace):
|
|
app = quart.Quart(__name__)
|
|
account = SimpleNamespace(
|
|
uuid='account-test',
|
|
user='test@example.com',
|
|
)
|
|
user_service = SimpleNamespace(
|
|
get_authenticated_account=AsyncMock(return_value=account),
|
|
)
|
|
access = SimpleNamespace(
|
|
workspace=SimpleNamespace(uuid='workspace-test'),
|
|
membership=SimpleNamespace(
|
|
uuid='membership-test',
|
|
role='developer',
|
|
projection_revision=1,
|
|
),
|
|
execution=SimpleNamespace(
|
|
instance_uuid='instance-test',
|
|
placement_generation=1,
|
|
),
|
|
)
|
|
ap = SimpleNamespace(
|
|
agent_service=agent_service,
|
|
user_service=user_service,
|
|
apikey_service=SimpleNamespace(
|
|
authenticate_api_key=AsyncMock(return_value=None)
|
|
),
|
|
workspace_collaboration_service=SimpleNamespace(
|
|
resolve_account_workspace=AsyncMock(return_value=access)
|
|
),
|
|
)
|
|
AgentsRouterGroup = import_module('langbot.pkg.api.http.controller.groups.agents').AgentsRouterGroup
|
|
group = AgentsRouterGroup(ap, app)
|
|
await group.initialize()
|
|
return app.test_client()
|
|
|
|
|
|
async def test_create_agent_returns_bad_request_for_invalid_runner_config():
|
|
message = 'agent config runner_config must be an object'
|
|
agent_service = SimpleNamespace(create_agent=AsyncMock(side_effect=ValueError(message)))
|
|
client = await _create_test_client(agent_service)
|
|
|
|
response = await client.post(
|
|
'/api/v1/agents',
|
|
json={'name': 'Invalid Agent', 'config': {'runner_config': []}},
|
|
headers={'Authorization': 'Bearer test-token'},
|
|
)
|
|
|
|
assert response.status_code == 400
|
|
assert await response.get_json() == {'code': -1, 'msg': message}
|
|
agent_service.create_agent.assert_awaited_once_with(
|
|
ANY,
|
|
{'name': 'Invalid Agent', 'config': {'runner_config': []}},
|
|
)
|
|
|
|
|
|
async def test_update_agent_returns_bad_request_for_invalid_runner_config():
|
|
message = 'agent config runner.id must be a string'
|
|
agent_service = SimpleNamespace(update_agent=AsyncMock(side_effect=ValueError(message)))
|
|
client = await _create_test_client(agent_service)
|
|
|
|
response = await client.put(
|
|
'/api/v1/agents/agent-1',
|
|
json={'config': {'runner': {'id': 7}}},
|
|
headers={'Authorization': 'Bearer test-token'},
|
|
)
|
|
|
|
assert response.status_code == 400
|
|
assert await response.get_json() == {'code': -1, 'msg': message}
|
|
agent_service.update_agent.assert_awaited_once_with(
|
|
ANY,
|
|
'agent-1',
|
|
{'config': {'runner': {'id': 7}}},
|
|
)
|