class TestSentryLoggerSerivce(VumiTestCase): def setUp(self): import vumi.sentry self.client = DummySentryClient() self.patch(vumi.sentry, 'vumi_raven_client', lambda dsn: self.client) self.logger = LogPublisher() self.service = SentryLoggerService("http://example.com/", "test.logger", "worker-1", logger=self.logger) @inlineCallbacks def test_logging(self): yield self.service.startService() self.logger.msg("Hello", logLevel=logging.WARN) self.assertEqual(self.client.messages, [(("Hello", ), { 'data': { 'level': 30, 'logger': 'test.logger' }, 'tags': { 'worker-id': 'worker-1' } })]) del self.client.messages[:] yield self.service.stopService() self.logger.msg("Foo", logLevel=logging.WARN) self.assertEqual(self.client.messages, []) @inlineCallbacks def test_stop_not_running(self): yield self.service.stopService() self.assertFalse(self.service.running) @inlineCallbacks def test_start_stop(self): self.assertFalse(self.service.registered()) self.assertEqual(self.client.teardowns, 0) yield self.service.startService() self.assertTrue(self.service.registered()) yield self.service.stopService() self.assertFalse(self.service.registered()) self.assertEqual(self.client.teardowns, 1)
class TestSentryLoggerSerivce(VumiTestCase): def setUp(self): import vumi.sentry self.client = DummySentryClient() self.patch(vumi.sentry, 'vumi_raven_client', lambda dsn: self.client) self.logger = LogPublisher() self.service = SentryLoggerService("http://example.com/", "test.logger", "worker-1", logger=self.logger) @inlineCallbacks def test_logging(self): yield self.service.startService() self.logger.msg("Hello", logLevel=logging.WARN) self.assertEqual(self.client.messages, [ (("Hello",), {'data': {'level': 30, 'logger': 'test.logger'}, 'tags': {'worker-id': 'worker-1'}}) ]) del self.client.messages[:] yield self.service.stopService() self.logger.msg("Foo", logLevel=logging.WARN) self.assertEqual(self.client.messages, []) @inlineCallbacks def test_stop_not_running(self): yield self.service.stopService() self.assertFalse(self.service.running) @inlineCallbacks def test_start_stop(self): self.assertFalse(self.service.registered()) self.assertEqual(self.client.teardowns, 0) yield self.service.startService() self.assertTrue(self.service.registered()) yield self.service.stopService() self.assertFalse(self.service.registered()) self.assertEqual(self.client.teardowns, 1)