import os
from datetime import datetime

import pandas as pd
from pytest import fixture
from socceraction.data.opta import OptaCompetitionSchema, OptaGameSchema
from socceraction.data.opta.parsers import F1JSONParser


@fixture()
def f1json_parser() -> F1JSONParser:
    path = os.path.join(
        os.path.dirname(__file__),
        os.pardir,
        os.pardir,
        os.pardir,
        "datasets",
        "opta",
        "tournament-2017-8.json",
    )
    return F1JSONParser(str(path))


def test_extract_competitions(f1json_parser: F1JSONParser) -> None:
    competitions = f1json_parser.extract_competitions()
    assert len(competitions) == 1
    assert competitions[(8, 2017)] == {
        "competition_id": 8,
        "season_id": 2017,
        "competition_name": "English Premier League",
        "season_name": "2017",
    }
    OptaCompetitionSchema.validate(pd.DataFrame.from_dict(competitions, orient="index"))


def test_extract_games(f1json_parser: F1JSONParser) -> None:
    games = f1json_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, 19, 45),
        "home_team_id": 3,
        "away_team_id": 13,
        "home_score": 4,
        "away_score": 3,
    }
    OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index"))