def testTokenExpiryFromJson(self):
cred = Oauth2WithReauthCredentials.from_json(
json.dumps({
'access_token': 'access_token',
'client_id': 'client_id',
'client_secret': 'client_secret',
'refresh_token': 'refresh_token',
'token_expiry': 'foo',
'token_uri': 'token_uri',
'user_agent': 'user_agent',
'invalid': False
}))
self.assertEqual(None, cred.token_expiry)
cred = Oauth2WithReauthCredentials.from_json(
json.dumps({
'access_token': 'access_token',
'client_id': 'client_id',
'client_secret': 'client_secret',
'refresh_token': 'refresh_token',
'token_expiry': '2018-03-02T21:26:13Z',
'token_uri': 'token_uri',
'user_agent': 'user_agent',
'invalid': False
}))
self.assertEqual(datetime.datetime(2018, 3, 2, 21, 26, 13),
cred.token_expiry)
def _Load(self):
"""Loads credentials from given application default credential file."""
with open(self._credential_file) as file_obj:
credentials = json.load(file_obj)
client_scope = _GetClientScopeFromFlags()
if credentials['type'] == oauth2client_4_0.client.AUTHORIZED_USER:
return Oauth2WithReauthCredentials(
access_token=None,
client_id=credentials['client_id'],
client_secret=credentials['client_secret'],
refresh_token=credentials['refresh_token'],
token_expiry=None,
token_uri=oauth2client_4_0.GOOGLE_TOKEN_URI,
user_agent=_CLIENT_USER_AGENT,
scopes=client_scope)
else: # Service account
credentials['type'] = oauth2client_4_0.client.SERVICE_ACCOUNT
service_account_credentials = (
oauth2client_4_0.service_account.ServiceAccountCredentials.
from_json_keyfile_dict(
keyfile_dict=credentials,
scopes=client_scope,
token_uri=oauth2client_4_0.GOOGLE_TOKEN_URI,
revoke_uri=oauth2client_4_0.GOOGLE_REVOKE_URI))
service_account_credentials._user_agent = _CLIENT_USER_AGENT # pylint: disable=protected-access
return service_account_credentials
def _get_creds(self):
return Oauth2WithReauthCredentials(access_token='old_token',
client_id='id',
client_secret='secret',
refresh_token='old_refresh_token',
token_expiry=datetime.datetime(
2018, 3, 2, 21, 26, 13),
token_uri='token_uri',
user_agent='user_agent')
def testFromOAuth2Credentials(self):
orig = client.OAuth2Credentials(access_token='at',
client_id='ci',
client_secret='cs',
refresh_token='rt',
token_expiry='te',
token_uri='tu',
user_agent='ua')
cred = Oauth2WithReauthCredentials.from_OAuth2Credentials(orig)
self.assertEqual('Oauth2WithReauthCredentials',
cred.__class__.__name__)
self.assertEqual('ci', cred.client_id)
self.assertEqual('cs', cred.client_secret)
def _GetReauthCredentials(oauth2_creds):
reauth_creds = Oauth2WithReauthCredentials.from_OAuth2Credentials(
oauth2_creds)
reauth_creds.store = oauth2_creds.store
return reauth_creds