def test_close_subscriber_does_close(self): with self.app.app_context(): subscriber = self.app.data.find_one("subscribers", None) self.assertTrue(subscriber.get("is_active")) PublishService().close_transmitter(subscriber, PublishQueueError.bad_schedule_error()) subscriber = self.app.data.find_one("subscribers", None) self.assertFalse(subscriber.get("is_active"))
def test_close_subscriber_doesnt_close(self): with self.app.app_context(): subscriber = self.app.data.find_one('subscribers', None) self.assertTrue(subscriber.get('is_active')) PublishService().close_transmitter(subscriber, PublishQueueError.unknown_format_error()) subscriber = self.app.data.find_one('subscribers', None) self.assertTrue(subscriber.get('is_active'))
def test_transmit_closes_subscriber(self): def mock_transmit(*args): raise PublishQueueError.bad_schedule_error() with self.app.app_context(): subscriber = self.app.data.find('subscribers', None, None)[0] publish_service = PublishService() publish_service._transmit = mock_transmit with assert_raises(PublishQueueError): publish_service.transmit(self.queue_items[0]) subscriber = self.app.data.find('subscribers', None, None)[0] self.assertFalse(subscriber.get('is_active')) self.assertIsNotNone(subscriber.get('last_closed'))