mirror of
https://github.com/nvbn/thefuck.git
synced 2025-01-18 12:06:04 +00:00
74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
import pytest
|
|
from io import BytesIO
|
|
from thefuck.rules.git_checkout import match, get_branches, get_new_command
|
|
from thefuck.types import Command
|
|
|
|
|
|
def did_not_match(target, did_you_forget=False):
|
|
error = ("error: pathspec '{}' did not match any "
|
|
"file(s) known to git.".format(target))
|
|
if did_you_forget:
|
|
error = ("{}\nDid you forget to 'git add'?'".format(error))
|
|
return error
|
|
|
|
|
|
@pytest.fixture
|
|
def git_branch(mocker, branches):
|
|
mock = mocker.patch('subprocess.Popen')
|
|
mock.return_value.stdout = BytesIO(branches)
|
|
return mock
|
|
|
|
|
|
@pytest.mark.parametrize('command', [
|
|
Command('git checkout unknown', did_not_match('unknown')),
|
|
Command('git commit unknown', did_not_match('unknown'))])
|
|
def test_match(command):
|
|
assert match(command)
|
|
|
|
|
|
@pytest.mark.parametrize('command', [
|
|
Command('git submodule update unknown',
|
|
did_not_match('unknown', True)),
|
|
Command('git checkout known', ''),
|
|
Command('git commit known', '')])
|
|
def test_not_match(command):
|
|
assert not match(command)
|
|
|
|
|
|
@pytest.mark.parametrize('branches, branch_list', [
|
|
(b'', []),
|
|
(b'* master', ['master']),
|
|
(b' remotes/origin/master', ['master']),
|
|
(b' remotes/origin/test/1', ['test/1']),
|
|
(b' remotes/origin/test/1/2/3', ['test/1/2/3']),
|
|
(b' test/1', ['test/1']),
|
|
(b' test/1/2/3', ['test/1/2/3']),
|
|
(b' remotes/origin/HEAD -> origin/master', []),
|
|
(b' just-another-branch', ['just-another-branch']),
|
|
(b'* master\n just-another-branch', ['master', 'just-another-branch']),
|
|
(b'* master\n remotes/origin/master\n just-another-branch',
|
|
['master', 'master', 'just-another-branch'])])
|
|
def test_get_branches(branches, branch_list, git_branch):
|
|
git_branch(branches)
|
|
assert list(get_branches()) == branch_list
|
|
|
|
|
|
@pytest.mark.parametrize('branches, command, new_command', [
|
|
(b'',
|
|
Command('git checkout unknown', did_not_match('unknown')),
|
|
['git checkout -b unknown']),
|
|
(b'',
|
|
Command('git commit unknown', did_not_match('unknown')),
|
|
['git branch unknown && git commit unknown']),
|
|
(b' test-random-branch-123',
|
|
Command('git checkout tst-rdm-brnch-123',
|
|
did_not_match('tst-rdm-brnch-123')),
|
|
['git checkout test-random-branch-123', 'git checkout -b tst-rdm-brnch-123']),
|
|
(b' test-random-branch-123',
|
|
Command('git commit tst-rdm-brnch-123',
|
|
did_not_match('tst-rdm-brnch-123')),
|
|
['git commit test-random-branch-123'])])
|
|
def test_get_new_command(branches, command, new_command, git_branch):
|
|
git_branch(branches)
|
|
assert get_new_command(command) == new_command
|