Method test_group_defer_newstyle

test/orm/test_deferred.py:142–186

(self, deferred_parameter: Union[bool, None])

    @testing.combinations(True, False, None, argnames="deferred_parameter")
    def test_group_defer_newstyle(self, deferred_parameter: Union[bool, None]):
        class Base(DeclarativeBase):
            pass

        class Order(Base):
            __tablename__ = "orders"

            id: Mapped[int] = mapped_column(primary_key=True)
            user_id: Mapped[int]
            address_id: Mapped[int]

            if deferred_parameter is None:
                isopen: Mapped[bool] = mapped_column(deferred_group="g1")
                description: Mapped[str] = mapped_column(deferred_group="g1")
            else:
                isopen: Mapped[bool] = mapped_column(
                    deferred=deferred_parameter, deferred_group="g1"
                )
                description: Mapped[str] = mapped_column(
                    deferred=deferred_parameter, deferred_group="g1"
                )

        if deferred_parameter is not False:
            self.assert_compile(
                select(Order),
                "SELECT orders.id, orders.user_id, orders.address_id "
                "FROM orders",
            )
            self.assert_compile(
                select(Order).options(undefer_group("g1")),
                "SELECT orders.id, orders.user_id, orders.address_id, "
                "orders.isopen, orders.description "
                "FROM orders",
            )
        else:
            self.assert_compile(
                select(Order),
                "SELECT orders.id, orders.user_id, orders.address_id, "
                "orders.isopen, orders.description FROM orders",
            )
            self.assert_compile(
                select(Order).options(undefer_group("g1")),
                "SELECT orders.id, orders.user_id, orders.address_id, "
                "orders.isopen, orders.description FROM orders",
            )

    def test_defer_primary_key(self):
        """what happens when we try to defer the primary key?"""