orm/tests/test_raw_sql.py

14 class Person(db.Entity):  class
27             Person(id=1, name='John', age=30, dob=date(1985, 1, 1))
28             Person(id=2, name='Mike', age=32, dob=date(1983, 5, 20))
29             Person(id=3, name='Mary', age=20, dob=date(1995, 2, 15))
38         persons = select(p for p in Person if raw_sql('abs("p"."age") > 25'))[:]
39         self.assertEqual(set(persons), {Person[1], Person[2]})
44         persons = select(p for p in Person if raw_sql('abs("p"."age")') > 25)[:]
45         self.assertEqual(set(persons), {Person[1], Person[2]})
51         persons = select(p for p in Person if raw_sql('abs("p"."age") > $x'))[:]
52         self.assertEqual(set(persons), {Person[1], Person[2]})
59         persons = select(p for p in Person if raw_sql(s))[:]
60         self.assertEqual(set(persons), {Person[2], Person[3]})
67         persons = select(p for p in Person if cond)[:]
68         self.assertEqual(set(persons), {Person[2], Person[3]})
74         persons = select(p for p in Person if raw_sql('p.dob < $x'))[:]
75         self.assertEqual(set(persons), {Person[1], Person[2]})
82         persons = select(p for p in Person if raw_sql('p.age > $(x + y)'))[:]
83         self.assertEqual(set(persons), {Person[1], Person[2]})
88         persons = select(p for p in Person if raw_sql('p.dob < $date.today()'))[:]
89         self.assertEqual(set(persons), {Person[1], Person[2], Person[3]})
94         names = select(raw_sql('UPPER(p.name)') for p in Person)[:]
100         dates = select(raw_sql('(p.dob)') for p in Person).order_by(lambda: p.id)[:]
106 …     dates = select(raw_sql('(p.dob)', result_type=date) for p in Person).order_by(lambda: p.id)[:]
113         persons = Person.select(lambda p: p.age > raw_sql('$x'))[:]
114         self.assertEqual(set(persons), {Person[1], Person[2]})
120         persons = select(p for p in Person).filter(lambda p: p.age > raw_sql('$x'))[:]
121         self.assertEqual(set(persons), {Person[1], Person[2]})
127         persons = Person.select().filter(raw_sql('p.age > $x'))[:]
128         self.assertEqual(set(persons), {Person[1], Person[2]})
135 …persons = Person.select(lambda p: raw_sql("UPPER(p.name) || $x") == raw_sql("UPPER($y || '123')"))…
136         self.assertEqual(set(persons), {Person[1]})
143         q = select(p for p in Person if p.age > x and p.age < raw_sql('$y'))
148         self.assertEqual(set(persons), {Person[1]})
154         persons = Person.select().order_by(lambda p: raw_sql('SUBSTR(p.dob, $x)'))[:]
155         self.assertEqual(persons, [Person[1], Person[3], Person[2]])
161         persons = Person.select().order_by(raw_sql('SUBSTR(p.dob, $x)'))[:]
162         self.assertEqual(persons, [Person[1], Person[3], Person[2]])
169         select(p for p in Person if raw_sql(p.name))[:]
177         select(p for p in Person if raw_sql('p.dob < $x'))[:]
182         persons = select(p for p in Person if p.id == 1 and raw_sql('$x') is None)[:]
183         self.assertEqual(persons, [Person[1]])