class CoreTest(TestCase): def setUp(self): self.application = Application() self.pubsub = BasePubSub(self.application) self.project_id = 'test' self.namespace = 'test' self.channel = 'test' def test_get_subscription_key(self): subscription_key = self.pubsub.get_subscription_key( self.project_id, self.namespace, self.channel) self.assertTrue(isinstance(subscription_key, str)) def test_add_subscription(self): self.pubsub.add_subscription(self.project_id, self.namespace, self.channel, FakeClient()) self.assertTrue( self.pubsub.get_subscription_key(self.project_id, self.namespace, self.channel) in self.pubsub.subscriptions) def test_remove_subscription(self): self.pubsub.remove_subscription(self.project_id, self.namespace, self.channel, FakeClient()) self.assertTrue( self.pubsub.get_subscription_key(self.project_id, self.namespace, self.channel) not in self.pubsub.subscriptions)