CiteGeist/tests/test_app_server.py

from types import SimpleNamespace

from citegeist import BibliographyStore
from citegeist.app_api import LiteratureExplorerApi
from citegeist.bootstrap import BootstrapResult
from citegeist.app_server import (
    LiteratureExplorerAppServer,
    _extract_bearer_token,
    _request_is_authorized,
    create_request_handler,
)


class FakeBootstrapper:
    def __init__(self) -> None:
        self.calls: list[dict] = []

    def bootstrap(self, store, **kwargs):
        self.calls.append(dict(kwargs))
        return [
            BootstrapResult(
                citation_key="graph2026topic",
                origin="topic",
                created=True,
                score=4.0,
                title="Graph Topic Result",
                year="2026",
            )
        ]


def test_literature_explorer_app_server_dispatch_search():
    store = BibliographyStore()
    try:
        store.ingest_bibtex(
            """
@article{seed2024,
  author = {Seed, Alice},
  title = {Graph Topic Result},
  year = {2024}
}
"""
        )
        server = LiteratureExplorerAppServer(LiteratureExplorerApi(store))

        payload = server.dispatch("search", {"query": "graph", "limit": 5})

        assert payload["results"][0]["citation_key"] == "seed2024"
    finally:
        store.close()


def test_literature_explorer_app_server_dispatch_exports_topic_bibtex():
    store = BibliographyStore()
    try:
        store.ingest_bibtex(
            """
@article{seed2024,
  author = {Seed, Alice},
  title = {Graph Topic Result},
  year = {2024}
}
"""
        )
        store.add_entry_topic("seed2024", topic_slug="graph-topic", topic_name="Graph Topic", source_label="seed")
        server = LiteratureExplorerAppServer(LiteratureExplorerApi(store))

        payload = server.dispatch("export_topic_bibtex", {"topic_slug": "graph-topic"})

        assert payload["entry_count"] == 1
        assert "@article{seed2024," in payload["bibtex"]
    finally:
        store.close()


def test_literature_explorer_app_server_dispatch_expand_topic_with_new_controls():
    store = BibliographyStore()
    try:
        store.ingest_bibtex(
            """
@article{seed2024,
  author = {Seed, Alice},
  title = {Graph Topic Result},
  year = {2024}
}
"""
        )
        store.add_entry_topic("seed2024", topic_slug="graph-topic", topic_name="Graph Topic", source_label="seed")
        server = LiteratureExplorerAppServer(LiteratureExplorerApi(store))

        payload = server.dispatch(
            "expand_topic",
            {
                "topic_slug": "graph-topic",
                "relation_type": "both",
                "max_rounds": 3,
                "recent_years": 5,
                "target_recent_entries": 10,
                "preview_only": True,
            },
        )

        assert payload["preview"] is True
        assert "results" in payload
    finally:
        store.close()


def test_literature_explorer_app_server_dispatch_bootstrap_with_new_caps():
    store = BibliographyStore()
    try:
        bootstrapper = FakeBootstrapper()
        server = LiteratureExplorerAppServer(LiteratureExplorerApi(store, bootstrapper=bootstrapper))

        payload = server.dispatch(
            "bootstrap",
            {
                "topic": "graph topic",
                "topic_slug": "graph-topic",
                "topic_name": "Graph Topic",
                "preview_only": True,
                "expand": False,
                "expansion_mode": "cites",
                "expansion_rounds": 2,
                "recent_years": 5,
                "target_recent_entries": 4,
                "max_expanded_entries": 75,
                "max_expand_seconds": 12.5,
            },
        )

        assert payload["preview"] is True
        assert "results" in payload
        assert bootstrapper.calls[0]["expansion_mode"] == "cites"
        assert bootstrapper.calls[0]["max_expanded_entries"] == 75
        assert bootstrapper.calls[0]["max_expand_seconds"] == 12.5
    finally:
        store.close()


def test_literature_explorer_http_handler_class_can_be_created():
    store = BibliographyStore()
    try:
        store.ingest_bibtex(
            """
@article{seed2024,
  author = {Seed, Alice},
  title = {Graph Topic Result},
  year = {2024}
}
"""
        )
        app_server = LiteratureExplorerAppServer(LiteratureExplorerApi(store))
        handler = create_request_handler(app_server)

        assert handler is not None
        assert issubclass(handler, object)
    finally:
        store.close()


def test_request_authorization_accepts_bearer_and_header_token():
    headers = SimpleNamespace(
        get=lambda key, default="": {
            "Authorization": "Bearer secret-token",
            "X-API-Token": "secret-token",
        }.get(key, default)
    )

    assert _extract_bearer_token(headers) == "secret-token"
    assert _request_is_authorized(headers, "secret-token") is True


def test_request_authorization_rejects_missing_or_wrong_token():
    missing_headers = SimpleNamespace(get=lambda key, default="": default)
    wrong_headers = SimpleNamespace(
        get=lambda key, default="": {
            "Authorization": "Bearer wrong-token",
            "X-API-Token": "",
        }.get(key, default)
    )

    assert _request_is_authorized(missing_headers, "secret-token") is False
    assert _request_is_authorized(wrong_headers, "secret-token") is False
    assert _request_is_authorized(missing_headers, None) is True