def testGetForAccountException(self):
self.StartObjectPatch(
store,
'Load',
side_effect=google_auth_exceptions.GoogleAuthError())
with self.assertRaises(refresh_token.LoadingCredentialsError):
refresh_token.GetForAccount('my-account')
def testActivateAndRetrieve(self):
self.StartObjectPatch(store, 'Refresh', autospec=True)
refresh_token.ActivateCredentials('my-account', 'my-refresh-token')
self.assertEqual('my-account', properties.VALUES.core.account.Get())
self.assertEqual('my-refresh-token',
refresh_token.GetForAccount('my-account'))
def testActivate(self):
self.StartObjectPatch(store, 'Refresh', autospec=True)
account = '*****@*****.**'
token = 'asdf'
self.Run(
'auth activate-refresh-token {account} {token}'.format(
account=account, token=token))
self.assertEqual(token, refresh_token.GetForAccount(account))
self.assertEqual('*****@*****.**', properties.VALUES.core.account.Get())
self.assertEqual('junkproj', properties.VALUES.core.project.Get())
self.AssertOutputEquals('')
self.AssertErrEquals(
'Activated refresh token credentials: [[email protected]]\n')
def testActivateStdin(self):
self.StartObjectPatch(store, 'Refresh', autospec=True)
account = '*****@*****.**'
token = 'asdf'
self.WriteInput(token)
self.Run(
'--project=myproj auth activate-refresh-token {account}'.format(
account=account))
self.assertEqual(token, refresh_token.GetForAccount(account))
self.assertEqual('*****@*****.**', properties.VALUES.core.account.Get())
self.assertEqual('myproj', properties.VALUES.core.project.Get())
self.AssertOutputEquals('')
self.AssertErrEquals(
'{"ux": "PROMPT_RESPONSE", "message": "Refresh token: "}'
'Activated refresh token credentials: [[email protected]]\n')
def Run(self, args):
"""Run the helper command."""
return {'refresh_token': refresh_token.GetForAccount(args.account)}