from thefuck.types import RulesNamesList, Settings, \
    SortedCorrectedCommandsSequence
from tests.utils import Rule, CorrectedCommand


def test_rules_names_list():
    assert RulesNamesList(['bash', 'lisp']) == ['bash', 'lisp']
    assert RulesNamesList(['bash', 'lisp']) == RulesNamesList(['bash', 'lisp'])
    assert Rule('lisp') in RulesNamesList(['lisp'])
    assert Rule('bash') not in RulesNamesList(['lisp'])


def test_update_settings():
    settings = Settings({'key': 'val'})
    new_settings = settings.update(key='new-val', unset='unset-value')
    assert new_settings.key == 'val'
    assert new_settings.unset == 'unset-value'
    assert settings.key == 'val'


class TestSortedCorrectedCommandsSequence(object):
    def test_realises_generator_only_on_demand(self, settings):
        should_realise = False

        def gen():
            nonlocal should_realise
            yield CorrectedCommand('git commit')
            yield CorrectedCommand('git branch', priority=200)
            assert should_realise
            yield CorrectedCommand('git checkout', priority=100)

        commands = SortedCorrectedCommandsSequence(gen(), settings)
        assert commands[0] == CorrectedCommand('git commit')
        should_realise = True
        assert commands[1] == CorrectedCommand('git checkout', priority=100)
        assert commands[2] == CorrectedCommand('git branch', priority=200)

    def test_remove_duplicates(self, settings):
        side_effect = lambda *_: None
        seq = SortedCorrectedCommandsSequence(
            iter([CorrectedCommand('ls', priority=100),
                  CorrectedCommand('ls', priority=200),
                  CorrectedCommand('ls', side_effect, 300)]),
            settings)
        assert set(seq) == {CorrectedCommand('ls', priority=100),
                            CorrectedCommand('ls', side_effect, 300)}