import unittest

from gcstruct.gchr import GCHR


class TestGchr(unittest.TestCase):
    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(self):
        gchr = GCHR(self.base_dir_1)
        self.assertEqual(len(gchr.account_bookings), 1)
        self.assertEqual(gchr.account_bookings[0].name, "GuV_Bilanz_Salden.csv")

    def test_multiple_booking_files(self):
        gchr = GCHR(self.base_dir_2)
        self.assertEqual(len(gchr.account_bookings), 2)
        self.assertEqual(gchr.account_bookings[0].name, "GuV_Bilanz_Salden.csv")
        self.assertEqual(gchr.account_bookings[1].name, "GuV_Bilanz_Salden_deop03.csv")