Python cleanup_subscriptions Examples

Example #1

File: bigquery_test.py Project: xsm110/Beam15.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)

Example #2

 def _cleanup_pubsub(self):
     test_utils.cleanup_subscriptions(self.sub_client, [self.input_sub])
     test_utils.cleanup_topics(self.pub_client, [self.input_topic])

Example #3

 def tearDown(self):
   """Delete all created topics and subs."""
   test_utils.cleanup_subscriptions(
       self.sub_client, [self.input_sub, self.output_sub])
   test_utils.cleanup_topics(
       self.pub_client, [self.input_topic, self.output_topic])

Example #4

 def test_cleanup_topics(self):
   mock_topics = mock.MagicMock()
   mock_topics.exist.return_value = True
   utils.cleanup_subscriptions([mock_topics])
   self.assertTrue(mock_topics.delete.called)

Example #5

File: streaming_wordcount_it_test.py Project: AfterShip/aftership-beam

 def tearDown(self):
     test_utils.cleanup_subscriptions(self.sub_client,
                                      [self.input_sub, self.output_sub])
     test_utils.cleanup_topics(self.pub_client,
                               [self.input_topic, self.output_topic])

Example #6

 def tearDown(self):
     super(GroupByKeyStreamingTest, self).tearDown()
     test_utils.cleanup_subscriptions(self.sub_client,
                                      [self.input_sub, self.output_sub])
     test_utils.cleanup_topics(self.pub_client,
                               [self.input_topic, self.output_topic])

Example #7

File: streaming_wordcount_it_test.py Project: charlesccychen/incubator-beam

 def tearDown(self):
   test_utils.cleanup_subscriptions(self.sub_client,
                                    [self.input_sub, self.output_sub])
   test_utils.cleanup_topics(self.pub_client,
                             [self.input_topic, self.output_topic])

Example #8

File: leader_board_it_test.py Project: lukecwik/incubator-beam

 def _cleanup_pubsub(self):
   test_utils.cleanup_subscriptions([self.input_sub])
   test_utils.cleanup_topics([self.input_topic])

Example #9

 def test_cleanup_subscriptions(self):
     sub_client = mock.Mock()
     sub = mock.Mock()
     sub.name = 'test_sub'
     utils.cleanup_subscriptions(sub_client, [sub])
     sub_client.delete_subscription.assert_called_with(sub.name)

Example #10

File: streaming_wordcount_it_test.py Project: zahiralward/beam

 def _cleanup_pubsub(self):
   test_utils.cleanup_subscriptions([self.input_sub, self.output_sub])
   test_utils.cleanup_topics([self.input_topic, self.output_topic])

Example #11

File: test_utils_test.py Project: charlesccychen/incubator-beam

 def test_cleanup_subscriptions(self):
   sub_client = mock.Mock()
   sub = mock.Mock()
   sub.name = 'test_sub'
   utils.cleanup_subscriptions(sub_client, [sub])
   sub_client.delete_subscription.assert_called_with(sub.name)