Method test_w_new_entities

test/ext/test_baked.py:639–655

Test that the query can have its entities modified in an arbitrary callable, and that this new entity list is preserved when the query is invoked.

(self)

        )

    def test_w_new_entities(self):
        """Test that the query can have its entities modified in
        an arbitrary callable, and that this new entity list is preserved
        when the query is invoked.

        """
        User = self.classes.User

        bq = self.bakery(lambda s: s.query(User.id, User.name))

        bq += lambda q: q._legacy_from_self().with_entities(
            func.count(User.id)
        )

        for i in range(3):
            session = fixture_session()
            eq_(bq(session).all(), [(4,)])

    def test_conditional_step(self):
        """Test a large series of conditionals and assert that