From b9c69004e453fa392dfba5debd87b17e83a10dcb Mon Sep 17 00:00:00 2001 From: Umair Khan Date: Fri, 16 Jun 2017 09:49:25 +0500 Subject: [PATCH] google_oauth2_test: Pass headers in all requests. 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. --- zerver/tests/test_auth_backends.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/zerver/tests/test_auth_backends.py b/zerver/tests/test_auth_backends.py index 3a9656dbbd..02a61f2087 100644 --- a/zerver/tests/test_auth_backends.py +++ b/zerver/tests/test_auth_backends.py @@ -766,7 +766,7 @@ class GoogleOAuthTest(ZulipTestCase): return result # 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) if 'google' not in result.url: return result @@ -779,7 +779,7 @@ class GoogleOAuthTest(ZulipTestCase): with mock.patch("requests.post", return_value=token_response), ( mock.patch("requests.get", return_value=account_response)): result = self.client_get("/accounts/login/google/done/", - dict(state=csrf_state)) + dict(state=csrf_state), **headers) return result class GoogleSubdomainLoginTest(GoogleOAuthTest):