mirror of
				https://github.com/nvbn/thefuck.git
				synced 2025-10-30 22:54:14 +00:00 
			
		
		
		
	#87 Add ability to fix branch names in git_checkout rule
				
					
				
			This commit is contained in:
		| @@ -12,6 +12,11 @@ def did_not_match(target, did_you_forget=False): | ||||
|     return error | ||||
|  | ||||
|  | ||||
| @pytest.fixture | ||||
| def get_branches(mocker): | ||||
|     return mocker.patch('thefuck.rules.git_checkout') | ||||
|  | ||||
|  | ||||
| @pytest.mark.parametrize('command', [ | ||||
|     Command(script='git checkout unknown', stderr=did_not_match('unknown')), | ||||
|     Command(script='git commit unknown', stderr=did_not_match('unknown'))]) | ||||
| @@ -28,10 +33,19 @@ def test_not_match(command): | ||||
|     assert not match(command, None) | ||||
|  | ||||
|  | ||||
| @pytest.mark.parametrize('command, new_command', [ | ||||
|     (Command(script='git checkout unknown', stderr=did_not_match('unknown')), | ||||
| @pytest.mark.parametrize('branches, command, new_command', [ | ||||
|     ([], | ||||
|      Command(script='git checkout unknown', stderr=did_not_match('unknown')), | ||||
|      'git branch unknown && git checkout unknown'), | ||||
|     (Command('git commit unknown', stderr=did_not_match('unknown')), | ||||
|      'git branch unknown && git commit unknown')]) | ||||
| def test_get_new_command(command, new_command): | ||||
|     ([], | ||||
|      Command('git commit unknown', stderr=did_not_match('unknown')), | ||||
|      'git branch unknown && git commit unknown'), | ||||
|     (['master'], | ||||
|      Command(script='git checkout amster', stderr=did_not_match('amster')), | ||||
|      'git checkout master'), | ||||
|     (['master'], | ||||
|      Command(script='git commit amster', stderr=did_not_match('amster')), | ||||
|      'git commit master')]) | ||||
| def test_get_new_command(branches, command, new_command, get_branches): | ||||
|     get_branches.return_value = branches | ||||
|     assert get_new_command(command, None) == new_command | ||||
|   | ||||
		Reference in New Issue
	
	Block a user