Initial commit

tests/rules/test_git_push.py

@@ -0,0 +1,24 @@
+import pytest
+from thefuck.main import Command
+from thefuck.rules.git_push import match, get_new_command
+
+
+@pytest.fixture
+def stderr():
+    return '''fatal: The current branch master has no upstream branch.
+To push the current branch and set the remote as upstream, use
+
+    git push --set-upstream origin master
+
+'''
+
+
+def test_match(stderr):
+    assert match(Command('git push master', '', stderr))
+    assert not match(Command('git push master', '', ''))
+    assert not match(Command('ls', '', stderr))
+
+
+def test_get_new_command(stderr):
+    assert get_new_command(Command('', '', stderr))\
+        == "git push --set-upstream origin master"

tests/rules/test_sudo.py

@@ -0,0 +1,12 @@
+from thefuck.main import Command
+from thefuck.rules.sudo import match, get_new_command
+
+
+def test_match():
+    assert match(Command('', '', 'Permission denied'))
+    assert match(Command('', '', 'permission denied'))
+    assert not match(Command('', '', ''))
+
+
+def test_get_new_command():
+    assert get_new_command(Command('ls', '', '')) == 'sudo ls'