def test_count(self): from papyrus.protocol import Protocol from mock import patch engine = self._get_engine() Session = self._get_session(engine) MappedClass = self._get_mapped_class() proto = Protocol(Session, MappedClass, "geom") # We make Query.count return the query and just check it includes # "SELECT". Yes, not so good! request = testing.DummyRequest() with patch('sqlalchemy.orm.query.Query.count', lambda q: q): query = proto.count(request) self.assertTrue(b"SELECT" in query_to_str(query, engine))
def test_count(self): from papyrus.protocol import Protocol from mock import patch engine = self._get_engine() Session = self._get_session(engine) MappedClass = self._get_mapped_class() proto = Protocol(Session, MappedClass, "geom") # We make Query.count return the query and just check it includes # "SELECT". Yes, not so good! request = testing.DummyRequest() with patch('sqlalchemy.orm.query.Query.count', lambda q : q): query = proto.count(request) self.assertTrue("SELECT" in query_to_str(query, engine))