def test_client_initializes_with_impersonated_service_account(
self, mock_impersonated_account):
mock_impersonated_account.return_value = self.mock_credentials
bigquery_utils.BigQueryUtils(
project_id=self.project_id,
service_account_name=self.service_account_name)
mock_impersonated_account.assert_called_once_with(self.service_account_name)
self.mock_client.assert_called_with(
project=self.project_id, credentials=self.mock_credentials)
def setUp(self):
super().setUp()
self.addCleanup(mock.patch.stopall)
self.project_id = 'project-id'
# Mock for google.cloud.bigquery.Client object
self.mock_client = mock.patch.object(
bigquery, 'Client', autospec=True).start()
self.mock_get_credentials = mock.patch.object(
cloud_auth, 'get_default_credentials', autospec=True).start()
self.mock_credentials = mock.Mock(credentials.Credentials, autospec=True)
self.mock_get_credentials.return_value = self.mock_credentials
self.service_account_name = (
'*****@*****.**')
self.bigquery_client = bigquery_utils.BigQueryUtils(self.project_id)