LangBot/tests/integration/api/test_bots.py

"""
API integration tests for bot endpoints.

Tests real HTTP API behavior for bot management.

Run: uv run pytest tests/integration/api/test_bots.py -q
"""

from __future__ import annotations

import pytest
from unittest.mock import MagicMock, AsyncMock, Mock
from types import SimpleNamespace

from tests.factories import FakeApp


pytestmark = pytest.mark.integration


@pytest.fixture(scope='module')
def mock_circular_import_chain():
    """Break circular import chain for API controller."""
    from tests.utils.import_isolation import isolated_sys_modules, MockLifecycleControlScope

    class FakeMinimalApplication:
        pass

    mock_app = MagicMock()
    mock_app.Application = FakeMinimalApplication

    mock_entities = MagicMock()
    mock_entities.LifecycleControlScope = MockLifecycleControlScope

    clear = [
        'langbot.pkg.api.http.controller.group',
        'langbot.pkg.api.http.controller.groups',
        'langbot.pkg.api.http.controller.groups.platform',
        'langbot.pkg.api.http.controller.groups.platform.bots',
        'langbot.pkg.api.http.controller.groups.platform.adapters',
        'langbot.pkg.api.http.controller.main',
    ]

    with isolated_sys_modules(
        mocks={
            'langbot.pkg.core.app': mock_app,
            'langbot.pkg.core.entities': mock_entities,
        },
        clear=clear,
    ):
        import langbot.pkg.api.http.controller.groups.platform.bots as _bots  # noqa: E402, F401

        yield


@pytest.fixture(scope='module')
def fake_bot_app():
    """Create FakeApp with bot services (module scope for reuse)."""
    app = FakeApp()

    app.instance_config.data.update(
        {
            'api': {'port': 5300},
            'system': {'allow_modify_login_info': True, 'limitation': {}},
        }
    )

    # Auth services
    account = SimpleNamespace(uuid='account-test', user='test@example.com')
    app.user_service = Mock()
    app.user_service.is_initialized = AsyncMock(return_value=True)
    app.user_service.verify_jwt_token = AsyncMock(return_value='test@example.com')
    app.user_service.get_user_by_email = AsyncMock(return_value=account)
    app.user_service.get_authenticated_account = AsyncMock(return_value=account)
    app.workspace_collaboration_service = SimpleNamespace(
        resolve_account_workspace=AsyncMock(
            return_value=SimpleNamespace(
                workspace=SimpleNamespace(uuid='workspace-test'),
                membership=SimpleNamespace(
                    uuid='membership-test',
                    role='owner',
                    projection_revision=0,
                ),
                execution=SimpleNamespace(instance_uuid='instance-test', placement_generation=1),
            )
        )
    )
    app.apikey_service = Mock()
    app.apikey_service.verify_api_key = AsyncMock(return_value=True)
    app.apikey_service.authenticate_api_key = AsyncMock(
        return_value=SimpleNamespace(
            instance_uuid='instance-test',
            placement_generation=1,
            api_key_uuid='api-key-test',
            workspace_uuid='workspace-test',
            permissions=frozenset(
                {
                    'resource.view',
                    'resource.manage',
                    'runtime.operate',
                    'provider_secret.manage',
                }
            ),
        )
    )

    # Bot service
    app.bot_service = Mock()
    app.bot_service.get_bots = AsyncMock(
        return_value=[
            {
                'uuid': 'test-bot-uuid',
                'name': 'Test Bot',
                'platform': 'telegram',
                'pipeline_uuid': 'test-pipeline-uuid',
            }
        ]
    )
    app.bot_service.get_runtime_bot_info = AsyncMock(
        return_value={
            'uuid': 'test-bot-uuid',
            'name': 'Test Bot',
            'platform': 'telegram',
            'pipeline_uuid': 'test-pipeline-uuid',
            'webhook_url': 'https://example.com/webhook/test-bot-uuid',
        }
    )
    app.bot_service.create_bot = AsyncMock(return_value={'uuid': 'new-bot-uuid'})
    app.bot_service.update_bot = AsyncMock(return_value={})
    app.bot_service.delete_bot = AsyncMock()
    app.bot_service.list_event_logs = AsyncMock(return_value=([{'uuid': 'log-1', 'message': 'test log'}], 1))
    app.bot_service.send_message = AsyncMock()

    # Platform manager
    app.platform_mgr = Mock()

    return app


@pytest.fixture(scope='module')
async def quart_test_client(fake_bot_app, http_controller_cls):
    """Create Quart test client (module scope to avoid route re-registration)."""
    controller = http_controller_cls(fake_bot_app)
    await controller.initialize()

    client = controller.quart_app.test_client()
    yield client


@pytest.mark.usefixtures('mock_circular_import_chain')
class TestBotEndpoints:
    """Tests for /api/v1/platform/bots endpoints."""

    @pytest.mark.asyncio
    async def test_get_bots_success(self, quart_test_client):
        """GET /api/v1/platform/bots returns bot list."""
        response = await quart_test_client.get('/api/v1/platform/bots', headers={'Authorization': 'Bearer test_token'})

        assert response.status_code == 200
        data = await response.get_json()
        assert data['code'] == 0
        assert 'data' in data
        assert 'bots' in data['data']

    @pytest.mark.asyncio
    async def test_create_bot_success(self, quart_test_client):
        """POST /api/v1/platform/bots creates new bot."""
        response = await quart_test_client.post(
            '/api/v1/platform/bots',
            headers={'Authorization': 'Bearer test_token'},
            json={'name': 'New Bot', 'platform': 'telegram', 'pipeline_uuid': 'test-pipeline'},
        )

        assert response.status_code == 200
        data = await response.get_json()
        assert data['code'] == 0
        assert 'uuid' in data['data']

    @pytest.mark.asyncio
    async def test_get_single_bot_success(self, quart_test_client):
        """GET /api/v1/platform/bots/{uuid} returns bot with runtime info."""
        response = await quart_test_client.get(
            '/api/v1/platform/bots/test-bot-uuid', headers={'Authorization': 'Bearer test_token'}
        )

        assert response.status_code == 200
        data = await response.get_json()
        assert data['code'] == 0
        assert 'bot' in data['data']

    @pytest.mark.asyncio
    async def test_update_bot_success(self, quart_test_client):
        """PUT /api/v1/platform/bots/{uuid} updates bot."""
        response = await quart_test_client.put(
            '/api/v1/platform/bots/test-bot-uuid',
            headers={'Authorization': 'Bearer test_token'},
            json={'name': 'Updated Bot'},
        )

        assert response.status_code == 200
        data = await response.get_json()
        assert data['code'] == 0

    @pytest.mark.asyncio
    async def test_delete_bot_success(self, quart_test_client):
        """DELETE /api/v1/platform/bots/{uuid} deletes bot."""
        response = await quart_test_client.delete(
            '/api/v1/platform/bots/test-bot-uuid', headers={'Authorization': 'Bearer test_token'}
        )

        assert response.status_code == 200


@pytest.mark.usefixtures('mock_circular_import_chain')
class TestBotLogsEndpoint:
    """Tests for bot logs endpoint."""

    @pytest.mark.asyncio
    async def test_get_bot_logs_success(self, quart_test_client):
        """POST /api/v1/platform/bots/{uuid}/logs returns logs."""
        response = await quart_test_client.post(
            '/api/v1/platform/bots/test-bot-uuid/logs',
            headers={'Authorization': 'Bearer test_token'},
            json={'from_index': -1, 'max_count': 10},
        )

        assert response.status_code == 200
        data = await response.get_json()
        assert data['code'] == 0
        assert 'logs' in data['data']
        assert 'total_count' in data['data']

    @pytest.mark.asyncio
    async def test_viewer_can_read_ordinary_bot_logs(self, quart_test_client, fake_bot_app):
        access = fake_bot_app.workspace_collaboration_service.resolve_account_workspace.return_value
        original_role = access.membership.role
        access.membership.role = 'viewer'
        fake_bot_app.bot_service.list_event_logs.reset_mock()
        try:
            response = await quart_test_client.post(
                '/api/v1/platform/bots/test-bot-uuid/logs',
                headers={'Authorization': 'Bearer test_token'},
                json={'from_index': -1, 'max_count': 10},
            )
        finally:
            access.membership.role = original_role

        assert response.status_code == 200
        assert (await response.get_json())['code'] == 0
        fake_bot_app.bot_service.list_event_logs.assert_awaited_once()


@pytest.mark.usefixtures('mock_circular_import_chain')
class TestBotSendMessageEndpoint:
    """Tests for bot send message endpoint."""

    @pytest.mark.asyncio
    async def test_send_message_success(self, quart_test_client):
        """POST /api/v1/platform/bots/{uuid}/send_message sends message."""
        response = await quart_test_client.post(
            '/api/v1/platform/bots/test-bot-uuid/send_message',
            headers={'Authorization': 'Bearer test_api_key'},
            json={
                'target_type': 'person',
                'target_id': 'user123',
                'message_chain': [{'type': 'text', 'text': 'Hello'}],
            },
        )

        assert response.status_code == 200
        data = await response.get_json()
        assert data['code'] == 0
        assert data['data']['sent'] is True

    @pytest.mark.asyncio
    async def test_send_message_missing_target_type(self, quart_test_client):
        """POST send_message without target_type returns 400."""
        response = await quart_test_client.post(
            '/api/v1/platform/bots/test-bot-uuid/send_message',
            headers={'Authorization': 'Bearer test_api_key'},
            json={'target_id': 'user123', 'message_chain': [{'type': 'text', 'text': 'Hello'}]},
        )

        assert response.status_code == 400
        data = await response.get_json()
        assert data['code'] == -1

    @pytest.mark.asyncio
    async def test_send_message_invalid_target_type(self, quart_test_client):
        """POST send_message with invalid target_type returns 400."""
        response = await quart_test_client.post(
            '/api/v1/platform/bots/test-bot-uuid/send_message',
            headers={'Authorization': 'Bearer test_api_key'},
            json={
                'target_type': 'invalid',
                'target_id': 'user123',
                'message_chain': [{'type': 'text', 'text': 'Hello'}],
            },
        )

        assert response.status_code == 400
        data = await response.get_json()
        assert data['code'] == -1