Ejemplo n.º 1
0
 def test_refresh_token(self, metadata):
     credentials = AppAssertionCredentials()
     self.assertIsNone(credentials.access_token)
     credentials.get_access_token()
     self.assertEqual(credentials.access_token, 'A')
     self.assertTrue(credentials.access_token_expired)
     credentials.get_access_token()
     self.assertEqual(credentials.access_token, 'B')
     self.assertFalse(credentials.access_token_expired)
Ejemplo n.º 2
0
 def test_refresh_token(self, get_info, get_token):
     http_request = mock.MagicMock()
     http_mock = mock.MagicMock(request=http_request)
     credentials = AppAssertionCredentials()
     credentials.invalid = False
     credentials.service_account_email = '*****@*****.**'
     self.assertIsNone(credentials.access_token)
     credentials.get_access_token(http=http_mock)
     self.assertEqual(credentials.access_token, 'A')
     self.assertTrue(credentials.access_token_expired)
     get_token.assert_called_with(http_request, service_account='*****@*****.**')
     credentials.get_access_token(http=http_mock)
     self.assertEqual(credentials.access_token, 'B')
     self.assertFalse(credentials.access_token_expired)
     get_token.assert_called_with(http_request, service_account='*****@*****.**')
     get_info.assert_not_called()
Ejemplo n.º 3
0
    def test_refresh_token(self, get_metadata):
        credentials = AppAssertionCredentials()
        self.assertIsNone(credentials.access_token)

        with mock.patch('oauth2client.contrib.gce._NOW',
                        side_effect=[datetime.datetime.min,
                                     datetime.datetime.max]):
            credentials.get_access_token()
            self.assertTrue(credentials.access_token_expired)

        force_refresh_time = datetime.datetime.max - datetime.timedelta(
            seconds=get_metadata.return_value['expires_in'])

        with mock.patch('oauth2client.contrib.gce._NOW',
                        side_effect=[force_refresh_time,
                                     datetime.datetime.min,
                                     datetime.datetime.min]):
            credentials.get_access_token()
            self.assertFalse(credentials.access_token_expired)
Ejemplo n.º 4
0
    def test_get_access_token(self):
        http = mock.MagicMock()
        http.request = mock.MagicMock(
            return_value=(mock.Mock(status=http_client.OK),
                          '{"accessToken": "this-is-a-token"}'))

        credentials = AppAssertionCredentials(['dummy_scope'])
        token = credentials.get_access_token(http=http)
        self.assertEqual('this-is-a-token', token.access_token)
        self.assertEqual(None, token.expires_in)

        http.request.assert_called_once_with(
            'http://metadata.google.internal/0.1/meta-data/service-accounts/'
            'default/acquire?scope=dummy_scope')
Ejemplo n.º 5
0
    def test_get_access_token(self):
        http = mock.MagicMock()
        http.request = mock.MagicMock(return_value=(mock.Mock(
            status=http_client.OK), '{"access_token": "this-is-a-token"}'))

        credentials = AppAssertionCredentials()
        token = credentials.get_access_token(http=http)
        self.assertEqual('this-is-a-token', token.access_token)
        self.assertEqual(None, token.expires_in)

        http.request.assert_called_once_with(
            'http://metadata.google.internal/computeMetadata/v1/instance/'
            'service-accounts/default/token',
            headers={'Metadata-Flavor': 'Google'})
Ejemplo n.º 6
0
    def test_get_access_token(self):
        http = mock.MagicMock()
        http.request = mock.MagicMock(
            return_value=(mock.Mock(status=200),
                          '{"accessToken": "this-is-a-token"}'))

        credentials = AppAssertionCredentials(['dummy_scope'])
        token = credentials.get_access_token(http=http)
        self.assertEqual('this-is-a-token', token.access_token)
        self.assertEqual(None, token.expires_in)

        http.request.assert_called_once_with(
            'http://metadata.google.internal/0.1/meta-data/service-accounts/'
            'default/acquire?scope=dummy_scope')
Ejemplo n.º 7
0
    def test_get_access_token(self):
        http = mock.MagicMock()
        http.request = mock.MagicMock(
            return_value=(mock.Mock(status=http_client.OK),
                          '{"accessToken": "this-is-a-token"}'))

        credentials = AppAssertionCredentials(['dummy_scope'])
        token = credentials.get_access_token(http=http)
        self.assertEqual('this-is-a-token', token.access_token)
        self.assertEqual(None, token.expires_in)

        http.request.assert_called_once_with(
            'http://metadata.google.internal/computeMetadata/v1/instance/'
            'service-accounts/default/acquire?scope=dummy_scope',
            headers={'Metadata-Flavor': 'Google'})