import pandas as pd

from gchr.gchr import GCHR
from gchr.gchr_bookings import GchrBookings
from gchr.gchr_translate import TRANSLATE

base_dir_1: str = "C:\\Projekte\\GCHR2_Testdaten\\Kunden\\Altermann"
base_dir_2: str = "C:\\Projekte\\GCHR2_Testdaten\\Kunden\\Koenig-und-Partner"


def test_single_booking_files():
    gchr = GCHR(base_dir_1)
    assert isinstance(gchr.bookings, GchrBookings)
    assert len(gchr.bookings.account_bookings) == 1
    assert gchr.bookings.account_bookings[0].name == "GuV_Bilanz_Salden.csv"


def test_multiple_booking_files():
    gchr = GCHR(base_dir_2)
    assert isinstance(gchr.bookings, GchrBookings)
    assert len(gchr.bookings.account_bookings) == 2
    assert gchr.bookings.account_bookings[0].name == "GuV_Bilanz_Salden.csv"
    assert gchr.bookings.account_bookings[1].name == "GuV_Bilanz_Salden_deop03.csv"


def test_translation_existing():
    gchr = GCHR(base_dir_1)
    df = gchr.df_translate
    assert isinstance(df, pd.DataFrame)
    assert df.shape[1] == 12
    assert list(df.columns) == TRANSLATE
    assert df.shape[0] > 0, "Translation not empty"


def test_all_periods():
    periods = GCHR.get_all_periods("2024-12-23")
    assert len(periods) == 12
    assert periods[0] == ("2023", "12")
    assert periods[-1] == ("2024", "11")