# vim: ft=python fileencoding=utf-8 sts=4 sw=4 et:

# Copyright 2017 Michal Siedlaczek <michal.siedlaczek@gmail.com>

# This file is part of qutebrowser.
#
# qutebrowser is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# qutebrowser is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with qutebrowser.  If not, see <http://www.gnu.org/licenses/>.


from os.path import basename, join

import pytest

from qutebrowser.browser.webengine import spell

AFRIKAANS = spell.Language('af-ZA',
                           'Afrikaans (South Africa)',
                           'af-ZA-3-0.bdic')
ENGLISH = spell.Language('en-US',
                         'English (United States)',
                         'en-US-7-1.bdic')
POLISH = spell.Language('pl-PL',
                        'Polish (Poland)',
                        'pl-PL-3-0.bdic')

LANGUAGE_LIST = [AFRIKAANS, ENGLISH, POLISH]


def test_get_installed_languages_empty(tmpdir, mocker):
    mocker.patch('qutebrowser.browser.webengine.spell.get_dictionary_dir',
                 lambda: '/some-non-existing-dir')
    assert spell.get_installed_languages() == []


def test_get_installed_languages_non_empty(tmpdir, mocker):
    mocker.patch('qutebrowser.browser.webengine.spell.get_dictionary_dir',
                 lambda: str(tmpdir))
    for lang in LANGUAGE_LIST:
        open(join(str(tmpdir), lang.file), 'w', encoding='UTF-8').close()
    for actual, expected in zip(spell.get_installed_languages(),
                                LANGUAGE_LIST):
        assert (actual.code, actual.name, actual.file) ==\
               (expected.code, expected.name, expected.file)


def test_get_available_languages():
    language_list = spell.get_available_languages()
    assert len(language_list) == 42
    first_lang = language_list[0]
    assert (first_lang.code, first_lang.name, first_lang.file) ==\
           (AFRIKAANS.code, AFRIKAANS.name, AFRIKAANS.file)


def test_filter_languages():
    filtered_languages = spell.filter_languages(LANGUAGE_LIST, ['af-ZA'])
    assert filtered_languages == [AFRIKAANS]
    filtered_languages = spell.filter_languages(LANGUAGE_LIST,
                                                ['pl-PL', 'en-US'])
    assert filtered_languages == [ENGLISH, POLISH]
    with pytest.raises(ValueError):
        spell.filter_languages(LANGUAGE_LIST, ['pl-PL', 'en-GB'])
    filtered_languages = spell.filter_languages(LANGUAGE_LIST,
                                                ['pl-PL-3-0.bdic'],
                                                by=lambda lang: lang.file)
    assert filtered_languages == [POLISH]


def test_install(tmpdir, mocker):
    mocker.patch('qutebrowser.browser.webengine.spell.get_dictionary_dir',
                 lambda: str(tmpdir))
    mocker.patch('qutebrowser.browser.webengine.spell.download_dictionary',
                 lambda url, dest: open(dest, 'w', encoding='UTF-8').close())
    spell.install(LANGUAGE_LIST)
    installed_files = [basename(str(file)) for file in tmpdir.listdir()]
    expected_files = [lang.file for lang in LANGUAGE_LIST]
    assert sorted(installed_files) == sorted(expected_files)


# TODO: move somewhere to be checked before a release
#def test_available_langs():
#    for lang in spell.get_available_languages():
#        lang_url = urljoin(spell.repository_url, lang.file)
#        response = head(lang_url)
#        assert response.status_code == 302