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))