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

# Copyright 2014-2015 Florian Bruhin (The Compiler) <mail@qutebrowser.org>
#
# 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/>.

"""Tests for qutebrowser.utils.utils."""

import sys
import enum
import datetime
import os.path

from PyQt5.QtCore import Qt
from PyQt5.QtGui import QColor
import pytest

from qutebrowser.utils import utils, qtutils


class Color(QColor):

    """A QColor with a nicer repr()."""

    def __repr__(self):
        return utils.get_repr(self, constructor=True, red=self.red(),
                              green=self.green(), blue=self.blue(),
                              alpha=self.alpha())


class TestEliding:

    """Test elide."""

    ELLIPSIS = '\u2026'

    def test_too_small(self):
        """Test eliding to 0 chars which should fail."""
        with pytest.raises(ValueError):
            utils.elide('foo', 0)

    def test_length_one(self):
        """Test eliding to 1 char which should yield ..."""
        assert utils.elide('foo', 1) == self.ELLIPSIS

    def test_fits(self):
        """Test eliding with a string which fits exactly."""
        assert utils.elide('foo', 3) == 'foo'

    def test_elided(self):
        """Test eliding with a string which should get elided."""
        assert utils.elide('foobar', 3) == 'fo' + self.ELLIPSIS


class TestReadFile:

    """Test read_file."""

    def test_readfile(self):
        """Read a test file."""
        content = utils.read_file(os.path.join('utils', 'testfile'))
        assert content.splitlines()[0] == "Hello World!"


class TestInterpolateColor:

    """Tests for interpolate_color.

    Attributes:
        white: The Color white as a valid Color for tests.
        white: The Color black as a valid Color for tests.
    """

    @pytest.fixture(autouse=True)
    def setup(self):
        self.white = Color('white')
        self.black = Color('black')

    def test_invalid_start(self):
        """Test an invalid start color."""
        with pytest.raises(qtutils.QtValueError):
            utils.interpolate_color(Color(), self.white, 0)

    def test_invalid_end(self):
        """Test an invalid end color."""
        with pytest.raises(qtutils.QtValueError):
            utils.interpolate_color(self.white, Color(), 0)

    def test_invalid_percentage(self):
        """Test an invalid percentage."""
        with pytest.raises(ValueError):
            utils.interpolate_color(self.white, self.white, -1)
        with pytest.raises(ValueError):
            utils.interpolate_color(self.white, self.white, 101)

    def test_invalid_colorspace(self):
        """Test an invalid colorspace."""
        with pytest.raises(ValueError):
            utils.interpolate_color(self.white, self.black, 10, QColor.Cmyk)

    def test_valid_percentages_rgb(self):
        """Test 0% and 100% in the RGB colorspace."""
        white = utils.interpolate_color(self.white, self.black, 0, QColor.Rgb)
        black = utils.interpolate_color(self.white, self.black, 100,
                                        QColor.Rgb)
        assert Color(white) == self.white
        assert Color(black) == self.black

    def test_valid_percentages_hsv(self):
        """Test 0% and 100% in the HSV colorspace."""
        white = utils.interpolate_color(self.white, self.black, 0, QColor.Hsv)
        black = utils.interpolate_color(self.white, self.black, 100,
                                        QColor.Hsv)
        assert Color(white) == self.white
        assert Color(black) == self.black

    def test_valid_percentages_hsl(self):
        """Test 0% and 100% in the HSL colorspace."""
        white = utils.interpolate_color(self.white, self.black, 0, QColor.Hsl)
        black = utils.interpolate_color(self.white, self.black, 100,
                                        QColor.Hsl)
        assert Color(white) == self.white
        assert Color(black) == self.black

    def test_interpolation_rgb(self):
        """Test an interpolation in the RGB colorspace."""
        color = utils.interpolate_color(Color(0, 40, 100), Color(0, 20, 200),
                                        50, QColor.Rgb)
        assert Color(color) == Color(0, 30, 150)

    def test_interpolation_hsv(self):
        """Test an interpolation in the HSV colorspace."""
        start = Color()
        stop = Color()
        start.setHsv(0, 40, 100)
        stop.setHsv(0, 20, 200)
        color = utils.interpolate_color(start, stop, 50, QColor.Hsv)
        expected = Color()
        expected.setHsv(0, 30, 150)
        assert Color(color) == expected

    def test_interpolation_hsl(self):
        """Test an interpolation in the HSL colorspace."""
        start = Color()
        stop = Color()
        start.setHsl(0, 40, 100)
        stop.setHsl(0, 20, 200)
        color = utils.interpolate_color(start, stop, 50, QColor.Hsl)
        expected = Color()
        expected.setHsl(0, 30, 150)
        assert Color(color) == expected


class TestFormatSeconds:

    """Tests for format_seconds.

    Class attributes:
        TESTS: A list of (input, output) tuples.
    """

    TESTS = [
        (-1, '-0:01'),
        (0, '0:00'),
        (59, '0:59'),
        (60, '1:00'),
        (60.4, '1:00'),
        (61, '1:01'),
        (-61, '-1:01'),
        (3599, '59:59'),
        (3600, '1:00:00'),
        (3601, '1:00:01'),
        (36000, '10:00:00'),
    ]

    @pytest.mark.parametrize('seconds, out', TESTS)
    def test_format_seconds(self, seconds, out):
        """Test format_seconds with several tests."""
        assert utils.format_seconds(seconds) == out


class TestFormatTimedelta:

    """Tests for format_timedelta.

    Class attributes:
        TESTS: A list of (input, output) tuples.
    """

    TESTS = [
        (datetime.timedelta(seconds=-1), '-1s'),
        (datetime.timedelta(seconds=0), '0s'),
        (datetime.timedelta(seconds=59), '59s'),
        (datetime.timedelta(seconds=120), '2m'),
        (datetime.timedelta(seconds=60.4), '1m'),
        (datetime.timedelta(seconds=63), '1m 3s'),
        (datetime.timedelta(seconds=-64), '-1m 4s'),
        (datetime.timedelta(seconds=3599), '59m 59s'),
        (datetime.timedelta(seconds=3600), '1h'),
        (datetime.timedelta(seconds=3605), '1h 5s'),
        (datetime.timedelta(seconds=3723), '1h 2m 3s'),
        (datetime.timedelta(seconds=3780), '1h 3m'),
        (datetime.timedelta(seconds=36000), '10h'),
    ]

    @pytest.mark.parametrize('td, out', TESTS)
    def test_format_seconds(self, td, out):
        """Test format_seconds with several tests."""
        assert utils.format_timedelta(td) == out


class TestFormatSize:

    """Tests for format_size.

    Class attributes:
        TESTS: A list of (input, output) tuples.
    """

    TESTS = [
        (-1024, '-1.00k'),
        (-1, '-1.00'),
        (0, '0.00'),
        (1023, '1023.00'),
        (1024, '1.00k'),
        (1034.24, '1.01k'),
        (1024 * 1024 * 2, '2.00M'),
        (1024 ** 10, '1024.00Y'),
        (None, '?.??'),
    ]

    KILO_TESTS = [(999, '999.00'), (1000, '1.00k'), (1010, '1.01k')]

    @pytest.mark.parametrize('size, out', TESTS)
    def test_format_size(self, size, out):
        """Test format_size with several tests."""
        assert utils.format_size(size) == out

    @pytest.mark.parametrize('size, out', TESTS)
    def test_suffix(self, size, out):
        """Test the suffix option."""
        assert utils.format_size(size, suffix='B') == out + 'B'

    @pytest.mark.parametrize('size, out', KILO_TESTS)
    def test_base(self, size, out):
        """Test with an alternative base."""
        assert utils.format_size(size, base=1000) == out


class TestKeyToString:

    """Test key_to_string."""

    def test_unicode_garbage_keys(self):
        """Test a special key where QKeyEvent::toString works incorrectly."""
        assert utils.key_to_string(Qt.Key_Blue) == 'Blue'

    def test_backtab(self):
        """Test if backtab is normalized to tab correctly."""
        assert utils.key_to_string(Qt.Key_Backtab) == 'Tab'

    def test_escape(self):
        """Test if escape is normalized to escape correctly."""
        assert utils.key_to_string(Qt.Key_Escape) == 'Escape'

    def test_letter(self):
        """Test a simple letter key."""
        assert utils.key_to_string(Qt.Key_A) == 'A'

    def test_unicode(self):
        """Test a printable unicode key."""
        assert utils.key_to_string(Qt.Key_degree) == '°'

    def test_special(self):
        """Test a non-printable key handled by QKeyEvent::toString."""
        assert utils.key_to_string(Qt.Key_F1) == 'F1'


class TestKeyEventToString:

    """Test keyevent_to_string."""

    def test_only_control(self, fake_keyevent_factory):
        """Test keyeevent when only control is pressed."""
        evt = fake_keyevent_factory(key=Qt.Key_Control,
                                    modifiers=Qt.ControlModifier)
        assert utils.keyevent_to_string(evt) is None

    def test_only_hyper_l(self, fake_keyevent_factory):
        """Test keyeevent when only Hyper_L is pressed."""
        evt = fake_keyevent_factory(key=Qt.Key_Hyper_L,
                                    modifiers=Qt.MetaModifier)
        assert utils.keyevent_to_string(evt) is None

    def test_only_key(self, fake_keyevent_factory):
        """Test with a simple key pressed."""
        evt = fake_keyevent_factory(key=Qt.Key_A)
        assert utils.keyevent_to_string(evt) == 'A'

    def test_key_and_modifier(self, fake_keyevent_factory):
        """Test with key and modifier pressed."""
        evt = fake_keyevent_factory(key=Qt.Key_A, modifiers=Qt.ControlModifier)
        assert utils.keyevent_to_string(evt) == 'Ctrl+A'

    def test_key_and_modifiers(self, fake_keyevent_factory):
        """Test with key and multiple modifier pressed."""
        evt = fake_keyevent_factory(
            key=Qt.Key_A, modifiers=(Qt.ControlModifier | Qt.AltModifier |
                                     Qt.MetaModifier | Qt.ShiftModifier))
        if sys.platform == 'darwin':
            assert utils.keyevent_to_string(evt) == 'Ctrl+Alt+Shift+A'
        else:
            assert utils.keyevent_to_string(evt) == 'Ctrl+Alt+Meta+Shift+A'


class TestNormalize:

    """Test normalize_keystr."""

    STRINGS = (
        ('Control+x', 'ctrl+x'),
        ('Windows+x', 'meta+x'),
        ('Mod1+x', 'alt+x'),
        ('Mod4+x', 'meta+x'),
        ('Control--', 'ctrl+-'),
        ('Windows++', 'meta++'),
        ('ctrl-x', 'ctrl+x'),
        ('control+x', 'ctrl+x')
    )

    @pytest.mark.parametrize('orig, repl', STRINGS)
    def test_normalize(self, orig, repl):
        """Test normalize with some strings."""
        assert utils.normalize_keystr(orig) == repl


class TestIsEnum:

    """Test is_enum."""

    def test_enum(self):
        """Test is_enum with an enum."""
        e = enum.Enum('Foo', 'bar, baz')
        assert utils.is_enum(e)

    def test_class(self):
        """Test is_enum with a non-enum class."""
        class Test:

            """Test class for is_enum."""

            pass

        assert not utils.is_enum(Test)

    def test_object(self):
        """Test is_enum with a non-enum object."""
        assert not utils.is_enum(23)


class TestRaises:

    """Test raises."""

    def do_raise(self):
        """Helper function which raises an exception."""
        raise Exception

    def do_nothing(self):
        """Helper function which does nothing."""
        pass

    def test_raises_single_exc_true(self):
        """Test raises with a single exception which gets raised."""
        assert utils.raises(ValueError, int, 'a')

    def test_raises_single_exc_false(self):
        """Test raises with a single exception which does not get raised."""
        assert not utils.raises(ValueError, int, '1')

    def test_raises_multiple_exc_true(self):
        """Test raises with multiple exceptions which get raised."""
        assert utils.raises((ValueError, TypeError), int, 'a')
        assert utils.raises((ValueError, TypeError), int, None)

    def test_raises_multiple_exc_false(self):
        """Test raises with multiple exceptions which do not get raised."""
        assert not utils.raises((ValueError, TypeError), int, '1')

    def test_no_args_true(self):
        """Test with no args and an exception which gets raised."""
        assert utils.raises(Exception, self.do_raise)

    def test_no_args_false(self):
        """Test with no args and an exception which does not get raised."""
        assert not utils.raises(Exception, self.do_nothing)

    def test_unrelated_exception(self):
        """Test with an unrelated exception."""
        with pytest.raises(Exception):
            utils.raises(ValueError, self.do_raise)


class TestForceEncoding:

    """Test force_encoding."""

    TESTS = [
        ('hello world', 'ascii', 'hello world'),
        ('hellö wörld', 'utf-8', 'hellö wörld'),
        ('hellö wörld', 'ascii', 'hell? w?rld'),
    ]

    @pytest.mark.parametrize('inp, enc, expected', TESTS)
    def test_fitting_ascii(self, inp, enc, expected):
        """Test force_encoding will yield expected text."""
        assert utils.force_encoding(inp, enc) == expected


class TestNewestSlice:

    """Test newest_slice."""

    def test_count_minus_two(self):
        """Test with a count of -2."""
        with pytest.raises(ValueError):
            utils.newest_slice([], -2)

    def test_count_minus_one(self):
        """Test with a count of -1 (all elements)."""
        items = range(20)
        sliced = utils.newest_slice(items, -1)
        assert list(sliced) == list(items)

    def test_count_zero(self):
        """Test with a count of 0 (no elements)."""
        items = range(20)
        sliced = utils.newest_slice(items, 0)
        assert list(sliced) == []

    def test_count_much_smaller(self):
        """Test with a count which is much smaller than the iterable."""
        items = range(20)
        sliced = utils.newest_slice(items, 5)
        assert list(sliced) == [15, 16, 17, 18, 19]

    def test_count_smaller(self):
        """Test with a count which is exactly one smaller."""
        items = range(5)
        sliced = utils.newest_slice(items, 4)
        assert list(sliced) == [1, 2, 3, 4]

    def test_count_equal(self):
        """Test with a count which is just as large as the iterable."""
        items = range(5)
        sliced = utils.newest_slice(items, 5)
        assert list(sliced) == list(items)

    def test_count_bigger(self):
        """Test with a count which is one bigger than the iterable."""
        items = range(5)
        sliced = utils.newest_slice(items, 6)
        assert list(sliced) == list(items)

    def test_count_much_bigger(self):
        """Test with a count which is much bigger than the iterable."""
        items = range(5)
        sliced = utils.newest_slice(items, 50)
        assert list(sliced) == list(items)