from datetime import datetime
from unittest.mock import Mock
import unittest

from parameterized import parameterized, param

from dateparser.custom_language_detection.fasttext import detect_languages as fast_text_detect_languages
from dateparser.custom_language_detection.langdetect import detect_languages as lang_detect_detect_languages
from dateparser import parse
from dateparser.date import DateDataParser
from dateparser.search import search_dates

detect_languages = Mock()
detect_languages.return_value = ["en"]


class CustomLangDetectParserTest(unittest.TestCase):
    def check_is_returned_list(self):
        self.assertEqual(type(self.result), list)

    @parameterized.expand([
        param(dt_string="14 June 2020", confidence_threshold=0.0),
        param(dt_string="26 July 2021", confidence_threshold=0.0)
    ])
    def test_custom_language_detect_fast_text(self, dt_string, confidence_threshold):
        self.result = fast_text_detect_languages(dt_string, confidence_threshold)
        self.check_is_returned_list()

    @parameterized.expand([
        param(dt_string="14 June 2020", confidence_threshold=0.0),
    ])
    def test_custom_language_detect_lang_detect(self, dt_string, confidence_threshold):
        self.result = lang_detect_detect_languages(dt_string, confidence_threshold)
        self.check_is_returned_list()

    @parameterized.expand([
        param(dt_string="10-10-2021", confidence_threshold=0.5),
    ])
    def test_lang_detect_doesnt_raise_error(self, dt_string, confidence_threshold):
        result = lang_detect_detect_languages(dt_string, confidence_threshold)
        assert result == []

    # Mock test for parse, search_dates and DateDataParser

    detect_languages = Mock()
    detect_languages.return_value = ["en"]

    # parse

    def when_date_is_parsed_using_parse(self, dt_string):
        self.result = parse(dt_string, detect_languages_function=detect_languages)

    def then_date_obj_exactly_is(self, expected_date_obj):
        self.assertEqual(expected_date_obj, self.result)

    @parameterized.expand([
        param("Tuesday Jul 22, 2014", datetime(2014, 7, 22, 0, 0, 0)),
    ])
    def test_custom_language_detect_mock_parse(self, dt_string, expected_date_obj):
        self.when_date_is_parsed_using_parse(dt_string)
        self.then_date_obj_exactly_is(expected_date_obj)

    # DateDataParser

    def when_date_is_parsed_using_with_datedataparser(self, dt_string):
        ddp = DateDataParser(detect_languages_function=detect_languages)
        self.result = ddp.get_date_data(dt_string)["date_obj"]

    @parameterized.expand([
        param("Tuesday Jul 22, 2014", datetime(2014, 7, 22, 0, 0, 0)),
    ])
    def test_custom_language_detect_mock_datedataparser(self, dt_string, expected_date_obj):
        self.when_date_is_parsed_using_with_datedataparser(dt_string)
        self.then_date_obj_exactly_is(expected_date_obj)

    # search_date

    def when_date_is_parsed_using_with_search_dates(self, dt_string):
        self.result = search_dates(dt_string, detect_languages_function=detect_languages)

    @parameterized.expand([
        param('January 3, 2017 - February 1st',
              [('January 3, 2017', datetime(2017, 1, 3, 0, 0)),
               ('February 1st', datetime(2017, 2, 1, 0, 0))]),
    ])
    def test_custom_language_detect_mock_search_dates(self, dt_string, expected_date_obj):
        self.when_date_is_parsed_using_with_search_dates(dt_string)
        self.then_date_obj_exactly_is(expected_date_obj)