summaryrefslogtreecommitdiff
path: root/test/orm/inheritance/_poly_fixtures.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/orm/inheritance/_poly_fixtures.py')
-rw-r--r--test/orm/inheritance/_poly_fixtures.py10
1 files changed, 4 insertions, 6 deletions
diff --git a/test/orm/inheritance/_poly_fixtures.py b/test/orm/inheritance/_poly_fixtures.py
index 77564bcdb..4253b4bee 100644
--- a/test/orm/inheritance/_poly_fixtures.py
+++ b/test/orm/inheritance/_poly_fixtures.py
@@ -2,9 +2,9 @@ from sqlalchemy import ForeignKey
from sqlalchemy import Integer
from sqlalchemy import String
from sqlalchemy import util
-from sqlalchemy.orm import create_session
from sqlalchemy.orm import polymorphic_union
from sqlalchemy.orm import relationship
+from sqlalchemy.orm import sessionmaker
from sqlalchemy.sql.selectable import LABEL_STYLE_TABLENAME_PLUS_COL
from sqlalchemy.testing import AssertsCompiledSQL
from sqlalchemy.testing import config
@@ -221,11 +221,9 @@ class _PolymorphicFixtureBase(fixtures.MappedTest, AssertsCompiledSQL):
cls.c2 = c2 = Company(name="Elbonia, Inc.")
c2.employees = [e3]
- sess = create_session(connection)
- sess.add(c1)
- sess.add(c2)
- sess.flush()
- sess.expunge_all()
+ with sessionmaker(connection, expire_on_commit=False).begin() as sess:
+ sess.add(c1)
+ sess.add(c2)
cls.all_employees = [e1, e2, b1, m1, e3]
cls.c1_employees = [e1, e2, b1, m1]