def tearDown(self): # Tear down PubSub test_utils.cleanup_topics(self.pub_client, [self.input_topic]) test_utils.cleanup_subscriptions(self.sub_client, [self.input_sub]) # Tear down BigQuery utils.delete_bq_dataset(self.project, self.dataset_ref)
def test_delete_bq_dataset(self, mock_client): utils.delete_bq_dataset('project', 'dataset_ref') mock_client.return_value.delete_dataset.assert_called_with( 'dataset_ref', delete_contents=mock.ANY)