import os
from datetime import datetime

import pandas as pd
from pytest import fixture
from socceraction.data.opta import OptaEventSchema, OptaGameSchema
from socceraction.data.opta.parsers import F24JSONParser


@fixture()
def f24json_parser() -> F24JSONParser:
    path = os.path.join(
        os.path.dirname(__file__),
        os.pardir,
        os.pardir,
        os.pardir,
        "datasets",
        "opta",
        "match-2017-8-918893.json",
    )
    return F24JSONParser(str(path))


def test_extract_games(f24json_parser: F24JSONParser) -> None:
    games = f24json_parser.extract_games()
    assert len(games) == 1
    assert games[918893] == {
        "game_id": 918893,
        "season_id": 2017,
        "competition_id": 8,
        "game_day": 1,
        "game_date": datetime(2017, 8, 11, 18, 45, 0, 0),
        "home_team_id": 3,
        "away_team_id": 13,
    }
    OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index"))


def test_extract_events(f24json_parser: F24JSONParser) -> None:
    events = f24json_parser.extract_events()
    assert len(events) == 1785
    assert events[(918893, 1815408644)] == {
        "game_id": 918893,
        "event_id": 1815408644,
        "period_id": 2,
        "team_id": 3,
        "player_id": 41792,
        "type_id": 5,
        "timestamp": datetime(2017, 8, 11, 20, 38, 49, 0),
        "minute": 94,
        "second": 57,
        "outcome": False,
        "start_x": 101.1,
        "start_y": 44.4,
        "end_x": 101.1,
        "end_y": 44.4,
        "qualifiers": {
            233: "690",
            56: "Center",
        },
        "assist": False,
        "keypass": False,
    }
    df = pd.DataFrame.from_dict(events, orient="index")
    df["type_name"] = "Added later"
    OptaEventSchema.validate(df)