diff --git a/tests/web/views/batch/test_neworder.py b/tests/web/views/batch/test_neworder.py index 8a34b26..f4b5dd8 100644 --- a/tests/web/views/batch/test_neworder.py +++ b/tests/web/views/batch/test_neworder.py @@ -86,9 +86,19 @@ class TestNewOrderBatchView(WebTestCase): def test_configure_row_grid(self): model = self.app.model view = self.make_view() + handler = view.batch_handler + + user = model.User(username="fred") + self.session.add(user) + batch = handler.make_batch(self.session, created_by=user) + self.session.add(batch) + self.session.commit() + grid = view.make_grid(model_class=model.NewOrderBatchRow) self.assertNotIn("total_price", grid.renderers) - view.configure_row_grid(grid) + with patch.object(view, "Session", return_value=self.session): + with patch.object(self.request, "matchdict", new={"uuid": batch.uuid}): + view.configure_row_grid(grid) self.assertIn("total_price", grid.renderers) def test_get_xref_buttons(self):