Fix tests

sigmavirus24 · sigmavirus24 · commit e511c07e08f9 · 2014-01-26T07:55:22.000-06:00
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -18,19 +18,19 @@ def test_authorize(self):
         self.gh.authorize.assert_called_with(*args)
 
     def test_login(self):
-        args = ('login', 'password', None)
+        args = ('login', 'password', None, None)
         with patch.object(github3.api.GitHub, 'login') as login:
             g = github3.login(*args)
             assert isinstance(g, github3.github.GitHub)
             assert not isinstance(g, github3.github.GitHubEnterprise)
             login.assert_called_with(*args)
 
     def test_enterprise_login(self):
-        args = ('login', 'password', None, 'http://ghe.invalid/')
+        args = ('login', 'password', None, 'http://ghe.invalid/', None)
         with patch.object(github3.api.GitHubEnterprise, 'login') as login:
             g = github3.login(*args)
             assert isinstance(g, github3.github.GitHubEnterprise)
-            login.assert_called_with(*args[:3])
+            login.assert_called_with('login', 'password', None, None)
 
     def test_gist(self):
         args = (123,)
