"""Report theme mapping for markdown grouping."""

from applepy.reporters.report_themes import theme_for_category, themes_present_in_order


def test_theme_for_known_categories() -> None:
    assert theme_for_category("Core") == "Core platform and controls"
    assert theme_for_category("Compliance") == "Compliance and configuration baselines"
    assert theme_for_category("MDM: Jamf") == "MDM local posture"


def test_theme_for_unknown_goes_to_other() -> None:
    assert theme_for_category("CustomVendor") == "Other findings"


def test_themes_present_in_order_sorts() -> None:
    got = themes_present_in_order({"Other findings", "Core platform and controls"})
    assert got[0] == "Core platform and controls"
    assert got[1] == "Other findings"