1
0
mirror of https://github.com/nvbn/thefuck.git synced 2025-03-20 01:28:56 +00:00
thefuck/tests/rules/test_git_push.py
2018-01-02 23:17:19 +00:00

55 lines
2.0 KiB
Python

import pytest
from thefuck.rules.git_push import match, get_new_command
from thefuck.types import Command
@pytest.fixture
def output(branch):
return '''fatal: The current branch {} has no upstream branch.
To push the current branch and set the remote as upstream, use
git push --set-upstream origin {}
'''.format(branch, branch)
@pytest.mark.parametrize('command', [
Command('git push', output('master')),
Command('git push master', output('master'))])
def test_match(command):
assert match(command)
@pytest.mark.parametrize('command', [
Command('git push master', ''),
Command('ls', output('master'))])
def test_not_match(command):
assert not match(command)
@pytest.mark.parametrize('command, new_command', [
(Command('git push master', output('master')),
'git push --set-upstream origin master'),
(Command('git push master', output('master')),
'git push --set-upstream origin master'),
(Command('git push master', output('master')),
'git push --set-upstream origin master'),
(Command('git push -u', output('master')),
'git push --set-upstream origin master'),
(Command('git push -u origin', output('master')),
'git push --set-upstream origin master'),
(Command('git push origin', output('master')),
'git push --set-upstream origin master'),
(Command('git push --set-upstream origin', output('master')),
'git push --set-upstream origin master'),
(Command('git push --quiet', output('master')),
'git push --set-upstream origin master --quiet'),
(Command('git push --quiet origin', output('master')),
'git push --set-upstream origin master --quiet'),
(Command('git -c test=test push --quiet origin', output('master')),
'git -c test=test push --set-upstream origin master --quiet'),
(Command('git push', output("test's")),
"git push --set-upstream origin test\\'s")])
def test_get_new_command(command, new_command):
assert get_new_command(command) == new_command