import os
from datetime import datetime

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


@fixture()
def f7xml_parser() -> F7XMLParser:
    path = os.path.join(
        os.path.dirname(__file__),
        os.pardir,
        os.pardir,
        os.pardir,
        "datasets",
        "opta",
        "f7-23-2018-1009316-matchresults.xml",
    )
    return F7XMLParser(str(path))


def test_extract_competitions(f7xml_parser: F7XMLParser) -> None:
    competitions = f7xml_parser.extract_competitions()
    assert len(competitions) == 1
    assert competitions[(23, 2018)] == {
        "competition_id": 23,
        "season_id": 2018,
        "competition_name": "Spanish La Liga",
        "season_name": "Season 2018/2019",
    }
    OptaCompetitionSchema.validate(pd.DataFrame.from_dict(competitions, orient="index"))


def test_extract_games(f7xml_parser: F7XMLParser) -> None:
    games = f7xml_parser.extract_games()
    assert len(games) == 1
    assert games[1009316] == {
        "game_id": 1009316,
        "season_id": 2018,
        "competition_id": 23,
        "game_day": 1,
        "game_date": datetime(2018, 8, 20, 21, 0),
        "home_team_id": 174,
        "away_team_id": 957,
        "home_score": 2,
        "away_score": 1,
        "duration": 96,
        "referee": "Adrián Cordero Vega",
        "venue": "San Mamés",
        "attendance": 38575,
        "home_manager": "Eduardo Berizzo",
        "away_manager": "Mauricio Pellegrino",
    }
    OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index"))


def test_extract_teams(f7xml_parser: F7XMLParser) -> None:
    teams = f7xml_parser.extract_teams()
    assert len(teams) == 2
    assert teams[957] == {
        "team_id": 957,
        "team_name": "Leganés",
    }
    assert teams[174] == {
        "team_id": 174,
        "team_name": "Athletic Club",
    }
    OptaTeamSchema.validate(pd.DataFrame.from_dict(teams, orient="index"))


def test_extract_players(f7xml_parser: F7XMLParser) -> None:
    players = f7xml_parser.extract_players()
    assert len(players) == 36
    assert players[(1009316, 242831)] == {
        "game_id": 1009316,
        "team_id": 174,
        "player_id": 242831,
        "player_name": "Peru Nolaskoain",
        "is_starter": True,
        "minutes_played": 96,
        "jersey_number": 31,
        "starting_position": "Defender",
    }
    OptaPlayerSchema.validate(pd.DataFrame.from_dict(players, orient="index"))