from datetime import datetime, timedelta
import re
import shutil

from review.stats import Board, UserBoard, DurationBoard, ReviewData, ReviewStats


def setup_function():
    shutil.copyfile("tests/datas/reviewstats_1.csv", "tests/reviews/reviewstats_1.csv")
    shutil.copyfile("tests/datas/reviewstats_2.csv", "tests/reviews/reviewstats_2.csv")


def load_reviewstats(filepath):
    final_datas = []
    try:
        with open(filepath) as file_handle:
            lines = file_handle.read().splitlines()
            for line in lines:
                if line.strip() != "":
                    datas = re.split(r"\s+", line)
                    if len(datas) == 2:
                        final_datas.append(
                            ReviewData(
                                datetime.strptime(datas[0], "%Y%m%d-%Hh%M"),
                                int(datas[1]),
                                None,
                            )
                        )
                    else:
                        final_datas.append(
                            ReviewData(
                                datetime.strptime(datas[0], "%Y%m%d-%Hh%M"),
                                int(datas[1]),
                                int(datas[2]),
                            )
                        )
    except FileNotFoundError:
        # no file, no stats
        pass

    return final_datas


def test_board():
    board = Board()
    assert board.datas == {}
    assert board["test"] == 0
    assert board.position("test") is None
    assert board.min == 0
    assert board.max == 0
    assert board.avg == 0
    assert board.sum == 0


def test_userboard():
    review = ReviewStats("tests/reviews/reviewstats_1.csv")
    review.load()
    board = review.users_board
    assert board["test"] == 0
    assert board[1] == 5
    assert board.min == 1
    assert board.max == 10
    assert board.sum == 51
    assert board.avg == 3.1875


def test_durationboard():
    review = ReviewStats("tests/reviews/reviewstats_1.csv")
    review.load()
    board = review.durations_board
    assert board["test"] == 0
    assert board.min == 0
    assert board.max == 0
    assert board.sum == 0
    assert board.avg == 0


def test_durationboard():
    review = ReviewStats("tests/reviews/reviewstats_2.csv")
    review.load()
    board = review.durations_board
    assert board["test"] == 0
    assert board.min == 3
    assert board.max == 17
    assert board.sum == 49
    assert board.avg == 7.0