import io
import time
from unittest import mock

import pytest

import main


SAMPLE_TIMEZONE_CSV = """Etc/UTC,ZZ,UTC,0,0,0
America/New_York,US,EST,0,-18000,0
Europe/London,GB,BST,0,0,1
"""

SAMPLE_COUNTRY_CSV = """ZZ,Unknown
US,United States
GB,United Kingdom
"""


def test_load_timezones_and_countries(monkeypatch):
    tzs = main.load_timezones()
    countries = main.load_countries()
    assert any(t['zone_name'] == 'America/New_York' for t in tzs)
    assert any(c['country_code'] == 'US' for c in countries)


def test_get_tz_and_country_info():
    timezones = [{'zone_name': 'A/B', 'country_code': 'US'}]
    countries = [{'country_code': 'US', 'country_name': 'United States'}]
    assert main.get_tz_info('A/B', timezones)['zone_name'] == 'A/B'
    assert main.get_country_info('US', countries)[
        'country_name'] == 'United States'
    assert main.get_tz_info('X/Y', timezones) is None
    assert main.get_country_info('XX', countries) is None


def test_create_embed_all_types(monkeypatch):
    # Prevent create_embed from trying to read actual CSV files by patching loaders
    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'}])

    # reminder
    emb = main.create_embed('reminder')
    assert emb['title'] == 'Reminder'
    assert '5 minute' in emb['description']
    assert emb['color'] == 0xe67e22

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

    # halftime (should include image)
    monkeypatch.setattr(main, 'where_is_it_420', lambda tzs, cs: [])
    emb = main.create_embed('halftime')
    assert emb['title'] == 'Halftime'
    assert emb['image'] is not None

    # 420 (should include image and appended tz info string when list empty)
    monkeypatch.setattr(main, 'where_is_it_420', lambda tzs, cs: [])
    emb = main.create_embed('420')
    assert emb['title'] == '420'
    assert emb['image'] is not None
    assert "not 4:20" in emb['description'] or "It's 4:20" in emb['description']

    # unknown
    emb = main.create_embed('nope')
    assert emb['description'] == 'Unknown notification type'


def test_where_is_it_420(monkeypatch):
    # Limit timezones to a predictable set
    monkeypatch.setattr(main.pytz, 'all_timezones', ['Etc/UTC'])

    tzs = [{'zone_name': 'Etc/UTC', 'country_code': 'ZZ'}]
    countries = [{'country_code': 'ZZ', 'country_name': 'Nowhere'}]

    class FakeDatetime:
        @staticmethod
        def now(tz):
            class R:
                hour = 4
            return R()

    monkeypatch.setattr(main, 'datetime', FakeDatetime)
    monkeypatch.setattr(main, 'get_tz_info', lambda name,
                        t: tzs[0] if name == 'Etc/UTC' else None)
    monkeypatch.setattr(main, 'get_country_info', lambda code,
                        c: countries[0] if code == 'ZZ' else None)

    res = main.where_is_it_420(tzs, countries)
    assert res == ['Nowhere']


def test_main_exits_quickly(monkeypatch):
    # Patch send_notification so it doesn't perform network
    monkeypatch.setattr(main, 'send_notification', lambda x: None)
    # Make schedule.run_pending raise KeyboardInterrupt to exit loop
    monkeypatch.setattr(main.schedule, 'run_pending', lambda: (
        _ for _ in ()).throw(KeyboardInterrupt()))
    # Patch time.sleep to no-op
    monkeypatch.setattr(main.time, 'sleep', lambda s: None)
    # Ensure WEBHOOK_URL present to avoid early return
    monkeypatch.setenv('DISCORD_WEBHOOK_URL', 'http://example.com/webhook')
    main.WEBHOOK_URL = 'http://example.com/webhook'

    # Should exit quickly due to KeyboardInterrupt from run_pending
    main.main()