summaryrefslogtreecommitdiff
path: root/test/ext/test_hybrid.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/ext/test_hybrid.py')
-rw-r--r--test/ext/test_hybrid.py45
1 files changed, 23 insertions, 22 deletions
diff --git a/test/ext/test_hybrid.py b/test/ext/test_hybrid.py
index fbac35f7e..048a8b52d 100644
--- a/test/ext/test_hybrid.py
+++ b/test/ext/test_hybrid.py
@@ -17,6 +17,7 @@ from sqlalchemy.testing import AssertsCompiledSQL
from sqlalchemy.testing import eq_
from sqlalchemy.testing import fixtures
from sqlalchemy.testing import is_
+from sqlalchemy.testing.fixtures import fixture_session
from sqlalchemy.testing.schema import Column
@@ -70,14 +71,14 @@ class PropertyComparatorTest(fixtures.TestBase, AssertsCompiledSQL):
def test_query(self):
A = self._fixture()
- sess = Session()
+ sess = fixture_session()
self.assert_compile(
sess.query(A.value), "SELECT a.value AS a_value FROM a"
)
def test_aliased_query(self):
A = self._fixture()
- sess = Session()
+ sess = fixture_session()
self.assert_compile(
sess.query(aliased(A).value),
"SELECT a_1.value AS a_1_value FROM a AS a_1",
@@ -85,7 +86,7 @@ class PropertyComparatorTest(fixtures.TestBase, AssertsCompiledSQL):
def test_aliased_filter(self):
A = self._fixture()
- sess = Session()
+ sess = fixture_session()
self.assert_compile(
sess.query(aliased(A)).filter_by(value="foo"),
"SELECT a_1.value AS a_1_value, a_1.id AS a_1_id "
@@ -183,7 +184,7 @@ class PropertyExpressionTest(fixtures.TestBase, AssertsCompiledSQL):
def test_any(self):
A, B = self._relationship_fixture()
- sess = Session()
+ sess = fixture_session()
self.assert_compile(
sess.query(B).filter(B.as_.any(value=5)),
"SELECT b.id AS b_id FROM b WHERE EXISTS "
@@ -200,7 +201,7 @@ class PropertyExpressionTest(fixtures.TestBase, AssertsCompiledSQL):
def test_query(self):
A = self._fixture()
- sess = Session()
+ sess = fixture_session()
self.assert_compile(
sess.query(A).filter_by(value="foo"),
"SELECT a.value AS a_value, a.id AS a_id "
@@ -209,7 +210,7 @@ class PropertyExpressionTest(fixtures.TestBase, AssertsCompiledSQL):
def test_aliased_query(self):
A = self._fixture()
- sess = Session()
+ sess = fixture_session()
self.assert_compile(
sess.query(aliased(A)).filter_by(value="foo"),
"SELECT a_1.value AS a_1_value, a_1.id AS a_1_id "
@@ -489,7 +490,7 @@ class MethodExpressionTest(fixtures.TestBase, AssertsCompiledSQL):
def test_query(self):
A = self._fixture()
- sess = Session()
+ sess = fixture_session()
self.assert_compile(
sess.query(A).filter(A.value(5) == "foo"),
"SELECT a.value AS a_value, a.id AS a_id "
@@ -498,7 +499,7 @@ class MethodExpressionTest(fixtures.TestBase, AssertsCompiledSQL):
def test_aliased_query(self):
A = self._fixture()
- sess = Session()
+ sess = fixture_session()
a1 = aliased(A)
self.assert_compile(
sess.query(a1).filter(a1.value(5) == "foo"),
@@ -508,7 +509,7 @@ class MethodExpressionTest(fixtures.TestBase, AssertsCompiledSQL):
def test_query_col(self):
A = self._fixture()
- sess = Session()
+ sess = fixture_session()
self.assert_compile(
sess.query(A.value(5)),
"SELECT foo(a.value, :foo_1) + :foo_2 AS anon_1 FROM a",
@@ -516,7 +517,7 @@ class MethodExpressionTest(fixtures.TestBase, AssertsCompiledSQL):
def test_aliased_query_col(self):
A = self._fixture()
- sess = Session()
+ sess = fixture_session()
self.assert_compile(
sess.query(aliased(A).value(5)),
"SELECT foo(a_1.value, :foo_1) + :foo_2 AS anon_1 FROM a AS a_1",
@@ -610,7 +611,7 @@ class BulkUpdateTest(fixtures.DeclarativeMappedTest, AssertsCompiledSQL):
def test_evaluate_hybrid_attr_indirect(self):
Person = self.classes.Person
- s = Session()
+ s = fixture_session()
jill = s.query(Person).get(3)
s.query(Person).update(
@@ -621,7 +622,7 @@ class BulkUpdateTest(fixtures.DeclarativeMappedTest, AssertsCompiledSQL):
def test_evaluate_hybrid_attr_plain(self):
Person = self.classes.Person
- s = Session()
+ s = fixture_session()
jill = s.query(Person).get(3)
s.query(Person).update(
@@ -632,7 +633,7 @@ class BulkUpdateTest(fixtures.DeclarativeMappedTest, AssertsCompiledSQL):
def test_fetch_hybrid_attr_indirect(self):
Person = self.classes.Person
- s = Session()
+ s = fixture_session()
jill = s.query(Person).get(3)
s.query(Person).update(
@@ -643,7 +644,7 @@ class BulkUpdateTest(fixtures.DeclarativeMappedTest, AssertsCompiledSQL):
def test_fetch_hybrid_attr_plain(self):
Person = self.classes.Person
- s = Session()
+ s = fixture_session()
jill = s.query(Person).get(3)
s.query(Person).update(
@@ -654,7 +655,7 @@ class BulkUpdateTest(fixtures.DeclarativeMappedTest, AssertsCompiledSQL):
def test_evaluate_hybrid_attr_w_update_expr(self):
Person = self.classes.Person
- s = Session()
+ s = fixture_session()
jill = s.query(Person).get(3)
s.query(Person).update(
@@ -665,7 +666,7 @@ class BulkUpdateTest(fixtures.DeclarativeMappedTest, AssertsCompiledSQL):
def test_fetch_hybrid_attr_w_update_expr(self):
Person = self.classes.Person
- s = Session()
+ s = fixture_session()
jill = s.query(Person).get(3)
s.query(Person).update(
@@ -676,7 +677,7 @@ class BulkUpdateTest(fixtures.DeclarativeMappedTest, AssertsCompiledSQL):
def test_evaluate_hybrid_attr_indirect_w_update_expr(self):
Person = self.classes.Person
- s = Session()
+ s = fixture_session()
jill = s.query(Person).get(3)
s.query(Person).update(
@@ -813,7 +814,7 @@ class SpecialObjectTest(fixtures.TestBase, AssertsCompiledSQL):
def test_query_one(self):
BankAccount, Amount = self.BankAccount, self.Amount
- session = Session()
+ session = fixture_session()
query = session.query(BankAccount).filter(
BankAccount.balance == Amount(10000, "cad")
@@ -829,7 +830,7 @@ class SpecialObjectTest(fixtures.TestBase, AssertsCompiledSQL):
def test_query_two(self):
BankAccount, Amount = self.BankAccount, self.Amount
- session = Session()
+ session = fixture_session()
# alternatively we can do the calc on the DB side.
query = (
@@ -858,7 +859,7 @@ class SpecialObjectTest(fixtures.TestBase, AssertsCompiledSQL):
def test_query_three(self):
BankAccount = self.BankAccount
- session = Session()
+ session = fixture_session()
query = session.query(BankAccount).filter(
BankAccount.balance.as_currency("cad")
@@ -879,7 +880,7 @@ class SpecialObjectTest(fixtures.TestBase, AssertsCompiledSQL):
def test_query_four(self):
BankAccount = self.BankAccount
- session = Session()
+ session = fixture_session()
# 4c. query all amounts, converting to "CAD" on the DB side
query = session.query(BankAccount.balance.as_currency("cad").amount)
@@ -892,7 +893,7 @@ class SpecialObjectTest(fixtures.TestBase, AssertsCompiledSQL):
def test_query_five(self):
BankAccount = self.BankAccount
- session = Session()
+ session = fixture_session()
# 4d. average balance in EUR
query = session.query(func.avg(BankAccount.balance.as_currency("eur")))