Exemplo n.º 1
0
 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)
Exemplo n.º 2
0
 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)