From 934eeaf4fcee84f22c4f58bdebc1d99ca5e7a31d Mon Sep 17 00:00:00 2001
From: Joseph Frazier <1212jtraceur@gmail.com>
Date: Fri, 30 Sep 2016 16:11:46 -0400
Subject: [PATCH] Test that `git push -u origin` still works

This was broken by https://github.com/nvbn/thefuck/pull/538
---
 tests/rules/test_git_push.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/rules/test_git_push.py b/tests/rules/test_git_push.py
index 7531472d..d6654964 100644
--- a/tests/rules/test_git_push.py
+++ b/tests/rules/test_git_push.py
@@ -23,5 +23,9 @@ def test_match(stderr):
 def test_get_new_command(stderr):
     assert get_new_command(Command('git push', stderr=stderr))\
         == "git push --set-upstream origin master"
+    assert get_new_command(Command('git push -u origin', stderr=stderr))\
+        == "git push --set-upstream origin master"
+    assert get_new_command(Command('git push --set-upstream origin', stderr=stderr))\
+        == "git push --set-upstream origin master"
     assert get_new_command(Command('git push --quiet', stderr=stderr))\
         == "git push --set-upstream origin master --quiet"