class TestJsonStreamListener(TestCase):
def setUp(self):
logger.manager.disable = logging.CRITICAL
self.listener = JsonStreamListener()
def test_on_exception(self):
self.assertTrue(self.listener.streaming_exception is None)
try:
raise Exception("testing")
except Exception as exception:
self.listener.on_exception(exception)
self.assertEqual(self.listener.streaming_exception, exception, "Saves exception")
def test_on_error(self):
self.assertFalse(self.listener.on_error(404))
def test_on_status(self):
fake_status = dict(id=12345652)
self.assertTrue(self.listener.on_status(fake_status))
