import unittest

from misc.csv_cleanup import cleanup_line


class TestCsvCleanup(unittest.TestCase):

    def test_cleanup_line_ident(self):
        simple_int = "1;2;3\n"
        simple_float = "1,20;2,322;3,434\n"
        simple_string = '"Test";"Default";"Simple Test"\n'

        self.assertEqual(cleanup_line(simple_int), simple_int)
        self.assertEqual(cleanup_line(simple_float), simple_float)
        self.assertEqual(cleanup_line(simple_string), simple_string)

    def test_cleanup_line_semicolon(self):
        string_1_in = '"String 1";"String;2";"Test 3"\n'
        string_1_out = '"String 1";"String2";"Test 3"\n'

        string_2_in = '"1;2;3"\n'
        string_2_out = '"123"\n'

        string_3_in = '"1";";";3\n'
        string_3_out = '"1";"";3\n'

        string_4_in = '";;;;;;;"\n'
        string_4_out = '""\n'

        string_5_in = '"";"";"";"";";;;"\n'
        string_5_out = '"";"";"";"";""\n'

        self.assertEqual(cleanup_line(string_1_in), string_1_out)
        self.assertEqual(cleanup_line(string_2_in), string_2_out)
        self.assertEqual(cleanup_line(string_3_in), string_3_out)
        self.assertEqual(cleanup_line(string_4_in), string_4_out)
        self.assertEqual(cleanup_line(string_5_in), string_5_out)

    def test_cleanup_line_quotes(self):
        self.maxDiff = None

        string_1_in = '"46100";"Kleinwerk. bis 250,--€ Service";0;"1040";"BMW";0;2021-01-04 00:00;45816797;"46100";2022-01-25 00:00;2;"ERGT22500205";"SchraubendreherSechskantschl. 3/8"";"Schlaugschraub";"";"5050";"DIVERSE";"CFO";"EINKAUF";0;"";1;"HD_INL";"999_SONST";0;12017234;2022-01-22 00:00;"6281206";2;"73188";"EKRGG-ET";"LL";1;"";"";0;"5050";"Teiledienst  Betrieb LL";2022-01-25 00:00;"50";"46100";-118,92;118,92;"";"46100";"1";"AHR";"50";"LL";"50";"LL";"Verbrenner"\n'
        string_1_out = '"46100";"Kleinwerk. bis 250,--€ Service";0;"1040";"BMW";0;2021-01-04 00:00;45816797;"46100";2022-01-25 00:00;2;"ERGT22500205";"SchraubendreherSechskantschl. 3/8";"Schlaugschraub";"";"5050";"DIVERSE";"CFO";"EINKAUF";0;"";1;"HD_INL";"999_SONST";0;12017234;2022-01-22 00:00;"6281206";2;"73188";"EKRGG-ET";"LL";1;"";"";0;"5050";"Teiledienst  Betrieb LL";2022-01-25 00:00;"50";"46100";-118,92;118,92;"";"46100";"1";"AHR";"50";"LL";"50";"LL";"Verbrenner"\n'

        string_2_in = '"INTB24103898";"INTERN_MM";"INTERN_MM";"Autohaus Reisacher GmbH";"Europastr. 1";"Memmingen";2024-11-21 00:00;2024-12-03 00:00;"BAR";"MM";"1080";"BMW";"KUNDE";"";0;0;"540";"";"";"INTERN";"";"Autohaus Reisacher GmbH";"Europastr. 1";"Memmingen";0;2024-12-03 00:00;"";"";"";"INTBEL";"";"AHR\\CKI";2;"Aufträge älter 60 Tage";"";0;"MM";"WBA11AL0807L48300";"M235i xDrive Gran Coupé";"INTB24103898";200000;"34356888167";"Bremsbelagfühler vorne links";"";1;19,78;17,17378000000000000000;0;19,78;19,78000000000000000000;"1040";"BMW";0,00000000000000000000;"INTERN";"511_ART";17,17378000000000000000;"";0;2;1;"";0;1753-01-01 00:00;0;"WAU24116299";70000;"";"39";"";"13304";"";"13304";"Christopher";"Kirmayer";"";"";"";"Christopher Kirmayer";"1";"10";"Intern";"BMW";"M235i xDrive Gran Coupé";"WBA11AL0807L48300 - M235i xDrive Gran Coupé - GA - EZ: 2022-09-29 - Anl.Dat.: 2022-07-25";"BMW";"AKTAUFTR";"";"AKTAUFTR";"";"";"Intern";"INTERN_MM";"Autohaus Reisacher GmbH BMW";"INTERN_MM - Autohaus Reisacher GmbH BMW";"";"";"";0;19,78;0;"INTB24103898";"34356888167";17,17;1;19,78;0;17,17;0;"";;;2024-12-03 00:00;"INTB24103898 - Aufträge älter 60 Tage - Autohaus Reisacher GmbH BMW";"Rechnungen älter 30 Tage";1;17;0,0588235;"";"";"";;0;;"INTERN_MM";"Autohaus Reisacher GmbH BMW";"";"Intern";"INTERN_MM - Autohaus Reisacher GmbH BMW";0;0;0;"1";"10";0;"Nachlass < 90 %";0;"Serviceberater";"1080";"Aufträge älter 30 Tage";"Rechnungen älter 30 Tage";"Aufträge älter 100 Tage";"Rechnungen älter 100 Tage";"M235i xDrive Gran Coupé";"WBA11AL0807L48300";"INTERN";0;0;"keine Angabe";"";"VK > EK";"Rechnung";"";"";"";"87700";"8";"87";"877";"8770";"87700";"Autohaus Reisacher GmbH BMW";"511_ART";0;0;0;0;"AHR";"MM";"";;;"";"";"Christopher Kirmayer";"Serviceberater";"Service";"";0;0;12;0,705882352941177;"Service";"PKW";"SCB BMW";1753-01-01 00:00;"SCB";"Christopher Kirmayer";"10";"1040";0;0;0;0;0;0;0;0;"""INTB24103898";"INTERN_MM";"INTERN_MM";"Autohaus Reisacher GmbH";"Europastr. 1";"Memmingen";2024-11-21 00:00;2024-12-03 00:00;"BAR";"MM";"1080";"BMW";"KUNDE";"";0;0;"540";"";"";"INTERN";"";"Autohaus Reisacher GmbH";"Europastr. 1";"Memmingen";0;2024-12-03 00:00;"";"";"";"INTBEL";"";"AHR\\CKI";2;"Aufträge älter 60 Tage";"";0;"MM";"WBA11AL0807L48300";"M235i xDrive Gran Coupé";"INTB24103898";250000;"80050";"Service Bremse hinten";"";8;13,5;0,00000000000000000000;0;108;108,00000000000000000000;"1040";"BMW";0,00000000000000000000;"INTERN";"011_LOHN";0,00000000000000000000;"";0;2;5;"";0;1753-01-01 00:00;0;"WAU24116299";130000;"0000614";"";"";"13304";"";"13304";"Christopher";"Kirmayer";"";"";"";"Christopher Kirmayer";"1";"10";"Intern";"BMW";"M235i xDrive Gran Coupé";"WBA11AL0807L48300 - M235i xDrive Gran Coupé - GA - EZ: 2022-09-29 - Anl.Dat.: 2022-07-25";"BMW";"AKTAUFTR";"";"AKTAUFTR";"";"";"Intern";"INTERN_MM";"Autohaus Reisacher GmbH BMW";"INTERN_MM - Autohaus Reisacher GmbH BMW";"";"";"";108;0;0;"";"";;1;0;0;0;8;"0000614";0;0;2024-12-03 00:00;"INTB24103898 - Aufträge älter 60 Tage - Autohaus Reisacher GmbH BMW";"Rechnungen älter 30 Tage";1;17;0,0588235;"";"";"MECH";0;40;0;"INTERN_MM";"Autohaus Reisacher GmbH BMW";"";"Intern";"INTERN_MM - Autohaus Reisacher GmbH BMW";0;0;0;"1";"10";0;"Nachlass < 90 %";0;"Serviceberater";"1080";"Aufträge älter 30 Tage";"Rechnungen älter 30 Tage";"Aufträge älter 100 Tage";"Rechnungen älter 100 Tage";"M235i xDrive Gran Coupé";"WBA11AL0807L48300";"INTERN";0;0;"keine Angabe";"011_LOHN";"VK > EK";"Rechnung";"";"";"";"87700";"8";"87";"877";"8770";"87700";"Autohaus Reisacher GmbH BMW";"011_LOHN";0;0;0;0;"AHR";"MM";"";;;"";"";"Christopher Kirmayer";"Serviceberater";"Service";"";13,5;108;12;0,705882352941177;"Service";"PKW";"SCB BMW";1753-01-01 00:00;"SCB";"Christopher Kirmayer";"10";"1040";0;0;0;0;0;0;0;0;""\n'
        string_2_out = '"INTB24103898";"INTERN_MM";"INTERN_MM";"Autohaus Reisacher GmbH";"Europastr. 1";"Memmingen";2024-11-21 00:00;2024-12-03 00:00;"BAR";"MM";"1080";"BMW";"KUNDE";"";0;0;"540";"";"";"INTERN";"";"Autohaus Reisacher GmbH";"Europastr. 1";"Memmingen";0;2024-12-03 00:00;"";"";"";"INTBEL";"";"AHR\\CKI";2;"Aufträge älter 60 Tage";"";0;"MM";"WBA11AL0807L48300";"M235i xDrive Gran Coupé";"INTB24103898";200000;"34356888167";"Bremsbelagfühler vorne links";"";1;19,78;17,17378000000000000000;0;19,78;19,78000000000000000000;"1040";"BMW";0,00000000000000000000;"INTERN";"511_ART";17,17378000000000000000;"";0;2;1;"";0;1753-01-01 00:00;0;"WAU24116299";70000;"";"39";"";"13304";"";"13304";"Christopher";"Kirmayer";"";"";"";"Christopher Kirmayer";"1";"10";"Intern";"BMW";"M235i xDrive Gran Coupé";"WBA11AL0807L48300 - M235i xDrive Gran Coupé - GA - EZ: 2022-09-29 - Anl.Dat.: 2022-07-25";"BMW";"AKTAUFTR";"";"AKTAUFTR";"";"";"Intern";"INTERN_MM";"Autohaus Reisacher GmbH BMW";"INTERN_MM - Autohaus Reisacher GmbH BMW";"";"";"";0;19,78;0;"INTB24103898";"34356888167";17,17;1;19,78;0;17,17;0;"";;;2024-12-03 00:00;"INTB24103898 - Aufträge älter 60 Tage - Autohaus Reisacher GmbH BMW";"Rechnungen älter 30 Tage";1;17;0,0588235;"";"";"";;0;;"INTERN_MM";"Autohaus Reisacher GmbH BMW";"";"Intern";"INTERN_MM - Autohaus Reisacher GmbH BMW";0;0;0;"1";"10";0;"Nachlass < 90 %";0;"Serviceberater";"1080";"Aufträge älter 30 Tage";"Rechnungen älter 30 Tage";"Aufträge älter 100 Tage";"Rechnungen älter 100 Tage";"M235i xDrive Gran Coupé";"WBA11AL0807L48300";"INTERN";0;0;"keine Angabe";"";"VK > EK";"Rechnung";"";"";"";"87700";"8";"87";"877";"8770";"87700";"Autohaus Reisacher GmbH BMW";"511_ART";0;0;0;0;"AHR";"MM";"";;;"";"";"Christopher Kirmayer";"Serviceberater";"Service";"";0;0;12;0,705882352941177;"Service";"PKW";"SCB BMW";1753-01-01 00:00;"SCB";"Christopher Kirmayer";"10";"1040";0;0;0;0;0;0;0;0;"INTB24103898";"INTERN_MM";"INTERN_MM";"Autohaus Reisacher GmbH";"Europastr. 1";"Memmingen";2024-11-21 00:00;2024-12-03 00:00;"BAR";"MM";"1080";"BMW";"KUNDE";"";0;0;"540";"";"";"INTERN";"";"Autohaus Reisacher GmbH";"Europastr. 1";"Memmingen";0;2024-12-03 00:00;"";"";"";"INTBEL";"";"AHR\\CKI";2;"Aufträge älter 60 Tage";"";0;"MM";"WBA11AL0807L48300";"M235i xDrive Gran Coupé";"INTB24103898";250000;"80050";"Service Bremse hinten";"";8;13,5;0,00000000000000000000;0;108;108,00000000000000000000;"1040";"BMW";0,00000000000000000000;"INTERN";"011_LOHN";0,00000000000000000000;"";0;2;5;"";0;1753-01-01 00:00;0;"WAU24116299";130000;"0000614";"";"";"13304";"";"13304";"Christopher";"Kirmayer";"";"";"";"Christopher Kirmayer";"1";"10";"Intern";"BMW";"M235i xDrive Gran Coupé";"WBA11AL0807L48300 - M235i xDrive Gran Coupé - GA - EZ: 2022-09-29 - Anl.Dat.: 2022-07-25";"BMW";"AKTAUFTR";"";"AKTAUFTR";"";"";"Intern";"INTERN_MM";"Autohaus Reisacher GmbH BMW";"INTERN_MM - Autohaus Reisacher GmbH BMW";"";"";"";108;0;0;"";"";;1;0;0;0;8;"0000614";0;0;2024-12-03 00:00;"INTB24103898 - Aufträge älter 60 Tage - Autohaus Reisacher GmbH BMW";"Rechnungen älter 30 Tage";1;17;0,0588235;"";"";"MECH";0;40;0;"INTERN_MM";"Autohaus Reisacher GmbH BMW";"";"Intern";"INTERN_MM - Autohaus Reisacher GmbH BMW";0;0;0;"1";"10";0;"Nachlass < 90 %";0;"Serviceberater";"1080";"Aufträge älter 30 Tage";"Rechnungen älter 30 Tage";"Aufträge älter 100 Tage";"Rechnungen älter 100 Tage";"M235i xDrive Gran Coupé";"WBA11AL0807L48300";"INTERN";0;0;"keine Angabe";"011_LOHN";"VK > EK";"Rechnung";"";"";"";"87700";"8";"87";"877";"8770";"87700";"Autohaus Reisacher GmbH BMW";"011_LOHN";0;0;0;0;"AHR";"MM";"";;;"";"";"Christopher Kirmayer";"Serviceberater";"Service";"";13,5;108;12;0,705882352941177;"Service";"PKW";"SCB BMW";1753-01-01 00:00;"SCB";"Christopher Kirmayer";"10";"1040";0;0;0;0;0;0;0;0;""\n'

        self.assertEqual(cleanup_line(string_1_in), string_1_out)
        self.assertEqual(cleanup_line(string_2_in), string_2_out)

    def test_cleanup_carriage_return(self):
        string_1_in = '"String 1";"String\r2";"Test 3"\n'
        string_1_out = '"String 1";"String2";"Test 3"\n'

        string_2_in = '"82132";"Erl. Schadensersatz FH 0%";"Erlöse Leasingschäden Freier Handel 0%";"abu";1;"2";"82132";2025-09-19 00:00;202509;1293654;"1";"0";0,00;-25,21;0,000;0,000;2102;"01";"";"";"";"";"";"";"";"";"";"";"";2222009;"";"";"";"82132";29436839;2025-09-19 00:00;"ego";"GSOR44";"8345";"V";"3";"Fzg.  579857      Spitze Dachantennenstab abgenagt,\rerneue";"01";"AH Tabor Achern";0;2102;"Gebrauchtwagen (Lagerzugang)";"Gebrauchtwagen (Lagerzugang)";"01";"1";-25,21;"dere01";0,000;"1293654 - Fzg.  579857      Spitze Dachantennenstab abgenagt,\rerneue";"8";;;;;"";"1"\n'
        string_2_out = '"82132";"Erl. Schadensersatz FH 0%";"Erlöse Leasingschäden Freier Handel 0%";"abu";1;"2";"82132";2025-09-19 00:00;202509;1293654;"1";"0";0,00;-25,21;0,000;0,000;2102;"01";"";"";"";"";"";"";"";"";"";"";"";2222009;"";"";"";"82132";29436839;2025-09-19 00:00;"ego";"GSOR44";"8345";"V";"3";"Fzg.  579857      Spitze Dachantennenstab abgenagt,erneue";"01";"AH Tabor Achern";0;2102;"Gebrauchtwagen (Lagerzugang)";"Gebrauchtwagen (Lagerzugang)";"01";"1";-25,21;"dere01";0,000;"1293654 - Fzg.  579857      Spitze Dachantennenstab abgenagt,erneue";"8";;;;;"";"1"\n'

        self.assertEqual(cleanup_line(string_1_in), string_1_out)
        self.assertEqual(cleanup_line(string_2_in), string_2_out)