class TestMiddleware(TestCase): def setUp(self): self.middleware = Marimo() def tearDown(self): del self.middleware def test_process_request_marimo_widgets_added(self): req = mock.Mock() self.middleware.process_request(req) self.assertTrue(isinstance(req.marimo_widgets, list)) def test_process_request_marimo_writecapture_delay_added(self): req = mock.Mock() self.middleware.process_request(req) self.assertTrue(isinstance(req.marimo_writecapture_delay, MarimoEventContainer)) def test_process_response_marimo_widgets_added(self): req = mock.Mock() req.marimo_writecapture_delay = MarimoEventContainer() req.marimo_widgets = ['dummywidget'] resp = mock.Mock() resp.content = "dummytext ${MARIMO} moredumbtext" self.middleware.process_response(req, resp) self.assertTrue("dummywidget" in resp.content) def test_process_response_marimo_writecapture_delay_added(self): req = mock.Mock() req.marimo_widgets = [] req.marimo_writecapture_delay = MarimoEventContainer("documentready") resp = mock.Mock() resp.content = "dummytext ${MARIMO} moredumbtext" self.middleware.process_response(req, resp) self.assertTrue("documentready" in resp.content)
def setUp(self): self.middleware = Marimo()