mirror of
https://github.com/nvbn/thefuck.git
synced 2024-10-05 18:31:10 +01:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
import pytest
|
|
from thefuck.rules.git_main_master import match, get_new_command
|
|
from thefuck.types import Command
|
|
|
|
|
|
@pytest.fixture
|
|
def output(branch_name):
|
|
if not branch_name:
|
|
return ""
|
|
output_str = u"error: pathspec '{}' did not match any file(s) known to git"
|
|
return output_str.format(branch_name)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"script, branch_name",
|
|
[
|
|
("git checkout main", "main"),
|
|
("git checkout master", "master"),
|
|
("git show main", "main"),
|
|
],
|
|
)
|
|
def test_match(script, branch_name, output):
|
|
assert match(Command(script, output))
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"script, branch_name",
|
|
[
|
|
("git checkout master", ""),
|
|
("git checkout main", ""),
|
|
("git checkout wibble", "wibble"),
|
|
],
|
|
)
|
|
def test_not_match(script, branch_name, output):
|
|
assert not match(Command(script, output))
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"script, branch_name, new_command",
|
|
[
|
|
("git checkout main", "main", "git checkout master"),
|
|
("git checkout master", "master", "git checkout main"),
|
|
("git checkout wibble", "wibble", "git checkout wibble"),
|
|
],
|
|
)
|
|
def test_get_new_command(script, branch_name, new_command, output):
|
|
assert get_new_command(Command(script, output)) == new_command
|