from __future__ import annotations

import pytest
from sqlalchemy.orm import sessionmaker

from services.knowledge_fs import runtime
from services.knowledge_fs.product_remote import KnowledgeFSOperationUnavailableError
from tests.helpers.unit.config_override import apply_config_overrides
from tests.helpers.unit.knowledge_fs_core_fakes import Marker, RecordingCallable


def test_runtime_fails_closed_when_the_remote_endpoint_is_not_configured(monkeypatch: pytest.MonkeyPatch) -> None:
    apply_config_overrides(monkeypatch, KNOWLEDGE_FS_BASE_URL=None)

    with pytest.raises(KnowledgeFSOperationUnavailableError, match="not configured"):
        runtime.create_knowledge_fs_runtime(sessionmaker())


def test_process_runtime_is_built_once_for_the_application_session_factory(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    session_maker = sessionmaker()
    built_runtime = Marker("runtime")
    factory = RecordingCallable(return_value=built_runtime)
    monkeypatch.setattr(runtime, "create_knowledge_fs_runtime", factory)

    first = runtime.get_knowledge_fs_runtime(session_maker)
    second = runtime.get_knowledge_fs_runtime(session_maker)

    assert first is built_runtime
    assert second is built_runtime
    factory.assert_called_once_with(session_maker)


def test_runtime_wires_one_shared_authorization_and_remote_graph(monkeypatch: pytest.MonkeyPatch) -> None:
    apply_config_overrides(
        monkeypatch,
        KNOWLEDGE_FS_BASE_URL="https://knowledge-fs.test",
        KNOWLEDGE_FS_RETRIEVAL_TEST_TIMEOUT_SECONDS=75.0,
        KNOWLEDGE_FS_TIMEOUT_SECONDS=12.0,
        KNOWLEDGE_FS_PRODUCT_MAX_RESPONSE_BYTES=4096,
    )

    factory_names = (
        "DifyKnowledgeFSProductRBACPort",
        "HTTPKnowledgeFSProductRemoteClient",
        "KnowledgeFSAppAdmissionService",
        "KnowledgeFSAppBindingManagementService",
        "KnowledgeFSAppExecutionCapabilityService",
        "KnowledgeFSBatchCapabilityBroker",
        "KnowledgeFSCapabilityBroker",
        "KnowledgeFSControlPlaneService",
        "KnowledgeFSControlSpaceCommandService",
        "KnowledgeFSServiceApiAuthorizationService",
        "KnowledgeFSDataFacade",
        "KnowledgeFSProductApplicationService",
        "KnowledgeFSProductService",
        "KnowledgeFSRevocationCommandProducer",
        "KnowledgeFSWorkspaceCutoverService",
        "KnowledgeFSWorkspaceGreenfieldInitializer",
        "SQLKnowledgeFSAppCatalog",
        "SQLKnowledgeFSWorkspaceMemberPort",
        "SQLKnowledgeFSWorkspaceRuntimeGate",
        "SQLAlchemyKnowledgeFSCapabilityIssuanceAuditor",
        "create_configured_knowledge_fs_capability_issuer",
        "get_knowledge_fs_lifecycle_remote",
    )
    factories: dict[str, RecordingCallable] = {}
    for name in factory_names:
        factory = RecordingCallable(return_value=Marker(name), name=name)
        factories[name] = factory
        monkeypatch.setattr(runtime, name, factory)

    session_maker = sessionmaker()
    result = runtime.create_knowledge_fs_runtime(session_maker)

    assert result.application is factories["KnowledgeFSProductApplicationService"].return_value
    assert result.app_admission is factories["KnowledgeFSAppAdmissionService"].return_value
    assert result.app_bindings is factories["KnowledgeFSAppBindingManagementService"].return_value
    assert result.app_capabilities is factories["KnowledgeFSAppExecutionCapabilityService"].return_value
    assert result.broker is factories["KnowledgeFSCapabilityBroker"].return_value
    assert result.control_plane is factories["KnowledgeFSControlPlaneService"].return_value
    assert result.service_api_authorization is factories["KnowledgeFSServiceApiAuthorizationService"].return_value
    assert result.facade is factories["KnowledgeFSDataFacade"].return_value

    factories["HTTPKnowledgeFSProductRemoteClient"].assert_called_once_with(
        base_url="https://knowledge-fs.test",
        retrieval_test_timeout_seconds=75.0,
        timeout_seconds=12.0,
        max_response_bytes=4096,
    )
    factories["create_configured_knowledge_fs_capability_issuer"].assert_called_once_with(
        audit=factories["SQLAlchemyKnowledgeFSCapabilityIssuanceAuditor"].return_value
    )
    assert (
        factories["KnowledgeFSProductService"].last_call.kwargs["cutover_gate"]
        is factories["SQLKnowledgeFSWorkspaceRuntimeGate"].return_value
    )
    factories["KnowledgeFSWorkspaceCutoverService"].assert_called_once_with(
        session_maker,
        remote_factory=factories["get_knowledge_fs_lifecycle_remote"],
    )
    factories["KnowledgeFSWorkspaceGreenfieldInitializer"].assert_called_once_with(
        session_maker,
        cutover=factories["KnowledgeFSWorkspaceCutoverService"].return_value,
    )
    factories["SQLKnowledgeFSWorkspaceRuntimeGate"].assert_called_once_with(
        session_maker,
        initializer=factories["KnowledgeFSWorkspaceGreenfieldInitializer"].return_value,
    )
    assert (
        factories["KnowledgeFSProductApplicationService"].last_call.kwargs["rbac"]
        is factories["DifyKnowledgeFSProductRBACPort"].return_value
    )
    assert factories["KnowledgeFSAppExecutionCapabilityService"].last_call.kwargs["admission"] is result.app_admission