feat: add basic support for rows grid for master, batch views

tests/views/test_batch.py

@@ -3,6 +3,7 @@
 import datetime
 from unittest.mock import patch, MagicMock
 
+from sqlalchemy import orm
 from pyramid.httpexceptions import HTTPFound
 
 from wuttjamaican.db import model
@@ -19,12 +20,23 @@ class MockBatchRow(model.BatchRowMixin, model.Base):
     __tablename__ = 'testing_batch_mock_row'
     __batch_class__ = MockBatch
 
+MockBatch.__row_class__ = MockBatchRow
+
 class MockBatchHandler(BatchHandler):
     model_class = MockBatch
 
 
 class TestBatchMasterView(WebTestCase):
 
+    def setUp(self):
+        self.setup_web()
+
+        # nb. create MockBatch, MockBatchRow
+        model.Base.metadata.create_all(bind=self.session.bind)
+
+    def make_view(self):
+        return mod.BatchMasterView(self.request)
+
     def test_get_batch_handler(self):
         self.assertRaises(NotImplementedError, mod.BatchMasterView, self.request)
 
@@ -200,3 +212,78 @@ class TestBatchMasterView(WebTestCase):
                         self.session.commit()
                         # nb. should give up waiting after 1 second
                         self.assertRaises(RuntimeError, view.populate_thread, batch.uuid)
+
+    def test_get_row_model_class(self):
+        handler = MockBatchHandler(self.config)
+        with patch.object(mod.BatchMasterView, 'get_batch_handler', return_value=handler):
+            view = self.make_view()
+
+            self.assertRaises(AttributeError, view.get_row_model_class)
+
+            # row class determined from batch class
+            with patch.object(mod.BatchMasterView, 'model_class', new=MockBatch, create=True):
+                cls = view.get_row_model_class()
+                self.assertIs(cls, MockBatchRow)
+
+            self.assertRaises(AttributeError, view.get_row_model_class)
+
+            # view may specify row class
+            with patch.object(mod.BatchMasterView, 'row_model_class', new=MockBatchRow, create=True):
+                cls = view.get_row_model_class()
+                self.assertIs(cls, MockBatchRow)
+
+    def test_get_row_grid_data(self):
+        handler = MockBatchHandler(self.config)
+        model = self.app.model
+
+        user = model.User(username='barney')
+        self.session.add(user)
+
+        batch = handler.make_batch(self.session, created_by=user)
+        self.session.add(batch)
+        row = handler.make_row()
+        handler.add_row(batch, row)
+        self.session.flush()
+
+        with patch.object(mod.BatchMasterView, 'get_batch_handler', return_value=handler):
+
+            view = self.make_view()
+            self.assertRaises(AttributeError, view.get_row_grid_data, batch)
+
+            Session = MagicMock(return_value=self.session)
+            Session.query.side_effect = lambda m: self.session.query(m)
+            with patch.multiple(mod.BatchMasterView, create=True,
+                                Session=Session,
+                                model_class=MockBatch):
+
+                view = self.make_view()
+                data = view.get_row_grid_data(batch)
+                self.assertIsInstance(data, orm.Query)
+                self.assertEqual(data.count(), 1)
+
+    def test_configure_row_grid(self):
+        handler = MockBatchHandler(self.config)
+        model = self.app.model
+
+        user = model.User(username='barney')
+        self.session.add(user)
+
+        batch = handler.make_batch(self.session, created_by=user)
+        self.session.add(batch)
+        row = handler.make_row()
+        handler.add_row(batch, row)
+        self.session.flush()
+
+        with patch.object(mod.BatchMasterView, 'get_batch_handler', return_value=handler):
+
+            Session = MagicMock(return_value=self.session)
+            Session.query.side_effect = lambda m: self.session.query(m)
+            with patch.multiple(mod.BatchMasterView, create=True,
+                                Session=Session,
+                                model_class=MockBatch):
+
+                with patch.object(self.request, 'matchdict', new={'uuid': batch.uuid}):
+                    view = self.make_view()
+                    grid = view.make_row_model_grid(batch)
+                    self.assertIn('sequence', grid.labels)
+                    self.assertEqual(grid.labels['sequence'], "Seq.")