import pytest
from theheck.rules.git_rm_local_modifications import match, get_new_command
from theheck.types import Command


@pytest.fixture
def output(target):
    return ('error: the following file has local modifications:\n    {}\n(use '
            '--cached to keep the file, or -f to force removal)').format(target)


@pytest.mark.parametrize('script, target', [
    ('git rm foo', 'foo'),
    ('git rm foo bar', 'bar')])
def test_match(output, script, target):
    assert match(Command(script, output))


@pytest.mark.parametrize('script', ['git rm foo', 'git rm foo bar', 'git rm'])
def test_not_match(script):
    assert not match(Command(script, ''))


@pytest.mark.parametrize('script, target, new_command', [
    ('git rm foo', 'foo', ['git rm --cached foo', 'git rm -f foo']),
    ('git rm foo bar', 'bar', ['git rm --cached foo bar', 'git rm -f foo bar'])])
def test_get_new_command(output, script, target, new_command):
    assert get_new_command(Command(script, output)) == new_command

[((326, 424), 'pytest.mark.parametrize', 'pytest.mark.parametrize', (['"""script, target"""', "[('git rm foo', 'foo'), ('git rm foo bar', 'bar')]"], {}), "('script, target', [('git rm foo', 'foo'), (\n    'git rm foo bar', 'bar')])\n", (349, 424), False, 'import pytest\n'), ((514, 591), 'pytest.mark.parametrize', 'pytest.mark.parametrize', (['"""script"""', "['git rm foo', 'git rm foo bar', 'git rm']"], {}), "('script', ['git rm foo', 'git rm foo bar', 'git rm'])\n", (537, 591), False, 'import pytest\n'), ((665, 871), 'pytest.mark.parametrize', 'pytest.mark.parametrize', (['"""script, target, new_command"""', "[('git rm foo', 'foo', ['git rm --cached foo', 'git rm -f foo']), (\n    'git rm foo bar', 'bar', ['git rm --cached foo bar', 'git rm -f foo bar'])]"], {}), "('script, target, new_command', [('git rm foo',\n    'foo', ['git rm --cached foo', 'git rm -f foo']), ('git rm foo bar',\n    'bar', ['git rm --cached foo bar', 'git rm -f foo bar'])])\n", (688, 871), False, 'import pytest\n'), ((486, 509), 'theheck.types.Command', 'Command', (['script', 'output'], {}), '(script, output)\n', (493, 509), False, 'from theheck.types import Command\n'), ((641, 660), 'theheck.types.Command', 'Command', (['script', '""""""'], {}), "(script, '')\n", (648, 660), False, 'from theheck.types import Command\n'), ((963, 986), 'theheck.types.Command', 'Command', (['script', 'output'], {}), '(script, output)\n', (970, 986), False, 'from theheck.types import Command\n')]