FM / suite

/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/suite/

/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/suite

Name	Size	Mode	Actions
test_cte.py	6788	0644	edit dl rm
test_cte.pyc	6518	0644	edit dl rm
test_cte.pyo	6518	0644	edit dl rm
test_ddl.py	2870	0644	edit dl rm
test_ddl.pyc	4498	0644	edit dl rm
test_ddl.pyo	4498	0644	edit dl rm
test_dialect.py	4572	0644	edit dl rm
test_dialect.pyc	5785	0644	edit dl rm
test_dialect.pyo	5650	0644	edit dl rm
test_insert.py	9586	0644	edit dl rm
test_insert.pyc	11356	0644	edit dl rm
test_insert.pyo	11030	0644	edit dl rm
test_reflection.py	37468	0644	edit dl rm
test_reflection.pyc	37964	0644	edit dl rm
test_reflection.pyo	37345	0644	edit dl rm
test_results.py	11504	0644	edit dl rm
test_results.pyc	14638	0644	edit dl rm
test_results.pyo	14086	0644	edit dl rm
test_select.py	19984	0644	edit dl rm
test_select.pyc	30094	0644	edit dl rm
test_select.pyo	30094	0644	edit dl rm
test_sequence.py	4661	0644	edit dl rm
test_sequence.pyc	6943	0644	edit dl rm
test_sequence.pyo	6943	0644	edit dl rm
test_types.py	29632	0644	edit dl rm
test_types.pyc	39399	0644	edit dl rm
test_types.pyo	39054	0644	edit dl rm
test_update_delete.py	1478	0644	edit dl rm
test_update_delete.pyc	2748	0644	edit dl rm
test_update_delete.pyo	2611	0644	edit dl rm
__init__.py	358	0644	edit dl rm
__init__.pyc	473	0644	edit dl rm
__init__.pyo	473	0644	edit dl rm

Edit: /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/suite/test_cte.py (6788B)

from .. import config
from .. import fixtures
from ..assertions import eq_
from ..schema import Column
from ..schema import Table
from ... import ForeignKey
from ... import Integer
from ... import select
from ... import String
from ... import testing

class CTETest(fixtures.TablesTest):
    __backend__ = True
    __requires__ = ("ctes",)

run_inserts = "each"
    run_deletes = "each"

@classmethod
    def define_tables(cls, metadata):
        Table(
            "some_table",
            metadata,
            Column("id", Integer, primary_key=True),
            Column("data", String(50)),
            Column("parent_id", ForeignKey("some_table.id")),
        )

Table(
            "some_other_table",
            metadata,
            Column("id", Integer, primary_key=True),
            Column("data", String(50)),
            Column("parent_id", Integer),
        )

@classmethod
    def insert_data(cls):
        config.db.execute(
            cls.tables.some_table.insert(),
            [
                {"id": 1, "data": "d1", "parent_id": None},
                {"id": 2, "data": "d2", "parent_id": 1},
                {"id": 3, "data": "d3", "parent_id": 1},
                {"id": 4, "data": "d4", "parent_id": 3},
                {"id": 5, "data": "d5", "parent_id": 3},
            ],
        )

def test_select_nonrecursive_round_trip(self):
        some_table = self.tables.some_table

with config.db.connect() as conn:
            cte = (
                select([some_table])
                .where(some_table.c.data.in_(["d2", "d3", "d4"]))
                .cte("some_cte")
            )
            result = conn.execute(
                select([cte.c.data]).where(cte.c.data.in_(["d4", "d5"]))
            )
            eq_(result.fetchall(), [("d4",)])

def test_select_recursive_round_trip(self):
        some_table = self.tables.some_table

with config.db.connect() as conn:
            cte = (
                select([some_table])
                .where(some_table.c.data.in_(["d2", "d3", "d4"]))
                .cte("some_cte", recursive=True)
            )

cte_alias = cte.alias("c1")
            st1 = some_table.alias()
            # note that SQL Server requires this to be UNION ALL,
            # can't be UNION
            cte = cte.union_all(
                select([st1]).where(st1.c.id == cte_alias.c.parent_id)
            )
            result = conn.execute(
                select([cte.c.data])
                .where(cte.c.data != "d2")
                .order_by(cte.c.data.desc())
            )
            eq_(
                result.fetchall(),
                [("d4",), ("d3",), ("d3",), ("d1",), ("d1",), ("d1",)],
            )

def test_insert_from_select_round_trip(self):
        some_table = self.tables.some_table
        some_other_table = self.tables.some_other_table

with config.db.connect() as conn:
            cte = (
                select([some_table])
                .where(some_table.c.data.in_(["d2", "d3", "d4"]))
                .cte("some_cte")
            )
            conn.execute(
                some_other_table.insert().from_select(
                    ["id", "data", "parent_id"], select([cte])
                )
            )
            eq_(
                conn.execute(
                    select([some_other_table]).order_by(some_other_table.c.id)
                ).fetchall(),
                [(2, "d2", 1), (3, "d3", 1), (4, "d4", 3)],
            )

@testing.requires.ctes_with_update_delete
    @testing.requires.update_from
    def test_update_from_round_trip(self):
        some_table = self.tables.some_table
        some_other_table = self.tables.some_other_table

with config.db.connect() as conn:
            conn.execute(
                some_other_table.insert().from_select(
                    ["id", "data", "parent_id"], select([some_table])
                )
            )

cte = (
                select([some_table])
                .where(some_table.c.data.in_(["d2", "d3", "d4"]))
                .cte("some_cte")
            )
            conn.execute(
                some_other_table.update()
                .values(parent_id=5)
                .where(some_other_table.c.data == cte.c.data)
            )
            eq_(
                conn.execute(
                    select([some_other_table]).order_by(some_other_table.c.id)
                ).fetchall(),
                [
                    (1, "d1", None),
                    (2, "d2", 5),
                    (3, "d3", 5),
                    (4, "d4", 5),
                    (5, "d5", 3),
                ],
            )

@testing.requires.ctes_with_update_delete
    @testing.requires.delete_from
    def test_delete_from_round_trip(self):
        some_table = self.tables.some_table
        some_other_table = self.tables.some_other_table

cte = (
                select([some_table])
                .where(some_table.c.data.in_(["d2", "d3", "d4"]))
                .cte("some_cte")
            )
            conn.execute(
                some_other_table.delete().where(
                    some_other_table.c.data == cte.c.data
                )
            )
            eq_(
                conn.execute(
                    select([some_other_table]).order_by(some_other_table.c.id)
                ).fetchall(),
                [(1, "d1", None), (5, "d5", 3)],
            )

@testing.requires.ctes_with_update_delete
    def test_delete_scalar_subq_round_trip(self):

some_table = self.tables.some_table
        some_other_table = self.tables.some_other_table

cte = (
                select([some_table])
                .where(some_table.c.data.in_(["d2", "d3", "d4"]))
                .cte("some_cte")
            )
            conn.execute(
                some_other_table.delete().where(
                    some_other_table.c.data
                    == select([cte.c.data]).where(
                        cte.c.id == some_other_table.c.id
                    )
                )
            )
            eq_(
                conn.execute(
                    select([some_other_table]).order_by(some_other_table.c.id)
                ).fetchall(),
                [(1, "d1", None), (5, "d5", 3)],
            )

cmd: