Skip to content

Commit 35dabdb

Browse files
committed
Fix lint.
1 parent 109d2a3 commit 35dabdb

File tree

2 files changed

+14
-19
lines changed

2 files changed

+14
-19
lines changed

google_auth_oauthlib/flow.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,8 @@ class Flow(object):
9595

9696
def __init__(
9797
self, oauth2session, client_type, client_config,
98-
redirect_uri=None, code_verifier=None, autogenerate_code_verifier=False):
98+
redirect_uri=None, code_verifier=None,
99+
autogenerate_code_verifier=False):
99100
"""
100101
Args:
101102
oauth2session (requests_oauthlib.OAuth2Session):
@@ -159,15 +160,23 @@ def from_client_config(cls, client_config, scopes, **kwargs):
159160

160161
# these args cannot be passed to requests_oauthlib.OAuth2Session
161162
code_verifier = kwargs.pop('code_verifier', None)
162-
autogenerate_code_verifier = kwargs.pop('autogenerate_code_verifier', None)
163+
autogenerate_code_verifier = kwargs.pop(
164+
'autogenerate_code_verifier', None)
163165

164166
session, client_config = (
165167
google_auth_oauthlib.helpers.session_from_client_config(
166168
client_config, scopes, **kwargs))
167169

168170
redirect_uri = kwargs.get('redirect_uri', None)
169171

170-
return cls(session, client_type, client_config, redirect_uri, code_verifier, autogenerate_code_verifier)
172+
return cls(
173+
session,
174+
client_type,
175+
client_config,
176+
redirect_uri,
177+
code_verifier,
178+
autogenerate_code_verifier
179+
)
171180

172181
@classmethod
173182
def from_client_secrets_file(cls, client_secrets_file, scopes, **kwargs):

tests/test_flow.py

Lines changed: 2 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,8 @@ def test_authorization_url_access_type(self, instance):
144144
def test_authorization_url_generated_verifier(self):
145145
scope = 'scope_one'
146146
instance = flow.Flow.from_client_config(
147-
CLIENT_SECRETS_INFO, scopes=[scope], autogenerate_code_verifier=True)
147+
CLIENT_SECRETS_INFO, scopes=[scope],
148+
autogenerate_code_verifier=True)
148149
authorization_url_path = mock.patch.object(
149150
instance.oauth2session, 'authorization_url',
150151
wraps=instance.oauth2session.authorization_url)
@@ -237,21 +238,6 @@ def set_token(*args, **kwargs):
237238
with fetch_token_patch as fetch_token_mock:
238239
yield fetch_token_mock
239240

240-
@mock.patch('google_auth_oauthlib.flow.input', autospec=True)
241-
def test_run_console(self, input_mock, instance, mock_fetch_token):
242-
input_mock.return_value = mock.sentinel.code
243-
credentials = instance.run_console()
244-
245-
assert credentials.token == mock.sentinel.access_token
246-
assert credentials._refresh_token == mock.sentinel.refresh_token
247-
assert credentials.id_token == mock.sentinel.id_token
248-
249-
mock_fetch_token.assert_called_with(
250-
CLIENT_SECRETS_INFO['web']['token_uri'],
251-
client_secret=CLIENT_SECRETS_INFO['web']['client_secret'],
252-
code=mock.sentinel.code,
253-
code_verifier=None)
254-
255241
@mock.patch('google_auth_oauthlib.flow.input', autospec=True)
256242
def test_run_console(self, input_mock, instance, mock_fetch_token):
257243
input_mock.return_value = mock.sentinel.code

0 commit comments

Comments
 (0)