import main


def test_create_embed_all_types(monkeypatch):
    monkeypatch.setattr(main, "load_timezones", lambda: [
        {"zone_name": "Etc/UTC", "country_code": "ZZ"}
    ])
    monkeypatch.setattr(main, "load_countries", lambda: [
        {"country_code": "ZZ", "country_name": "Nowhere"}
    ])

    emb = main.create_embed("reminder")
    assert emb["title"] == "Reminder"
    assert "5 minute" in emb["description"]
    assert emb["color"] == 0xE67E22

    emb = main.create_embed("reminder_halftime")
    assert emb["title"] == "Reminder halftime"
    assert "Half-time in 5 minutes" in emb["description"]

    monkeypatch.setattr(main, "where_is_it_420", lambda tzs, cs, **kwargs: [])
    emb = main.create_embed("halftime")
    assert emb["title"] == "Halftime"
    assert emb["image"] is not None

    monkeypatch.setattr(main, "where_is_it_420", lambda tzs, cs, **kwargs: [])
    emb = main.create_embed("420")
    assert emb["title"] == "420"
    assert emb["image"] is not None

    emb = main.create_embed("nope")
    assert emb["description"] == "Unknown notification type"