import os
from datetime import datetime

import pandas as pd
from pytest import fixture
from socceraction.data.opta import (
    OptaGameSchema,
    OptaPlayerSchema,
    OptaTeamSchema,
)
from socceraction.data.opta.parsers import F9JSONParser


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


def test_extract_games(f9json_parser: F9JSONParser) -> None:
    games = f9json_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),
        "home_team_id": 3,
        "away_team_id": 13,
        "home_score": 4,
        "away_score": 3,
        "attendance": 59387,
        "duration": 96,
        "referee": "Mike Dean",
        "venue": None,
        "home_manager": None,
        "away_manager": None,
    }
    OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index"))


def test_extract_teams(f9json_parser: F9JSONParser) -> None:
    teams = f9json_parser.extract_teams()
    assert len(teams) == 2
    assert teams[3] == {
        "team_id": 3,
        "team_name": "Arsenal",
    }
    assert teams[13] == {
        "team_id": 13,
        "team_name": "Leicester City",
    }
    OptaTeamSchema.validate(pd.DataFrame.from_dict(teams, orient="index"))


def test_extract_players(f9json_parser: F9JSONParser) -> None:
    players = f9json_parser.extract_players()
    assert len(players) == 27
    assert players[(918893, 11334)] == {
        "game_id": 918893,
        "player_id": 11334,
        "player_name": "Petr Cech",
        "team_id": 3,
        "jersey_number": 33,
        "minutes_played": 96,
        "starting_position": "Goalkeeper",
        "is_starter": True,
    }
    OptaPlayerSchema.validate(pd.DataFrame.from_dict(players, orient="index"))