Method test_annotated_fetch

test/sql/test_external_traversal.py:526–573

test #9526

(self, limit_type: testing.Variation, dialect)

    @testing.variation("limit_type", ["limit", "fetch"])
    @testing.variation("dialect", ["default", "oracle"])
    def test_annotated_fetch(self, limit_type: testing.Variation, dialect):
        """test #9526"""

        if limit_type.limit:
            stmt = select(column("q")).limit(1)
        elif limit_type.fetch:
            stmt = select(column("q")).fetch(1)
        else:
            limit_type.fail()

        stmt = _deep_annotate(stmt, {"foo": "bar"})

        if limit_type.limit:
            if dialect.default:
                self.assert_compile(
                    stmt,
                    "SELECT q LIMIT :param_1",
                    use_literal_execute_for_simple_int=True,
                    dialect=dialect.name,
                )
            elif dialect.oracle:
                self.assert_compile(
                    stmt,
                    "SELECT q FROM DUAL FETCH FIRST "
                    "__[POSTCOMPILE_param_1] ROWS ONLY",
                    dialect=dialect.name,
                )
            else:
                dialect.fail()
        elif limit_type.fetch:
            if dialect.default:
                self.assert_compile(
                    stmt,
                    "SELECT q FETCH FIRST __[POSTCOMPILE_param_1] ROWS ONLY",
                    use_literal_execute_for_simple_int=True,
                    dialect=dialect.name,
                )
            elif dialect.oracle:
                self.assert_compile(
                    stmt,
                    "SELECT q FROM DUAL FETCH FIRST "
                    "__[POSTCOMPILE_param_1] ROWS ONLY",
                    dialect=dialect.name,
                )
            else:
                dialect.fail()
        else:
            limit_type.fail()

    @testing.combinations((null(),), (true(),))
    def test_dont_adapt_singleton_elements(self, elem):