paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-06 15:03:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

google_oauth2_test: Pass headers in all requests.

Browse Source 
Currently we only pass headers in the first client_get call but
sometimes the effective request which reaches the view is through
a later call to the client_get in this function. Due to which
headers are not passed.
This commit is contained in:
Umair Khan
2017-06-16 09:49:25 +05:00
committed by Tim Abbott
parent 1ce8258f2e
commit b9c69004e4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
zerver/tests/test_auth_backends.py
View File

@@ -766,7 +766,7 @@ class GoogleOAuthTest(ZulipTestCase):
return result return result
# Now do the /google/send/ request # Now do the /google/send/ request
result = self.client_get(result.url) result = self.client_get(result.url, **headers)
self.assertEqual(result.status_code, 302) self.assertEqual(result.status_code, 302)
if 'google' not in result.url: if 'google' not in result.url:
return result return result
@@ -779,7 +779,7 @@ class GoogleOAuthTest(ZulipTestCase):
with mock.patch("requests.post", return_value=token_response), ( with mock.patch("requests.post", return_value=token_response), (
mock.patch("requests.get", return_value=account_response)): mock.patch("requests.get", return_value=account_response)):
result = self.client_get("/accounts/login/google/done/", result = self.client_get("/accounts/login/google/done/",
dict(state=csrf_state)) dict(state=csrf_state), **headers)
return result return result
class GoogleSubdomainLoginTest(GoogleOAuthTest): class GoogleSubdomainLoginTest(GoogleOAuthTest):