from unfccc_ghg_data.unfccc_crf_reader.unfccc_crf_reader_core import (
    filter_category,
    get_info_from_crf_filename,
    get_latest_date_for_country,
)


def test_get_latest_date_for_country():
    # RUS CRF
    expected = "22082023"
    date = get_latest_date_for_country("RUS", 2023, submission_type="CRF")
    assert date == expected

    # AUS CRT
    expected = "12042024"
    date = get_latest_date_for_country("AUS", 1, submission_type="CRT")
    assert date == expected

    # RUS CRT
    expected = "20241108"
    date = get_latest_date_for_country("RUS", 1, submission_type="CRT")
    assert date == expected


def test_get_info_from_crf_filename():
    # crf
    filename = "BLR_2021_1990_30032021_192048.xlsx"
    expected = {
        "party": "BLR",
        "submission_year": 2021,
        "data_year": 1990,
        "date": "30032021",
        "extra": "192048",
    }
    assert expected == get_info_from_crf_filename(filename)

    # crt
    filename = "GUY-CRT-2024-V0.3-1992-20240927-191031_started.xlsx"
    expected = {
        "party": "GUY",
        "submission_year": 2024,
        "data_year": 1992,
        "date": "20240927",
        "extra": "191031_started",
        "version": "V0.3",
    }
    assert expected == get_info_from_crf_filename(filename)


def test_filter_category():
    # general
    map_gen = ["Option C (country-specific):", ["\\IGNORE"], 4]
    assert filter_category(map_gen, "MOZ") == map_gen

    # country specific
    expected = [
        "Other (as specified in table 3(I).A)",
        ["3.A.1.C"],
        5,
    ]
    expected_remove = ["\\REMOVE", ["3.A.1.C"], 5]
    # exclude multiple
    map_excl_multiple = [
        "\\C!-AUS-MLT-LUX-POL-SVN-USA\\ Other (as specified in table 3(I).A)",
        ["3.A.1.C"],
        5,
    ]
    assert filter_category(map_excl_multiple, "MOZ") == expected
    assert filter_category(map_excl_multiple, "MLT") == expected_remove

    # exclude single
    map_excl_single = [
        "\\C!-AUS\\ Other (as specified in table 3(I).A)",
        ["3.A.1.C"],
        5,
    ]
    expected = [
        "Other (as specified in table 3(I).A)",
        ["3.A.1.C"],
        5,
    ]
    expected_remove = ["\\REMOVE", ["3.A.1.C"], 5]
    assert filter_category(map_excl_single, "MOZ") == expected
    assert filter_category(map_excl_single, "AUS") == expected_remove

    # include multiple
    map_incl_multiple = [
        "\\C-AUS-NLD\\ Other (as specified in table 3(I).A)",
        ["3.A.1.C"],
        5,
    ]
    assert filter_category(map_incl_multiple, "MOZ") == expected_remove
    assert filter_category(map_incl_multiple, "AUS") == expected
    assert filter_category(map_incl_multiple, "NLD") == expected

    # include single
    map_incl_single = ["\\C-AUS\\ Other (as specified in table 3(I).A)", ["3.A.1.C"], 5]
    assert filter_category(map_incl_single, "MOZ") == expected_remove
    assert filter_category(map_incl_single, "AUS") == expected