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