from gfa_core.mas.utils.intent_parser import parse_ncbi_fetch_request


def test_ncbi_fetch_accepts_ids_suffix_and_protein_keywords():
    message = (
        "我的拟南芥spl基因家族的id列表放在"
        "/home/xia-lab/wutx/plantAnalysisAgent/ath_family/At.SPL.ids,"
        "请你帮我下载蛋白序列"
    )

    params = parse_ncbi_fetch_request(message)

    assert params["ncbiGeneIds"] == "/home/xia-lab/wutx/plantAnalysisAgent/ath_family/At.SPL.ids"
    assert params["ncbiSpecies"] == "Arabidopsis thaliana"
    assert params["ncbiSeqTypes"] == "protein"
    assert params["prefix"] == "At.SPL"


def test_ncbi_fetch_accepts_uploaded_filename_without_absolute_path():
    message = "我已经上传了 At.SPL.ids，请用这个id列表下载拟南芥SPL蛋白序列"

    params = parse_ncbi_fetch_request(message)

    assert params["ncbiGeneIds"] == "At.SPL.ids"
    assert params["ncbiSpecies"] == "Arabidopsis thaliana"
    assert params["ncbiSeqTypes"] == "protein"
    assert params["prefix"] == "At.SPL"


def test_ncbi_fetch_accepts_according_to_filename_without_spaces():
    message = "帮我根据At.SPL.ids下载蛋白序列"

    params = parse_ncbi_fetch_request(message)

    assert params["ncbiGeneIds"] == "At.SPL.ids"
    assert params["ncbiSeqTypes"] == "protein"
    assert params["prefix"] == "At.SPL"


def test_ncbi_fetch_does_not_truncate_filename_when_adjacent_to_chinese():
    params = parse_ncbi_fetch_request("请根据gene_ids.txt下载cds")

    assert params["ncbiGeneIds"] == "gene_ids.txt"
    assert params["ncbiSeqTypes"] == "cds"
    assert params["prefix"] == "gene_ids"


def test_ncbi_fetch_does_not_treat_downloaded_result_view_as_new_download():
    params = parse_ncbi_fetch_request("帮我把下载好的At.SPL家族序列的基因id列展示一下")

    assert params == {}