705 lines
21 KiB
Python
705 lines
21 KiB
Python
|
from .. import config
|
||
|
|
from .. import fixtures
|
||
|
|
from ..assertions import eq_
|
||
|
|
from ..assertions import in_
|
||
|
|
from ..schema import Column
|
||
|
|
from ..schema import Table
|
||
|
|
from ... import bindparam
|
||
|
|
from ... import case
|
||
|
|
from ... import Computed
|
||
|
|
from ... import false
|
||
|
|
from ... import func
|
||
|
|
from ... import Integer
|
||
|
|
from ... import literal_column
|
||
|
|
from ... import null
|
||
|
|
from ... import select
|
||
|
|
from ... import String
|
||
|
|
from ... import testing
|
||
|
|
from ... import text
|
||
|
|
from ... import true
|
||
|
|
from ... import tuple_
|
||
|
|
from ... import union
|
||
|
|
from ... import util
|
||
|
|
|
||
|
|
|
||
|
|
class CollateTest(fixtures.TablesTest):
|
||
|
|
__backend__ = True
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def define_tables(cls, metadata):
|
||
|
|
Table(
|
||
|
|
"some_table",
|
||
|
|
metadata,
|
||
|
|
Column("id", Integer, primary_key=True),
|
||
|
|
Column("data", String(100)),
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def insert_data(cls):
|
||
|
|
config.db.execute(
|
||
|
|
cls.tables.some_table.insert(),
|
||
|
|
[
|
||
|
|
{"id": 1, "data": "collate data1"},
|
||
|
|
{"id": 2, "data": "collate data2"},
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
def _assert_result(self, select, result):
|
||
|
|
eq_(config.db.execute(select).fetchall(), result)
|
||
|
|
|
||
|
|
@testing.requires.order_by_collation
|
||
|
|
def test_collate_order_by(self):
|
||
|
|
collation = testing.requires.get_order_by_collation(testing.config)
|
||
|
|
|
||
|
|
self._assert_result(
|
||
|
|
select([self.tables.some_table]).order_by(
|
||
|
|
self.tables.some_table.c.data.collate(collation).asc()
|
||
|
|
),
|
||
|
|
[(1, "collate data1"), (2, "collate data2")],
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class OrderByLabelTest(fixtures.TablesTest):
|
||
|
|
"""Test the dialect sends appropriate ORDER BY expressions when
|
||
|
|
labels are used.
|
||
|
|
|
||
|
|
This essentially exercises the "supports_simple_order_by_label"
|
||
|
|
setting.
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
__backend__ = True
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def define_tables(cls, metadata):
|
||
|
|
Table(
|
||
|
|
"some_table",
|
||
|
|
metadata,
|
||
|
|
Column("id", Integer, primary_key=True),
|
||
|
|
Column("x", Integer),
|
||
|
|
Column("y", Integer),
|
||
|
|
Column("q", String(50)),
|
||
|
|
Column("p", String(50)),
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def insert_data(cls):
|
||
|
|
config.db.execute(
|
||
|
|
cls.tables.some_table.insert(),
|
||
|
|
[
|
||
|
|
{"id": 1, "x": 1, "y": 2, "q": "q1", "p": "p3"},
|
||
|
|
{"id": 2, "x": 2, "y": 3, "q": "q2", "p": "p2"},
|
||
|
|
{"id": 3, "x": 3, "y": 4, "q": "q3", "p": "p1"},
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
def _assert_result(self, select, result):
|
||
|
|
eq_(config.db.execute(select).fetchall(), result)
|
||
|
|
|
||
|
|
def test_plain(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
lx = table.c.x.label("lx")
|
||
|
|
self._assert_result(select([lx]).order_by(lx), [(1,), (2,), (3,)])
|
||
|
|
|
||
|
|
def test_composed_int(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
lx = (table.c.x + table.c.y).label("lx")
|
||
|
|
self._assert_result(select([lx]).order_by(lx), [(3,), (5,), (7,)])
|
||
|
|
|
||
|
|
def test_composed_multiple(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
lx = (table.c.x + table.c.y).label("lx")
|
||
|
|
ly = (func.lower(table.c.q) + table.c.p).label("ly")
|
||
|
|
self._assert_result(
|
||
|
|
select([lx, ly]).order_by(lx, ly.desc()),
|
||
|
|
[(3, util.u("q1p3")), (5, util.u("q2p2")), (7, util.u("q3p1"))],
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_plain_desc(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
lx = table.c.x.label("lx")
|
||
|
|
self._assert_result(
|
||
|
|
select([lx]).order_by(lx.desc()), [(3,), (2,), (1,)]
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_composed_int_desc(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
lx = (table.c.x + table.c.y).label("lx")
|
||
|
|
self._assert_result(
|
||
|
|
select([lx]).order_by(lx.desc()), [(7,), (5,), (3,)]
|
||
|
|
)
|
||
|
|
|
||
|
|
@testing.requires.group_by_complex_expression
|
||
|
|
def test_group_by_composed(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
expr = (table.c.x + table.c.y).label("lx")
|
||
|
|
stmt = (
|
||
|
|
select([func.count(table.c.id), expr])
|
||
|
|
.group_by(expr)
|
||
|
|
.order_by(expr)
|
||
|
|
)
|
||
|
|
self._assert_result(stmt, [(1, 3), (1, 5), (1, 7)])
|
||
|
|
|
||
|
|
|
||
|
|
class LimitOffsetTest(fixtures.TablesTest):
|
||
|
|
__backend__ = True
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def define_tables(cls, metadata):
|
||
|
|
Table(
|
||
|
|
"some_table",
|
||
|
|
metadata,
|
||
|
|
Column("id", Integer, primary_key=True),
|
||
|
|
Column("x", Integer),
|
||
|
|
Column("y", Integer),
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def insert_data(cls):
|
||
|
|
config.db.execute(
|
||
|
|
cls.tables.some_table.insert(),
|
||
|
|
[
|
||
|
|
{"id": 1, "x": 1, "y": 2},
|
||
|
|
{"id": 2, "x": 2, "y": 3},
|
||
|
|
{"id": 3, "x": 3, "y": 4},
|
||
|
|
{"id": 4, "x": 4, "y": 5},
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
def _assert_result(self, select, result, params=()):
|
||
|
|
eq_(config.db.execute(select, params).fetchall(), result)
|
||
|
|
|
||
|
|
def test_simple_limit(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
self._assert_result(
|
||
|
|
select([table]).order_by(table.c.id).limit(2),
|
||
|
|
[(1, 1, 2), (2, 2, 3)],
|
||
|
|
)
|
||
|
|
|
||
|
|
@testing.requires.offset
|
||
|
|
def test_simple_offset(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
self._assert_result(
|
||
|
|
select([table]).order_by(table.c.id).offset(2),
|
||
|
|
[(3, 3, 4), (4, 4, 5)],
|
||
|
|
)
|
||
|
|
|
||
|
|
@testing.requires.offset
|
||
|
|
def test_simple_limit_offset(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
self._assert_result(
|
||
|
|
select([table]).order_by(table.c.id).limit(2).offset(1),
|
||
|
|
[(2, 2, 3), (3, 3, 4)],
|
||
|
|
)
|
||
|
|
|
||
|
|
@testing.requires.offset
|
||
|
|
def test_limit_offset_nobinds(self):
|
||
|
|
"""test that 'literal binds' mode works - no bound params."""
|
||
|
|
|
||
|
|
table = self.tables.some_table
|
||
|
|
stmt = select([table]).order_by(table.c.id).limit(2).offset(1)
|
||
|
|
sql = stmt.compile(
|
||
|
|
dialect=config.db.dialect, compile_kwargs={"literal_binds": True}
|
||
|
|
)
|
||
|
|
sql = str(sql)
|
||
|
|
|
||
|
|
self._assert_result(sql, [(2, 2, 3), (3, 3, 4)])
|
||
|
|
|
||
|
|
@testing.requires.bound_limit_offset
|
||
|
|
def test_bound_limit(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
self._assert_result(
|
||
|
|
select([table]).order_by(table.c.id).limit(bindparam("l")),
|
||
|
|
[(1, 1, 2), (2, 2, 3)],
|
||
|
|
params={"l": 2},
|
||
|
|
)
|
||
|
|
|
||
|
|
@testing.requires.bound_limit_offset
|
||
|
|
def test_bound_offset(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
self._assert_result(
|
||
|
|
select([table]).order_by(table.c.id).offset(bindparam("o")),
|
||
|
|
[(3, 3, 4), (4, 4, 5)],
|
||
|
|
params={"o": 2},
|
||
|
|
)
|
||
|
|
|
||
|
|
@testing.requires.bound_limit_offset
|
||
|
|
def test_bound_limit_offset(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
self._assert_result(
|
||
|
|
select([table])
|
||
|
|
.order_by(table.c.id)
|
||
|
|
.limit(bindparam("l"))
|
||
|
|
.offset(bindparam("o")),
|
||
|
|
[(2, 2, 3), (3, 3, 4)],
|
||
|
|
params={"l": 2, "o": 1},
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class CompoundSelectTest(fixtures.TablesTest):
|
||
|
|
__backend__ = True
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def define_tables(cls, metadata):
|
||
|
|
Table(
|
||
|
|
"some_table",
|
||
|
|
metadata,
|
||
|
|
Column("id", Integer, primary_key=True),
|
||
|
|
Column("x", Integer),
|
||
|
|
Column("y", Integer),
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def insert_data(cls):
|
||
|
|
config.db.execute(
|
||
|
|
cls.tables.some_table.insert(),
|
||
|
|
[
|
||
|
|
{"id": 1, "x": 1, "y": 2},
|
||
|
|
{"id": 2, "x": 2, "y": 3},
|
||
|
|
{"id": 3, "x": 3, "y": 4},
|
||
|
|
{"id": 4, "x": 4, "y": 5},
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
def _assert_result(self, select, result, params=()):
|
||
|
|
eq_(config.db.execute(select, params).fetchall(), result)
|
||
|
|
|
||
|
|
def test_plain_union(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
s1 = select([table]).where(table.c.id == 2)
|
||
|
|
s2 = select([table]).where(table.c.id == 3)
|
||
|
|
|
||
|
|
u1 = union(s1, s2)
|
||
|
|
self._assert_result(u1.order_by(u1.c.id), [(2, 2, 3), (3, 3, 4)])
|
||
|
|
|
||
|
|
def test_select_from_plain_union(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
s1 = select([table]).where(table.c.id == 2)
|
||
|
|
s2 = select([table]).where(table.c.id == 3)
|
||
|
|
|
||
|
|
u1 = union(s1, s2).alias().select()
|
||
|
|
self._assert_result(u1.order_by(u1.c.id), [(2, 2, 3), (3, 3, 4)])
|
||
|
|
|
||
|
|
@testing.requires.order_by_col_from_union
|
||
|
|
@testing.requires.parens_in_union_contained_select_w_limit_offset
|
||
|
|
def test_limit_offset_selectable_in_unions(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
s1 = (
|
||
|
|
select([table])
|
||
|
|
.where(table.c.id == 2)
|
||
|
|
.limit(1)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
s2 = (
|
||
|
|
select([table])
|
||
|
|
.where(table.c.id == 3)
|
||
|
|
.limit(1)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
u1 = union(s1, s2).limit(2)
|
||
|
|
self._assert_result(u1.order_by(u1.c.id), [(2, 2, 3), (3, 3, 4)])
|
||
|
|
|
||
|
|
@testing.requires.parens_in_union_contained_select_wo_limit_offset
|
||
|
|
def test_order_by_selectable_in_unions(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
s1 = select([table]).where(table.c.id == 2).order_by(table.c.id)
|
||
|
|
s2 = select([table]).where(table.c.id == 3).order_by(table.c.id)
|
||
|
|
|
||
|
|
u1 = union(s1, s2).limit(2)
|
||
|
|
self._assert_result(u1.order_by(u1.c.id), [(2, 2, 3), (3, 3, 4)])
|
||
|
|
|
||
|
|
def test_distinct_selectable_in_unions(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
s1 = select([table]).where(table.c.id == 2).distinct()
|
||
|
|
s2 = select([table]).where(table.c.id == 3).distinct()
|
||
|
|
|
||
|
|
u1 = union(s1, s2).limit(2)
|
||
|
|
self._assert_result(u1.order_by(u1.c.id), [(2, 2, 3), (3, 3, 4)])
|
||
|
|
|
||
|
|
@testing.requires.parens_in_union_contained_select_w_limit_offset
|
||
|
|
def test_limit_offset_in_unions_from_alias(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
s1 = (
|
||
|
|
select([table])
|
||
|
|
.where(table.c.id == 2)
|
||
|
|
.limit(1)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
s2 = (
|
||
|
|
select([table])
|
||
|
|
.where(table.c.id == 3)
|
||
|
|
.limit(1)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
# this necessarily has double parens
|
||
|
|
u1 = union(s1, s2).alias()
|
||
|
|
self._assert_result(
|
||
|
|
u1.select().limit(2).order_by(u1.c.id), [(2, 2, 3), (3, 3, 4)]
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_limit_offset_aliased_selectable_in_unions(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
s1 = (
|
||
|
|
select([table])
|
||
|
|
.where(table.c.id == 2)
|
||
|
|
.limit(1)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
.alias()
|
||
|
|
.select()
|
||
|
|
)
|
||
|
|
s2 = (
|
||
|
|
select([table])
|
||
|
|
.where(table.c.id == 3)
|
||
|
|
.limit(1)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
.alias()
|
||
|
|
.select()
|
||
|
|
)
|
||
|
|
|
||
|
|
u1 = union(s1, s2).limit(2)
|
||
|
|
self._assert_result(u1.order_by(u1.c.id), [(2, 2, 3), (3, 3, 4)])
|
||
|
|
|
||
|
|
|
||
|
|
class ExpandingBoundInTest(fixtures.TablesTest):
|
||
|
|
__backend__ = True
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def define_tables(cls, metadata):
|
||
|
|
Table(
|
||
|
|
"some_table",
|
||
|
|
metadata,
|
||
|
|
Column("id", Integer, primary_key=True),
|
||
|
|
Column("x", Integer),
|
||
|
|
Column("y", Integer),
|
||
|
|
Column("z", String(50)),
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def insert_data(cls):
|
||
|
|
config.db.execute(
|
||
|
|
cls.tables.some_table.insert(),
|
||
|
|
[
|
||
|
|
{"id": 1, "x": 1, "y": 2, "z": "z1"},
|
||
|
|
{"id": 2, "x": 2, "y": 3, "z": "z2"},
|
||
|
|
{"id": 3, "x": 3, "y": 4, "z": "z3"},
|
||
|
|
{"id": 4, "x": 4, "y": 5, "z": "z4"},
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
def _assert_result(self, select, result, params=()):
|
||
|
|
eq_(config.db.execute(select, params).fetchall(), result)
|
||
|
|
|
||
|
|
def test_multiple_empty_sets(self):
|
||
|
|
# test that any anonymous aliasing used by the dialect
|
||
|
|
# is fine with duplicates
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(table.c.x.in_(bindparam("q", expanding=True)))
|
||
|
|
.where(table.c.y.in_(bindparam("p", expanding=True)))
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(stmt, [], params={"q": [], "p": []})
|
||
|
|
|
||
|
|
@testing.requires.tuple_in
|
||
|
|
def test_empty_heterogeneous_tuples(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(
|
||
|
|
tuple_(table.c.x, table.c.z).in_(
|
||
|
|
bindparam("q", expanding=True)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(stmt, [], params={"q": []})
|
||
|
|
|
||
|
|
@testing.requires.tuple_in
|
||
|
|
def test_empty_homogeneous_tuples(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(
|
||
|
|
tuple_(table.c.x, table.c.y).in_(
|
||
|
|
bindparam("q", expanding=True)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(stmt, [], params={"q": []})
|
||
|
|
|
||
|
|
def test_bound_in_scalar(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(table.c.x.in_(bindparam("q", expanding=True)))
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(stmt, [(2,), (3,), (4,)], params={"q": [2, 3, 4]})
|
||
|
|
|
||
|
|
@testing.requires.tuple_in
|
||
|
|
def test_bound_in_two_tuple(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(
|
||
|
|
tuple_(table.c.x, table.c.y).in_(
|
||
|
|
bindparam("q", expanding=True)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(
|
||
|
|
stmt, [(2,), (3,), (4,)], params={"q": [(2, 3), (3, 4), (4, 5)]}
|
||
|
|
)
|
||
|
|
|
||
|
|
@testing.requires.tuple_in
|
||
|
|
def test_bound_in_heterogeneous_two_tuple(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(
|
||
|
|
tuple_(table.c.x, table.c.z).in_(
|
||
|
|
bindparam("q", expanding=True)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(
|
||
|
|
stmt,
|
||
|
|
[(2,), (3,), (4,)],
|
||
|
|
params={"q": [(2, "z2"), (3, "z3"), (4, "z4")]},
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_empty_set_against_integer(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(table.c.x.in_(bindparam("q", expanding=True)))
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(stmt, [], params={"q": []})
|
||
|
|
|
||
|
|
def test_empty_set_against_integer_negation(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(table.c.x.notin_(bindparam("q", expanding=True)))
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(stmt, [(1,), (2,), (3,), (4,)], params={"q": []})
|
||
|
|
|
||
|
|
def test_empty_set_against_string(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(table.c.z.in_(bindparam("q", expanding=True)))
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(stmt, [], params={"q": []})
|
||
|
|
|
||
|
|
def test_empty_set_against_string_negation(self):
|
||
|
|
table = self.tables.some_table
|
||
|
|
|
||
|
|
stmt = (
|
||
|
|
select([table.c.id])
|
||
|
|
.where(table.c.z.notin_(bindparam("q", expanding=True)))
|
||
|
|
.order_by(table.c.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
self._assert_result(stmt, [(1,), (2,), (3,), (4,)], params={"q": []})
|
||
|
|
|
||
|
|
def test_null_in_empty_set_is_false(self):
|
||
|
|
stmt = select(
|
||
|
|
[
|
||
|
|
case(
|
||
|
|
[
|
||
|
|
(
|
||
|
|
null().in_(
|
||
|
|
bindparam("foo", value=(), expanding=True)
|
||
|
|
),
|
||
|
|
true(),
|
||
|
|
)
|
||
|
|
],
|
||
|
|
else_=false(),
|
||
|
|
)
|
||
|
|
]
|
||
|
|
)
|
||
|
|
in_(config.db.execute(stmt).fetchone()[0], (False, 0))
|
||
|
|
|
||
|
|
|
||
|
|
class LikeFunctionsTest(fixtures.TablesTest):
|
||
|
|
__backend__ = True
|
||
|
|
|
||
|
|
run_inserts = "once"
|
||
|
|
run_deletes = None
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def define_tables(cls, metadata):
|
||
|
|
Table(
|
||
|
|
"some_table",
|
||
|
|
metadata,
|
||
|
|
Column("id", Integer, primary_key=True),
|
||
|
|
Column("data", String(50)),
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def insert_data(cls):
|
||
|
|
config.db.execute(
|
||
|
|
cls.tables.some_table.insert(),
|
||
|
|
[
|
||
|
|
{"id": 1, "data": "abcdefg"},
|
||
|
|
{"id": 2, "data": "ab/cdefg"},
|
||
|
|
{"id": 3, "data": "ab%cdefg"},
|
||
|
|
{"id": 4, "data": "ab_cdefg"},
|
||
|
|
{"id": 5, "data": "abcde/fg"},
|
||
|
|
{"id": 6, "data": "abcde%fg"},
|
||
|
|
{"id": 7, "data": "ab#cdefg"},
|
||
|
|
{"id": 8, "data": "ab9cdefg"},
|
||
|
|
{"id": 9, "data": "abcde#fg"},
|
||
|
|
{"id": 10, "data": "abcd9fg"},
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
def _test(self, expr, expected):
|
||
|
|
some_table = self.tables.some_table
|
||
|
|
|
||
|
|
with config.db.connect() as conn:
|
||
|
|
rows = {
|
||
|
|
value
|
||
|
|
for value, in conn.execute(
|
||
|
|
select([some_table.c.id]).where(expr)
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
eq_(rows, expected)
|
||
|
|
|
||
|
|
def test_startswith_unescaped(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.startswith("ab%c"), {1, 2, 3, 4, 5, 6, 7, 8, 9, 10})
|
||
|
|
|
||
|
|
def test_startswith_autoescape(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.startswith("ab%c", autoescape=True), {3})
|
||
|
|
|
||
|
|
def test_startswith_sqlexpr(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(
|
||
|
|
col.startswith(literal_column("'ab%c'")),
|
||
|
|
{1, 2, 3, 4, 5, 6, 7, 8, 9, 10},
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_startswith_escape(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.startswith("ab##c", escape="#"), {7})
|
||
|
|
|
||
|
|
def test_startswith_autoescape_escape(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.startswith("ab%c", autoescape=True, escape="#"), {3})
|
||
|
|
self._test(col.startswith("ab#c", autoescape=True, escape="#"), {7})
|
||
|
|
|
||
|
|
def test_endswith_unescaped(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.endswith("e%fg"), {1, 2, 3, 4, 5, 6, 7, 8, 9})
|
||
|
|
|
||
|
|
def test_endswith_sqlexpr(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(
|
||
|
|
col.endswith(literal_column("'e%fg'")), {1, 2, 3, 4, 5, 6, 7, 8, 9}
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_endswith_autoescape(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.endswith("e%fg", autoescape=True), {6})
|
||
|
|
|
||
|
|
def test_endswith_escape(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.endswith("e##fg", escape="#"), {9})
|
||
|
|
|
||
|
|
def test_endswith_autoescape_escape(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.endswith("e%fg", autoescape=True, escape="#"), {6})
|
||
|
|
self._test(col.endswith("e#fg", autoescape=True, escape="#"), {9})
|
||
|
|
|
||
|
|
def test_contains_unescaped(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.contains("b%cde"), {1, 2, 3, 4, 5, 6, 7, 8, 9})
|
||
|
|
|
||
|
|
def test_contains_autoescape(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.contains("b%cde", autoescape=True), {3})
|
||
|
|
|
||
|
|
def test_contains_escape(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.contains("b##cde", escape="#"), {7})
|
||
|
|
|
||
|
|
def test_contains_autoescape_escape(self):
|
||
|
|
col = self.tables.some_table.c.data
|
||
|
|
self._test(col.contains("b%cd", autoescape=True, escape="#"), {3})
|
||
|
|
self._test(col.contains("b#cd", autoescape=True, escape="#"), {7})
|
||
|
|
|
||
|
|
|
||
|
|
class ComputedColumnTest(fixtures.TablesTest):
|
||
|
|
__backend__ = True
|
||
|
|
__requires__ = ("computed_columns",)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def define_tables(cls, metadata):
|
||
|
|
Table(
|
||
|
|
"square",
|
||
|
|
metadata,
|
||
|
|
Column("id", Integer, primary_key=True),
|
||
|
|
Column("side", Integer),
|
||
|
|
Column("area", Integer, Computed("side * side")),
|
||
|
|
Column("perimeter", Integer, Computed("4 * side")),
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def insert_data(cls):
|
||
|
|
with config.db.begin() as conn:
|
||
|
|
conn.execute(
|
||
|
|
cls.tables.square.insert(),
|
||
|
|
[{"id": 1, "side": 10}, {"id": 10, "side": 42}],
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_select_all(self):
|
||
|
|
with config.db.connect() as conn:
|
||
|
|
res = conn.execute(
|
||
|
|
select([text("*")])
|
||
|
|
.select_from(self.tables.square)
|
||
|
|
.order_by(self.tables.square.c.id)
|
||
|
|
).fetchall()
|
||
|
|
eq_(res, [(1, 10, 100, 40), (10, 42, 1764, 168)])
|
||
|
|
|
||
|
|
def test_select_columns(self):
|
||
|
|
with config.db.connect() as conn:
|
||
|
|
res = conn.execute(
|
||
|
|
select(
|
||
|
|
[self.tables.square.c.area, self.tables.square.c.perimeter]
|
||
|
|
)
|
||
|
|
.select_from(self.tables.square)
|
||
|
|
.order_by(self.tables.square.c.id)
|
||
|
|
).fetchall()
|
||
|
|
eq_(res, [(100, 40), (1764, 168)])
|