def testNoProject(self):
instance = publisher.BigQueryPublisher(self.table)
instance.PublishSamples(self.samples)
self.mock_vm_util.IssueRetryableCommand.assert_called_once_with([
'bq', 'load', '--autodetect',
'--source_format=NEWLINE_DELIMITED_JSON', self.table, mock.ANY
])
def testServiceAccountFlags_BothSpecified(self):
instance = publisher.BigQueryPublisher(
self.table,
service_account=mock.MagicMock(),
service_account_private_key_file=mock.MagicMock())
instance.PublishSamples(self.samples) # No error
self.mock_vm_util.IssueRetryableCommand.assert_called_once_with(mock.ANY)
def testNoSamples(self):
instance = publisher.BigQueryPublisher(self.table)
instance.PublishSamples([])
self.assertEqual([],
self.mock_vm_util.IssueRetryableCommand.mock_calls)
def testApplicationCredentialsFlag_WorkingNormal(self):
instance = publisher.BigQueryPublisher(
self.table, application_default_credential_file=mock.MagicMock())
instance.PublishSamples(self.samples) # No error
self.mock_vm_util.IssueRetryableCommand.assert_called_once_with(
mock.ANY)