def test_google_call_credentials_error(self):
callback_event = threading.Event()
def mock_callback(metadata, error):
self.assertIsNotNone(error)
callback_event.set()
call_creds = _auth.GoogleCallCredentials(MockExceptionGoogleCreds())
call_creds(None, mock_callback)
self.assertTrue(callback_event.wait(1.0))
def google_call_credentials(credentials):
"""Construct CallCredentials from GoogleCredentials.
Args:
credentials: A GoogleCredentials object from the oauth2client library.
Returns:
A CallCredentials object for use in a GRPCCallOptions object.
"""
return metadata_call_credentials(_auth.GoogleCallCredentials(credentials))
def test_google_call_credentials_success(self):
callback_event = threading.Event()
def mock_callback(metadata, error):
self.assertEqual(metadata, (('authorization', 'Bearer token'),))
self.assertIsNone(error)
callback_event.set()
call_creds = _auth.GoogleCallCredentials(MockGoogleCreds())
call_creds(None, mock_callback)
self.assertTrue(callback_event.wait(1.0))
