Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit ca7a8cc

Browse files
author
Jim Switzer
committed
Testing: Add expires_in parameter to TokenGenerator Mock()
1 parent 7f127aa commit ca7a8cc

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

oauth2/test/test_grant.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1810,7 +1810,7 @@ def test_read_validate_params(self):
18101810
handler = RefreshTokenHandler(access_token_store=access_token_store_mock,
18111811
client_store=client_store_mock,
18121812
scope_handler=scope_handler_mock,
1813-
token_generator=Mock())
1813+
token_generator=Mock(expires_in=600))
18141814

18151815
handler.read_validate_params(request=request_mock)
18161816

@@ -1834,7 +1834,7 @@ def test_read_validate_params_no_client_id(self):
18341834
handler = RefreshTokenHandler(access_token_store=Mock(),
18351835
client_store=Mock(),
18361836
scope_handler=Mock(),
1837-
token_generator=Mock())
1837+
token_generator=Mock(expires_in=600))
18381838

18391839
with self.assertRaises(OAuthInvalidError) as expected:
18401840
handler.read_validate_params(request_mock)
@@ -1851,7 +1851,7 @@ def test_read_validate_params_no_client_secret(self):
18511851
handler = RefreshTokenHandler(access_token_store=Mock(),
18521852
client_store=Mock(),
18531853
scope_handler=Mock(),
1854-
token_generator=Mock())
1854+
token_generator=Mock(expires_in=600))
18551855

18561856
with self.assertRaises(OAuthInvalidError) as expected:
18571857
handler.read_validate_params(request_mock)
@@ -1868,7 +1868,7 @@ def test_read_validate_params_no_refresh_token(self):
18681868
handler = RefreshTokenHandler(access_token_store=Mock(),
18691869
client_store=Mock(),
18701870
scope_handler=Mock(),
1871-
token_generator=Mock())
1871+
token_generator=Mock(expires_in=600))
18721872

18731873
with self.assertRaises(OAuthInvalidError) as expected:
18741874
handler.read_validate_params(request_mock)
@@ -1888,7 +1888,7 @@ def test_read_validate_params_client_not_found(self):
18881888
handler = RefreshTokenHandler(access_token_store=Mock(),
18891889
client_store=client_store_mock,
18901890
scope_handler=Mock(),
1891-
token_generator=Mock())
1891+
token_generator=Mock(expires_in=600))
18921892

18931893
with self.assertRaises(OAuthInvalidError) as expected:
18941894
handler.read_validate_params(request_mock)
@@ -1915,7 +1915,7 @@ def test_read_validate_params_invalid_client_secret(self):
19151915
handler = RefreshTokenHandler(access_token_store=Mock(),
19161916
client_store=client_store_mock,
19171917
scope_handler=Mock(),
1918-
token_generator=Mock())
1918+
token_generator=Mock(expires_in=600))
19191919

19201920
with self.assertRaises(OAuthInvalidError) as expected:
19211921
handler.read_validate_params(request_mock)
@@ -1943,7 +1943,7 @@ def test_read_validate_params_invalid_refresh_token(self):
19431943
handler = RefreshTokenHandler(access_token_store=access_token_store_mock,
19441944
client_store=client_store_mock,
19451945
scope_handler=Mock(),
1946-
token_generator=Mock())
1946+
token_generator=Mock(expires_in=600))
19471947

19481948
with self.assertRaises(OAuthInvalidError) as expected:
19491949
handler.read_validate_params(request_mock)
@@ -1975,7 +1975,7 @@ def test_read_validate_params_expired_access_token(self):
19751975
handler = RefreshTokenHandler(access_token_store=access_token_store_mock,
19761976
client_store=client_store_mock,
19771977
scope_handler=Mock(),
1978-
token_generator=Mock())
1978+
token_generator=Mock(expires_in=600))
19791979

19801980
with self.assertRaises(OAuthInvalidError) as expected:
19811981
handler.read_validate_params(request_mock)

0 commit comments

Comments
 (0)