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