BioMCP

from biomcp.articles.autocomplete import Entity, EntityRequest, autocomplete


async def test_autocomplete(anyio_backend, http_cache):
    # new cache for each call
    assert http_cache.count == 0

    # gene (compare using entity_id directly)
    request = EntityRequest(concept="gene", query="her2")
    entity = await autocomplete(request=request)
    assert entity.entity_id == "@GENE_ERBB2"

    # variant
    request = EntityRequest(concept="variant", query="BRAF V600E")
    assert await autocomplete(request=request) == Entity(
        _id="@VARIANT_p.V600E_BRAF_human",
        biotype="variant",
        name="p.V600E",
    )

    # disease
    request = EntityRequest(concept="disease", query="lung adenocarcinoma")
    assert await autocomplete(request=request) == Entity(
        _id="@DISEASE_Adenocarcinoma_of_Lung",
        biotype="disease",
        name="Adenocarcinoma of Lung",
        match="Multiple matches",
    )

    assert http_cache.count == 3

    # duplicate request uses the cached response
    request = EntityRequest(concept="gene", query="her2")
    entity = await autocomplete(request=request)
    assert entity.entity_id == "@GENE_ERBB2"
    assert http_cache.count == 3