import datetime

import pytz
from parameterized import param, parameterized

from dateparser.conf import Settings, apply_settings
from dateparser.search import search_dates
from dateparser.search.search import DateSearchWithDetection
from dateparser.timezone_parser import StaticTzInfo
from dateparser_data.settings import default_parsers
from tests import BaseTestCase

today = datetime.datetime.today()


class TestTranslateSearch(BaseTestCase):
    def setUp(self):
        super().setUp()
        self.search_with_detection = DateSearchWithDetection()
        self.exact_language_search = self.search_with_detection.search

    def run_search_dates_function_invalid_languages(self, text, languages, error_type):
        try:
            search_dates(text=text, languages=languages)
        except Exception as error:
            self.error = error
            self.assertIsInstance(self.error, error_type)

    def check_error_message(self, message):
        self.assertEqual(str(self.error), message)

    @parameterized.expand(
        [
            # English
            param("en", "Sep 03 2014"),
            param("en", "friday, 03 september 2014"),
            param("en", "Aug 06, 2018 05:05 PM CDT"),
            # Chinese
            param("zh", "1年11个月"),
            param("zh", "1年11個月"),
            param("zh", "2015年04月08日10点05"),
            param("zh", "2015年04月08日10:05"),
            param("zh", "2013年04月08日"),
            param("zh", "周一"),
            param("zh", "礼拜一"),
            param("zh", "周二"),
            param("zh", "礼拜二"),
            param("zh", "周三"),
            param("zh", "礼拜三"),
            param("zh", "星期日 2015年04月08日10:05"),
            param("zh", "周六 2013年04月08日"),
            param("zh", "下午3:30"),
            param("zh", "凌晨3:30"),
            param("zh", "中午"),
            # French
            param("fr", "20 Février 2012"),
            param("fr", "Mercredi 19 Novembre 2013"),
            param("fr", "18 octobre 2012 à 19 h 21 min"),
            # German
            param("de", "29. Juni 2007"),
            param("de", "Montag 5 Januar, 2015"),
            # Hungarian
            param("hu", "2016 augusztus 11"),
            param("hu", "2016-08-13 szombat 10:21"),
            param("hu", "2016. augusztus 14. vasárnap 10:21"),
            param("hu", "hétfő"),
            param("hu", "tegnapelőtt"),
            param("hu", "ma"),
            param("hu", "2 hónappal ezelőtt"),
            param("hu", "2016-08-13 szombat 10:21 GMT"),
            # Spanish
            param("es", "Miércoles 31 Diciembre 2014"),
            # Italian
            param("it", "Giovedi Maggio 29 2013"),
            param("it", "19 Luglio 2013"),
            # Portuguese
            param("pt", "22 de dezembro de 2014 às 02:38"),
            # Russian
            param("ru", "5 августа 2014 г в 12:00"),
            # Real: param('ru', "5 августа 2014 г. в 12:00"),
            # Turkish
            param("tr", "2 Ocak 2015 Cuma, 16:49"),
            # Czech
            param("cs", "22. prosinec 2014 v 2:38"),
            # Dutch
            param("nl", "maandag 22 december 2014 om 2:38"),
            # Romanian
            param("ro", "22 Decembrie 2014 la 02:38"),
            # Polish
            param("pl", "4 stycznia o 13:50"),
            param("pl", "29 listopada 2014 o 08:40"),
            # Ukrainian
            param("uk", "30 листопада 2013 о 04:27"),
            # Belarusian
            param("be", "5 снежня 2015 г у 12:00"),
            # Real: param('be', "5 снежня 2015 г. у 12:00"), Issue: Abbreviation segmentation.
            param("be", "11 верасня 2015 г у 12:11"),
            # Real: param('be', "11 верасня 2015 г. у 12:11"),
            param("be", "3 стд 2015 г у 10:33"),
            # Real: param('be', "3 стд 2015 г. у 10:33"),
            # Arabic
            param("ar", "6 يناير، 2015، الساعة 05:16 مساءً"),
            param("ar", "7 يناير، 2015، الساعة 11:00 صباحاً"),
            # Vietnamese
            # Disabled - wrong segmentation at "Thứ Năm"
            # param('vi', "Thứ Năm, ngày 8 tháng 1 năm 2015"),
            # Disabled - wrong segmentation at "Thứ Tư"
            # param('vi', "Thứ Tư, 07/01/2015 | 22:34"),
            param("vi", "9 Tháng 1 2015 lúc 15:08"),
            # Thai
            # Disabled - spacing differences
            # param('th', "เมื่อ กุมภาพันธ์ 09, 2015, 09:27:57 AM"),
            # param('th', "เมื่อ กรกฎาคม 05, 2012, 01:18:06 AM"),
            # Tagalog
            param("tl", "Biyernes Hulyo 3, 2015"),
            param("tl", "Pebrero 5, 2015 7:00 pm"),
            # Indonesian
            param("id", "06 Sep 2015"),
            param("id", "07 Feb 2015 20:15"),
            # Miscellaneous
            param("en", "2014-12-12T12:33:39-08:00"),
            param("en", "2014-10-15T16:12:20+00:00"),
            param("en", "28 Oct 2014 16:39:01 +0000"),
            # Disabled - wrong split at "a las".
            # param('es', "13 Febrero 2015 a las 23:00"),
            # Danish
            param("da", "Sep 03 2014"),
            param("da", "fredag, 03 september 2014"),
            param("da", "fredag d. 3 september 2014"),
            # Finnish
            param("fi", "maanantai tammikuu 16, 2015"),
            param("fi", "ma tammi 16, 2015"),
            param("fi", "tiistai helmikuu 16, 2015"),
            param("fi", "ti helmi 16, 2015"),
            param("fi", "keskiviikko maaliskuu 16, 2015"),
            param("fi", "ke maalis 16, 2015"),
            param("fi", "torstai huhtikuu 16, 2015"),
            param("fi", "to huhti 16, 2015"),
            param("fi", "perjantai toukokuu 16, 2015"),
            param("fi", "pe touko 16, 2015"),
            param("fi", "lauantai kesäkuu 16, 2015"),
            param("fi", "la kesä 16, 2015"),
            param("fi", "sunnuntai heinäkuu 16, 2015"),
            param("fi", "su heinä 16, 2015"),
            param("fi", "su elokuu 16, 2015"),
            param("fi", "su elo 16, 2015"),
            param("fi", "su syyskuu 16, 2015"),
            param("fi", "su syys 16, 2015"),
            param("fi", "su lokakuu 16, 2015"),
            param("fi", "su loka 16, 2015"),
            param("fi", "su marraskuu 16, 2015"),
            param("fi", "su marras 16, 2015"),
            param("fi", "su joulukuu 16, 2015"),
            param("fi", "su joulu 16, 2015"),
            param("fi", "1. tammikuuta, 2016"),
            param("fi", "tiistaina, 27. lokakuuta 2015"),
            # Japanese
            param("ja", "午後3時"),
            param("ja", "2時"),
            param("ja", "11時42分"),
            param("ja", "3ヶ月"),
            param("ja", "約53か月前"),
            param("ja", "3月"),
            param("ja", "十二月"),
            param("ja", "2月10日"),
            param("ja", "2013年2月"),
            param("ja", "2013年04月08日"),
            param("ja", "2016年03月24日 木曜日 10時05分"),
            param("ja", "2016年3月20日 21時40分"),
            param("ja", "2016年03月21日 23時05分11秒"),
            param("ja", "2016年3月21日(月) 14時48分"),
            param("ja", "2016年3月20日(日) 21時40分"),
            param("ja", "2016年3月20日 (日) 21時40分"),
            param("ja", "正午"),
            param("ja", "明後日"),
            param("ja", "明後日の正午"),
            # Hebrew
            param("he", "20 לאפריל 2012"),
            param("he", "יום רביעי ה-19 בנובמבר 2013"),
            param("he", "18 לאוקטובר 2012 בשעה 19:21"),
            # Disabled - wrong split at "יום ה'".
            # param('he', "יום ה' 6/10/2016"),
            param("he", "חצות"),
            param("he", "1 אחר חצות"),
            param("he", "3 לפנות בוקר"),
            param("he", "3 בבוקר"),
            param("he", "3 בצהריים"),
            param("he", "6 לפנות ערב"),
            param("he", "6 אחרי הצהריים"),
            param("he", "6 אחרי הצהרים"),
            # Bangla
            param("bn", "সেপ্টেম্বর 03 2014"),
            param("bn", "শুক্রবার, 03 সেপ্টেম্বর 2014"),
            # Hindi
            param("hi", "सोमवार 13 जून 1998"),
            param("hi", "मंगल 16 1786 12:18"),
            param("hi", "शनि 11 अप्रैल 2002 03:09"),
            # Swedish
            param("sv", "Sept 03 2014"),
            param("sv", "fredag, 03 september 2014"),
        ]
    )
    def test_search_date_string(self, shortname, datetime_string):
        result = self.exact_language_search.search(
            shortname, datetime_string, settings=Settings()
        )[1][0]
        self.assertEqual(result, datetime_string)

    @parameterized.expand(
        [
            # Arabic
            param(
                "ar",
                "في 29 يوليو 1938 غزت القوات اليابانية الاتحاد"
                " السوفييتي ووقعت أولى المعارك والتي انتصر فيها السوفييت، وعلى الرغم من ذلك رفضت"
                " اليابان الاعتراف بذلك وقررت في 11 مايو 1939 تحريك الحدود المنغولية حتى نهر غول،"
                " حيث وقعت معركة خالخين غول والتي انتصر فيها الجيش الأحمر على جيش كوانتونغ",
                [
                    ("في 29 يوليو 1938", datetime.datetime(1938, 7, 29, 0, 0)),
                    ("في 11 مايو 1939", datetime.datetime(1939, 5, 11, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Belarusian
            param(
                "be",
                "Пасля апублікавання Патсдамскай дэкларацыі 26 ліпеня 1945 года і адмовы Японіі капітуляваць "
                "на яе ўмовах ЗША скінулі атамныя бомбы.",
                [("26 ліпеня 1945 года і", datetime.datetime(1945, 7, 26, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Bulgarian
            param(
                "bg",
                "На 16 юни 1944 г. започват въздушни "
                "бомбардировки срещу Япония, използувайки новозавладените острови като бази.",
                [("На 16 юни 1944 г", datetime.datetime(1944, 6, 16, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Chinese
            param(
                "zh",
                "不過大多數人仍多把第二次世界大戰的爆發定為1939年9月1日德國入侵波蘭開始，這次入侵行動隨即導致英國與法國向德國宣戰。",
                [("1939年9月1", datetime.datetime(1939, 9, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Czech
            param(
                "cs",
                "V roce 1920 byla proto vytvořena Společnost národů, jež měla fungovat jako fórum, "
                "na němž měly národy mírovým způsobem urovnávat svoje spory.",
                [("1920", datetime.datetime(1920, 1, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Danish
            param(
                "da",
                "Krigen i Europa begyndte den 1. september 1939, da Nazi-Tyskland invaderede Polen, "
                "og endte med Nazi-Tysklands betingelsesløse overgivelse den 8. maj 1945.",
                [
                    ("1. september 1939", datetime.datetime(1939, 9, 1, 0, 0)),
                    ("8. maj 1945", datetime.datetime(1945, 5, 8, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Dutch
            param(
                "nl",
                " De meest dramatische uitbreiding van het conflict vond plaats op 22 juni 1941 met de "
                "Duitse aanval op de Sovjet-Unie.",
                [("22 juni 1941", datetime.datetime(1941, 6, 22, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # English
            param(
                "en",
                "I will meet you tomorrow at noon",
                [("tomorrow at noon", datetime.datetime(2000, 1, 2, 12, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            param(
                "en",
                "in a minute",
                [("in a minute", datetime.datetime(2000, 1, 1, 0, 1))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            param(
                "en",
                "last decade",
                [("last decade", datetime.datetime(1990, 1, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            param(
                "en",
                "July 13th.\r\n July 14th",
                [
                    ("July 13th", datetime.datetime(2000, 7, 13, 0, 0)),
                    ("July 14th", datetime.datetime(2000, 7, 14, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            param(
                "en",
                "last updated Aug 06, 2018 05:05 PM CDT",
                [
                    (
                        "Aug 06, 2018 05:05 PM CDT",
                        datetime.datetime(
                            2018,
                            8,
                            6,
                            17,
                            5,
                            tzinfo=StaticTzInfo(
                                "CDT", datetime.timedelta(seconds=-18000)
                            ),
                        ),
                    )
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            param(
                "en",
                "25th march 2015 , i need this report today.",
                [("25th march 2015", datetime.datetime(2015, 3, 25))],
                settings={
                    "PARSERS": [
                        parser
                        for parser in default_parsers
                        if parser != "relative-time"
                    ]
                },
            ),
            param(
                "en",
                "25th march 2015 , i need this report today.",
                [
                    ("25th march 2015", datetime.datetime(2015, 3, 25)),
                    ("today", datetime.datetime(2000, 1, 1)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            param(
                "en",
                "The employee has not submitted their documents till date",
                [("till date", datetime.datetime(2000, 1, 1))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Filipino / Tagalog
            param(
                "tl",
                "Maraming namatay sa mga Hapon hanggang sila'y sumuko noong Agosto 15, 1945.",
                [("noong Agosto 15, 1945", datetime.datetime(1945, 8, 15, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Finnish
            param(
                "fi",
                "Iso-Britannia ja Ranska julistivat sodan Saksalle 3. syyskuuta 1939.",
                [("3. syyskuuta 1939", datetime.datetime(1939, 9, 3, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # French
            param(
                "fr",
                "La 2e Guerre mondiale, ou Deuxième Guerre mondiale4, est un conflit armé à "
                "l'échelle planétaire qui dura du 1 septembre 1939 au 2 septembre 1945.",
                [
                    ("1 septembre 1939", datetime.datetime(1939, 9, 1, 0, 0)),
                    ("2 septembre 1945", datetime.datetime(1945, 9, 2, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Hebrew
            param(
                "he",
                'במרץ 1938 "אוחדה" אוסטריה עם גרמניה (אנשלוס). ',
                [("במרץ 1938", datetime.datetime(1938, 3, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Hindi
            param(
                "hi",
                "जुलाई 1937 में, मार्को-पोलो ब्रिज हादसे का बहाना लेकर जापान ने चीन पर हमला कर दिया और चीनी साम्राज्य "
                "की राजधानी बीजिंग पर कब्जा कर लिया,",
                [("जुलाई 1937 में", datetime.datetime(1937, 7, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Hungarian
            param(
                "hu",
                "A háború Európában 1945. május 8-án Németország feltétel nélküli megadásával, "
                "míg Ázsiában szeptember 2-án, Japán kapitulációjával fejeződött be.",
                [
                    ("1945. május 8-án", datetime.datetime(1945, 5, 8, 0, 0)),
                    ("szeptember 2-án", datetime.datetime(2000, 9, 2, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Georgian
            param(
                "ka",
                "1937 წელს დაიწყო იაპონია-ჩინეთის მეორე ომი.",
                [("1937", datetime.datetime(1937, 1, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # German
            param(
                "de",
                "Die UdSSR blieb gemäß dem Neutralitätspakt "
                "vom 13. April 1941 gegenüber Japan vorerst neutral.",
                [
                    ("Die", datetime.datetime(1999, 12, 28, 0, 0)),
                    ("13. April 1941", datetime.datetime(1941, 4, 13, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Indonesian
            param(
                "id",
                "Kekaisaran Jepang menyerah pada tanggal 15 Agustus 1945, sehingga mengakhiri perang "
                "di Asia dan memperkuat kemenangan total Sekutu atas Poros.",
                [("tanggal 15 Agustus 1945", datetime.datetime(1945, 8, 15, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Italian
            param(
                "it",
                " Con questo il 2 ottobre 1935 prese il via la campagna "
                "d'Etiopia. Il 9 maggio 1936 venne proclamato l'Impero. ",
                [
                    ("2 ottobre 1935", datetime.datetime(1935, 10, 2, 0, 0)),
                    ("9 maggio 1936", datetime.datetime(1936, 5, 9, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Japanese
            param(
                "ja",
                "1939年9月1日、ドイツ軍がポーランドへ侵攻したことが第二次世界大戦の始まりとされている。",
                [("1939年9月1", datetime.datetime(1939, 9, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Persian
            param(
                "fa",
                "نگ جهانی دوم جنگ جدی بین سپتامبر 1939 و 2 سپتامبر 1945 بود.",
                [
                    ("سپتامبر 1939", datetime.datetime(1939, 9, 1, 0, 0)),
                    ("2 سپتامبر 1945", datetime.datetime(1945, 9, 2, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Polish
            param(
                "pl",
                "II wojna światowa – największa wojna światowa w historii, "
                "trwająca od 1 września 1939 do 2 września 1945 (w Europie do 8 maja 1945)",
                [
                    ("1 września 1939", datetime.datetime(1939, 9, 1, 0, 0)),
                    ("2 września 1945 (w", datetime.datetime(1945, 9, 2, 0, 0)),
                    ("8 maja 1945", datetime.datetime(1945, 5, 8, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Portuguese
            param(
                "pt",
                "Em outubro de 1936, Alemanha e Itália formaram o Eixo Roma-Berlim.",
                [("Em outubro de 1936", datetime.datetime(1936, 10, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Romanian
            param(
                "ro",
                "Pe 17 septembrie 1939, după semnarea unui acord de încetare a focului cu Japonia, "
                "sovieticii au invadat Polonia dinspre est.",
                [("17 septembrie 1939", datetime.datetime(1939, 9, 17, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Russian
            param(
                "ru",
                "Втора́я мирова́я война́ (1 сентября 1939 — 2 сентября 1945) — "
                "война двух мировых военно-политических коалиций, ставшая крупнейшим вооружённым "
                "конфликтом в истории человечества.",
                [
                    ("1 сентября 1939", datetime.datetime(1939, 9, 1, 0, 0)),
                    ("2 сентября 1945", datetime.datetime(1945, 9, 2, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Spanish
            param(
                "es",
                "Desde finales de 1939 hasta inicios de 1941 Alemania conquistó o sometió "
                "gran parte de la Europa continental.",
                [
                    ("de 1939", datetime.datetime(1939, 1, 1, 0, 0)),
                    ("de 1941", datetime.datetime(1941, 1, 1, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            param(
                "es",
                "¡¡Ay!! En Madrid, a 17 de marzo de 1615. ¿Vos bueno?",
                [("a 17 de marzo de 1615", datetime.datetime(1615, 3, 17, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Swedish
            param(
                "sv",
                "Efter kommunisternas seger 1922 drog de allierade och Japan bort sina trupper.",
                [("1922", datetime.datetime(1922, 1, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Thai
            param(
                "th",
                "และเมื่อวันที่ 11 พฤษภาคม 1939 "
                "ญี่ปุ่นตัดสินใจขยายพรมแดนญี่ปุ่น-มองโกเลียขึ้นไปถึงแม่น้ำคัลคินกอลด้วยกำลัง",
                [("11 พฤษภาคม 1939", datetime.datetime(1939, 5, 11, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Turkish
            param(
                "tr",
                "Almanya’nın Polonya’yı işgal ettiği 1 Eylül 1939 savaşın başladığı "
                "tarih olarak genel kabul görür.",
                [("1 Eylül 1939", datetime.datetime(1939, 9, 1, 0, 0))],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Ukrainian
            param(
                "uk",
                "Інші дати, що розглядаються деякими авторами як дати початку війни: початок японської "
                "інтервенції в Маньчжурію 13 вересня 1931, початок другої японсько-китайської війни 7 "
                "липня 1937 року та початок угорсько-української війни 14 березня 1939 року.",
                [
                    ("13 вересня 1931", datetime.datetime(1931, 9, 13, 0, 0)),
                    ("7 липня 1937 року", datetime.datetime(1937, 7, 7, 0, 0)),
                    ("14 березня 1939 року", datetime.datetime(1939, 3, 14, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
            # Vietnamese
            param(
                "vi",
                "Ý theo gương Đức, đã tiến hành xâm lược Ethiopia năm 1935 và sát "
                "nhập Albania vào ngày 12 tháng 4 năm 1939.",
                [
                    ("năm 1935", datetime.datetime(1935, 1, 1, 0, 0)),
                    ("ngày 12 tháng 4 năm 1939", datetime.datetime(1939, 4, 12, 0, 0)),
                ],
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
            ),
        ]
    )
    @apply_settings
    def test_search_and_parse(self, shortname, string, expected, settings=None):
        result = self.exact_language_search.search_parse(
            shortname, string, settings=settings
        )
        self.assertEqual(result, expected)

    @parameterized.expand(
        [
            # English
            param(
                "en",
                "January 3, 2017 - February 1st",
                [
                    ("January 3, 2017", datetime.datetime(2017, 1, 3, 0, 0)),
                    ("February 1st", datetime.datetime(2017, 2, 1, 0, 0)),
                ],
            ),
            param(
                "en",
                "2014 was good! October was excellent!"
                " Friday, 21 was especially good!",
                [
                    ("2014", datetime.datetime(2014, today.month, today.day, 0, 0)),
                    ("October", datetime.datetime(2014, 10, today.day, 0, 0)),
                    ("Friday, 21", datetime.datetime(2014, 10, 21, 0, 0)),
                ],
            ),
            param(
                "en",
                """May 2020
                    July 2020
                    2023
                    January UTC
                    June 5 am utc
                    June 23th 5 pm EST
                    May 31, 8am UTC""",
                [
                    (
                        "May 2020",
                        datetime.datetime(
                            2020,
                            5,
                            datetime.datetime.now(tz=datetime.timezone.utc).day,
                            0,
                            0,
                        ),
                    ),
                    (
                        "July 2020",
                        datetime.datetime(
                            2020,
                            7,
                            datetime.datetime.now(tz=datetime.timezone.utc).day,
                            0,
                            0,
                        ),
                    ),
                    (
                        "2023",
                        datetime.datetime(
                            2023,
                            7,
                            datetime.datetime.now(tz=datetime.timezone.utc).day,
                            0,
                            0,
                        ),
                    ),
                    (
                        "January UTC",
                        datetime.datetime(
                            2023,
                            1,
                            datetime.datetime.now(tz=datetime.timezone.utc).day,
                            0,
                            0,
                            tzinfo=pytz.utc,
                        ),
                    ),
                    (
                        "June 5 am utc",
                        datetime.datetime(2023, 6, 5, 0, 0, tzinfo=pytz.utc),
                    ),
                    (
                        "June 23th 5 pm EST",
                        datetime.datetime(
                            2023, 6, 23, 17, 0, tzinfo=pytz.timezone("EST")
                        ),
                    ),
                    ("May 31", datetime.datetime(2023, 5, 31, 0, 0)),
                    ("8am UTC", datetime.datetime(2023, 8, 31, 0, 0, tzinfo=pytz.utc)),
                ],
            ),
            # Russian
            param(
                "ru",
                "19 марта 2001 был хороший день. 20 марта тоже был хороший день. 21 марта был отличный день.",
                [
                    ("19 марта 2001", datetime.datetime(2001, 3, 19, 0, 0)),
                    ("20 марта", datetime.datetime(2001, 3, 20, 0, 0)),
                    ("21 марта", datetime.datetime(2001, 3, 21, 0, 0)),
                ],
            ),
            param(
                "ru",
                "Андрей Дмитриевич Сахаров скончался 14 декабря в 1989 году от внезапной остановки сердца.",
                [("14 декабря в 1989 году", datetime.datetime(1989, 12, 14, 0, 0))],
            ),
            # relative dates
            param(
                "ru",
                "19 марта 2001. Сегодня был хороший день. 2 дня назад был хороший день. "
                "Вчера тоже был хороший день.",
                [
                    ("19 марта 2001", datetime.datetime(2001, 3, 19, 0, 0)),
                    ("Сегодня", datetime.datetime(2001, 3, 19, 0, 0)),
                    ("2 дня назад", datetime.datetime(2001, 3, 17, 0, 0)),
                    ("Вчера", datetime.datetime(2001, 3, 18, 0, 0)),
                ],
            ),
            param(
                "ru",
                "19 марта 2001. Сегодня был хороший день. Два дня назад был хороший день. Хорошая была неделя. "
                "Думаю, через неделю будет еще лучше.",
                [
                    ("19 марта 2001", datetime.datetime(2001, 3, 19, 0, 0)),
                    ("Сегодня", datetime.datetime(2001, 3, 19, 0, 0)),
                    ("Два дня назад", datetime.datetime(2001, 3, 17, 0, 0)),
                    ("через неделю", datetime.datetime(2001, 3, 26, 0, 0)),
                ],
            ),
            # Hungarian
            param(
                "hu",
                "1962 augusztus 11 Föld körüli pályára bocsátották a szovjet Vosztok-3 űrhajót, "
                "mely páros űrrepülést hajtott végre a másnap föld körüli pályára bocsátott Vosztok-4-gyel."
                "2 hónappal ezelőtt furcsa, nem forgó jellegű szédülést tapasztaltam.",
                [
                    ("1962 augusztus 11", datetime.datetime(1962, 8, 11, 0, 0)),
                    ("2 hónappal ezelőtt", datetime.datetime(1962, 6, 11, 0, 0)),
                ],
            ),
            # Vietnamese
            param(
                "vi",
                "1/1/1940. Vào tháng 8 năm 1940, với lực lượng lớn của Pháp tại Bắc Phi chính thức trung lập "
                "trong cuộc chiến, Ý mở một cuộc tấn công vào thuộc địa Somalia của Anh tại Đông Phi. "
                "Đến tháng 9 quân Ý vào đến Ai Cập (cũng đang dưới sự kiểm soát của Anh). ",
                [
                    ("1/1/1940", datetime.datetime(1940, 1, 1, 0, 0)),
                    ("tháng 8 năm 1940", datetime.datetime(1940, 8, 1, 0, 0)),
                    ("tháng 9", datetime.datetime(1940, 9, 1, 0, 0)),
                ],
            ),
        ]
    )
    @apply_settings
    def test_relative_base_setting(self, shortname, string, expected, settings=None):
        result = self.exact_language_search.search_parse(
            shortname, string, settings=settings
        )
        self.assertEqual(result, expected)

    @parameterized.expand(
        [
            # English
            param(
                "en",
                "July 12th, 2014. July 13th, July 14th",
                [
                    ("July 12th, 2014", datetime.datetime(2014, 7, 12, 0, 0)),
                    ("July 13th", datetime.datetime(2014, 7, 13, 0, 0)),
                    ("July 14th", datetime.datetime(2014, 7, 14, 0, 0)),
                ],
            ),
            param(
                "en",
                "2014. July 13th July 14th",
                [
                    ("2014", datetime.datetime(2014, today.month, today.day, 0, 0)),
                    ("July 13th", datetime.datetime(2014, 7, 13, 0, 0)),
                    ("July 14th", datetime.datetime(2014, 7, 14, 0, 0)),
                ],
            ),
            param(
                "en",
                "July 13th 2014 July 14th 2014",
                [
                    ("July 13th 2014", datetime.datetime(2014, 7, 13, 0, 0)),
                    ("July 14th 2014", datetime.datetime(2014, 7, 14, 0, 0)),
                ],
            ),
            param(
                "en",
                "July 13th 2014 July 14th",
                [
                    ("July 13th 2014", datetime.datetime(2014, 7, 13, 0, 0)),
                    ("July 14th", datetime.datetime(2014, 7, 14, 0, 0)),
                ],
            ),
            param(
                "en",
                "July 13th, 2014 July 14th, 2014",
                [
                    ("July 13th, 2014", datetime.datetime(2014, 7, 13, 0, 0)),
                    ("July 14th, 2014", datetime.datetime(2014, 7, 14, 0, 0)),
                ],
            ),
            param(
                "en",
                "2014. July 12th, July 13th, July 14th",
                [
                    ("2014", datetime.datetime(2014, today.month, today.day, 0, 0)),
                    ("July 12th", datetime.datetime(2014, 7, 12, 0, 0)),
                    ("July 13th", datetime.datetime(2014, 7, 13, 0, 0)),
                    ("July 14th", datetime.datetime(2014, 7, 14, 0, 0)),
                ],
            ),
            # Swedish
            param(
                "sv",
                "1938–1939 marscherade tyska soldater i Österrike samtidigt som "
                "österrikiska soldater marscherade i Berlin.",
                [
                    ("1938", datetime.datetime(1938, today.month, today.day, 0, 0)),
                    ("1939", datetime.datetime(1939, today.month, today.day, 0, 0)),
                ],
            ),
            # German
            param(
                "de",
                "Verteidiger der Stadt kapitulierten am 2. Mai 1945. Am 8. Mai 1945 (VE-Day) trat "
                "bedingungslose Kapitulation der Wehrmacht in Kraft",
                [
                    ("am 2. Mai 1945", datetime.datetime(1945, 5, 2, 0, 0)),
                    ("Am 8. Mai 1945", datetime.datetime(1945, 5, 8, 0, 0)),
                ],
            ),
        ]
    )
    @apply_settings
    def test_splitting_of_not_parsed(self, shortname, string, expected, settings=None):
        result = self.exact_language_search.search_parse(
            shortname, string, settings=settings
        )
        self.assertEqual(result, expected)

    @parameterized.expand(
        [
            # Arabic
            param(
                "ar",
                "في 29 يوليو 1938 غزت القوات اليابانية الاتحاد"
                " السوفييتي ووقعت أولى المعارك والتي انتصر فيها السوفييت، وعلى الرغم من ذلك رفضت"
                " اليابان الاعتراف بذلك وقررت في 11 مايو 1939 تحريك الحدود المنغولية حتى نهر غول،",
            ),
            # Belarusian
            param(
                "be",
                "Пасля апублікавання Патсдамскай дэкларацыі 26 ліпеня 1945 года і адмовы Японіі капітуляваць "
                "на яе ўмовах ЗША скінулі атамныя бомбы.",
            ),
            # Bulgarian
            param(
                "bg",
                "На 16 юни 1944 г. започват въздушни "
                "бомбардировки срещу Япония, използувайки новозавладените острови като бази.",
            ),
            # Chinese
            param("zh", "不過大多數人仍多把第二次世界大戰的爆發定為1939年9月1日德國入侵波蘭開始，2015年04月08日10点05。"),
            # Czech
            param(
                "cs",
                "V rok 1920 byla proto vytvořena Společnost národů, jež měla fungovat jako fórum, "
                "na němž měly národy mírovým způsobem urovnávat svoje spory.",
            ),
            # Danish
            param(
                "da",
                "Krigen i Europa begyndte den 1. september 1939, da Nazi-Tyskland invaderede Polen, "
                "og endte med Nazi-Tysklands betingelsesløse overgivelse den 8. marts 1945.",
            ),
            # Dutch
            param(
                "nl",
                " De meest dramatische uitbreiding van het conflict vond plaats op Maandag 22 juni 1941  met de "
                "Duitse aanval op de Sovjet-Unie.",
            ),
            # English
            param("en", "I will meet you tomorrow at noon"),
            # Filipino / Tagalog
            param(
                "tl",
                "Maraming namatay sa mga Hapon hanggang sila'y sumuko noong Agosto 15, 1945.",
            ),
            # Finnish
            param(
                "fi",
                "Iso-Britannia ja Ranska julistivat sodan Saksalle 3. syyskuuta 1939.",
            ),
            # French
            param(
                "fr",
                "La Seconde Guerre mondiale, ou Deuxième Guerre mondiale4, est un conflit armé à "
                "l'échelle planétaire qui dura du 1 septembre 1939 au 2 septembre 1945.",
            ),
            # Hebrew
            param("he", 'במרץ 1938 "אוחדה" אוסטריה עם גרמניה (אנשלוס). '),
            # Hindi
            param(
                "hi",
                "जुलाई 1937 में, मार्को-पोलो ब्रिज हादसे का बहाना लेकर जापान ने चीन पर हमला कर दिया और चीनी साम्राज्य "
                "की राजधानी बीजिंग पर कब्जा कर लिया,",
            ),
            # Hungarian
            param(
                "hu",
                "A háború Európában 1945. május 8-án Németország feltétel nélküli megadásával, "
                "míg Ázsiában szeptember 2-án, Japán kapitulációjával fejeződött be.",
            ),
            # Georgian
            param("ka", "1937 წელს დაიწყო იაპონია-ჩინეთის მეორე ომი."),
            # German
            param(
                "de",
                "Die UdSSR blieb dem Neutralitätspakt "
                "vom 13. April 1941 gegenüber Japan vorerst neutral.",
            ),
            # Indonesian
            param(
                "id",
                "Kekaisaran Jepang menyerah pada tanggal 15 Agustus 1945, sehingga mengakhiri perang "
                "di Asia dan memperkuat kemenangan total Sekutu atas Poros.",
            ),
            # Italian
            param(
                "it",
                " Con questo il 2 ottobre 1935 prese il via la campagna "
                "d'Etiopia. Il 9 maggio 1936 venne proclamato l'Impero. ",
            ),
            # Japanese
            param("ja", "1933年（昭和8年）12月23日午前6時39分、宮城（現：皇居）内の産殿にて誕生。"),
            # Persian
            param("fa", "نگ جهانی دوم جنگ جدی بین سپتامبر 1939 و 2 سپتامبر 1945 بود."),
            # Polish
            param(
                "pl",
                "II wojna światowa – największa wojna światowa w historii, "
                "trwająca od 1 września 1939 do 2 września 1945 (w Europie do 8 maja 1945)",
            ),
            # Portuguese
            param(
                "pt",
                "Em outubro de 1936, Alemanha e Itália formaram o Eixo Roma-Berlim.",
            ),
            # Romanian
            param(
                "ro",
                "Pe 17 septembrie 1939, după semnarea unui acord de încetare a focului cu Japonia, "
                "sovieticii au invadat Polonia dinspre est.",
            ),
            # Russian
            param(
                "ru",
                "Втора́я мирова́я война́ (1 сентября 1939 — 2 сентября 1945) — "
                "война двух мировых военно-политических коалиций, ставшая крупнейшим вооружённым "
                "конфликтом в истории человечества.",
            ),
            # Spanish
            param("es", "11 junio 2010"),
            param("es", "¡¡Ay!! En Madrid, a 17 de marzo de 1615. ¿Vos bueno?"),
            # Swedish
            param("sv", " den 15 augusti 1945 då Kejsardömet"),
            # Thai
            param(
                "th",
                "และเมื่อวันที่ 11 พฤษภาคม 1939 "
                "ญี่ปุ่นตัดสินใจขยายพรมแดนญี่ปุ่น-มองโกเลียขึ้นไปถึงแม่น้ำคัลคินกอลด้วยกำลัง",
            ),
            # Turkish
            param(
                "tr",
                "Almanya’nın Polonya’yı işgal ettiği 1 Eylül 1939 savaşın başladığı "
                "tarih olarak genel kabul görür.",
            ),
            # Ukrainian
            param(
                "uk",
                "Інші дати, що розглядаються деякими авторами як дати початку війни: початок японської "
                "інтервенції в Маньчжурію 13 вересня 1931, початок другої японсько-китайської війни 7 "
                "липня 1937 року та початок угорсько-української війни 14 березня 1939 року.",
            ),
            # Vietnamese
            param(
                "vi",
                "Ý theo gương Đức, đã tiến hành xâm lược Ethiopia năm 1935 và sát "
                "nhập Albania vào ngày 12 tháng 4 năm 1939.",
            ),
            # Only digits
            param("en", "2007"),
        ]
    )
    def test_detection(self, shortname, text):
        result = self.search_with_detection.detect_language(text, languages=None)
        self.assertEqual(result, shortname)

    @parameterized.expand(
        [
            param(
                text="19 марта 2001 был хороший день. 20 марта тоже был хороший день. 21 марта был отличный день.",
                languages=["en", "ru"],
                settings=None,
                expected=[
                    ("19 марта 2001", datetime.datetime(2001, 3, 19, 0, 0)),
                    ("20 марта", datetime.datetime(2001, 3, 20, 0, 0)),
                    ("21 марта", datetime.datetime(2001, 3, 21, 0, 0)),
                ],
            ),
            param(
                text="Em outubro de 1936, Alemanha e Itália formaram o Eixo Roma-Berlim.",
                languages=None,
                settings={"RELATIVE_BASE": datetime.datetime(2000, 1, 1)},
                expected=[("Em outubro de 1936", datetime.datetime(1936, 10, 1, 0, 0))],
            ),
            param(
                text="19 марта 2001, 20 марта, 21 марта был отличный день.",
                languages=["en", "ru"],
                settings=None,
                expected=[
                    ("19 марта 2001", datetime.datetime(2001, 3, 19, 0, 0)),
                    ("20 марта", datetime.datetime(2001, 3, 20, 0, 0)),
                    ("21 марта", datetime.datetime(2001, 3, 21, 0, 0)),
                ],
            ),
            # Dates not found
            param(text="", languages=None, settings=None, expected=None),
            # Language not detected
            param(text="Привет", languages=["en"], settings=None, expected=None),
            # ZeroDivisionError
            param(
                text="DECEMBER 21 19.87 87",
                languages=None,
                settings=None,
                expected=[("DECEMBER 21 19", datetime.datetime(2019, 12, 21, 0, 0))],
            ),
            param(
                text="bonjour, pouvez vous me joindre svp par telephone 08 11 58 54 41",
                languages=None,
                settings={"STRICT_PARSING": True},
                expected=None,
            ),
            param(text="a Americ", languages=None, settings=None, expected=None),
            # Date with comma and apostrophe
            param(
                text="9/3/2017  , ",
                languages=["en"],
                settings=None,
                expected=[("9/3/2017", datetime.datetime(2017, 9, 3, 0, 0))],
            ),
            param(
                text="9/3/2017  ' ",
                languages=["en"],
                settings=None,
                expected=[("9/3/2017", datetime.datetime(2017, 9, 3, 0, 0))],
            ),
        ]
    )
    def test_date_search_function(self, text, languages, settings, expected):
        result = search_dates(text, languages=languages, settings=settings)
        self.assertEqual(result, expected)

    @parameterized.expand(
        [
            param(
                text="15 de outubro de 1936",
                add_detected_language=True,
                expected=[
                    (
                        "15 de outubro de 1936",
                        datetime.datetime(1936, 10, 15, 0, 0),
                        "pt",
                    )
                ],
            ),
            param(
                text="15 de outubro de 1936",
                add_detected_language=False,
                expected=[
                    ("15 de outubro de 1936", datetime.datetime(1936, 10, 15, 0, 0))
                ],
            ),
        ]
    )
    def test_search_dates_returning_detected_languages_if_requested(
        self, text, add_detected_language, expected
    ):
        result = search_dates(text, add_detected_language=add_detected_language)
        self.assertEqual(result, expected)

    @parameterized.expand(
        [
            param(text="19 марта 2001", languages="wrong type: str instead of list"),
        ]
    )
    def test_date_search_function_invalid_languages_type(self, text, languages):
        self.run_search_dates_function_invalid_languages(
            text=text, languages=languages, error_type=TypeError
        )
        self.check_error_message(
            "languages argument must be a list (<class 'str'> given)"
        )

    @parameterized.expand(
        [
            param(text="19 марта 2001", languages=["unknown language code"]),
        ]
    )
    def test_date_search_function_invalid_language_code(self, text, languages):
        self.run_search_dates_function_invalid_languages(
            text=text, languages=languages, error_type=ValueError
        )
        self.check_error_message("Unknown language(s): 'unknown language code'")
