from sqlalchemy import desc from sqlalchemy import exc as sa_exc from sqlalchemy import exists from sqlalchemy import func from sqlalchemy import select from sqlalchemy import testing from sqlalchemy import true from sqlalchemy.orm import aliased from sqlalchemy.orm import defaultload from sqlalchemy.orm import join from sqlalchemy.orm import joinedload from sqlalchemy.orm import selectinload from sqlalchemy.orm import subqueryload from sqlalchemy.orm import with_parent from sqlalchemy.orm import with_polymorphic from sqlalchemy.testing import assert_raises from sqlalchemy.testing import eq_ from sqlalchemy.testing.assertsql import CompiledSQL from sqlalchemy.testing.fixtures import fixture_session from ._poly_fixtures import _Polymorphic from ._poly_fixtures import _PolymorphicAliasedJoins from ._poly_fixtures import _PolymorphicJoins from ._poly_fixtures import _PolymorphicPolymorphic from ._poly_fixtures import _PolymorphicUnions from ._poly_fixtures import Boss from ._poly_fixtures import Company from ._poly_fixtures import Engineer from ._poly_fixtures import Machine from ._poly_fixtures import Manager from ._poly_fixtures import Paperwork from ._poly_fixtures import Person class _PolymorphicTestBase: __backend__ = True __dialect__ = "default_enhanced" @classmethod def setup_mappers(cls): super().setup_mappers() global people, engineers, managers, boss global companies, paperwork, machines people, engineers, managers, boss, companies, paperwork, machines = ( cls.tables.people, cls.tables.engineers, cls.tables.managers, cls.tables.boss, cls.tables.companies, cls.tables.paperwork, cls.tables.machines, ) @classmethod def insert_data(cls, connection): super().insert_data(connection) global all_employees, c1_employees, c2_employees global c1, c2, e1, e2, e3, b1, m1 c1, c2, all_employees, c1_employees, c2_employees = ( cls.c1, cls.c2, cls.all_employees, cls.c1_employees, cls.c2_employees, ) e1, e2, e3, b1, m1 = cls.e1, cls.e2, cls.e3, cls.b1, cls.m1 @testing.requires.ctes def test_cte_clone_issue(self): """test #8357""" sess = fixture_session() cte = select(Engineer.person_id).cte(name="test_cte") stmt = ( select(Engineer) .where(exists().where(Engineer.person_id == cte.c.person_id)) .where(exists().where(Engineer.person_id == cte.c.person_id)) ).order_by(Engineer.person_id) self.assert_compile( stmt, "WITH test_cte AS (SELECT engineers.person_id AS person_id " "FROM people JOIN engineers ON people.person_id = " "engineers.person_id) SELECT engineers.person_id, " "people.person_id AS person_id_1, people.company_id, " "people.name, people.type, engineers.status, " "engineers.engineer_name, engineers.primary_language FROM people " "JOIN engineers ON people.person_id = engineers.person_id WHERE " "(EXISTS (SELECT * FROM test_cte WHERE engineers.person_id = " "test_cte.person_id)) AND (EXISTS (SELECT * FROM test_cte " "WHERE engineers.person_id = test_cte.person_id)) " "ORDER BY engineers.person_id", ) result = sess.scalars(stmt) eq_( result.all(), [ Engineer(name="dilbert"), Engineer(name="wally"), Engineer(name="vlad"), ], ) def test_loads_at_once(self): """ Test that all objects load from the full query, when with_polymorphic is used. """ sess = fixture_session() def go(): eq_( sess.query(Person).order_by(Person.person_id).all(), all_employees, ) count = {"": 14, "Polymorphic": 9}.get(self.select_type, 10) self.assert_sql_count(testing.db, go, count) def test_primary_eager_aliasing_joinedload(self): # For both joinedload() and subqueryload(), if the original q is # not loading the subclass table, the joinedload doesn't happen. sess = fixture_session() def go(): eq_( sess.query(Person) .order_by(Person.person_id) .options(joinedload(Engineer.machines))[1:3], all_employees[1:3], ) count = {"": 6, "Polymorphic": 3}.get(self.select_type, 4) self.assert_sql_count(testing.db, go, count) def test_primary_eager_aliasing_subqueryload(self): # test that subqueryload does not occur because the parent # row cannot support it sess = fixture_session() def go(): eq_( sess.query(Person) .order_by(Person.person_id) .options(subqueryload(Engineer.machines)) .all(), all_employees, ) count = {"": 14, "Polymorphic": 7}.get(self.select_type, 8) self.assert_sql_count(testing.db, go, count) def test_primary_eager_aliasing_selectinload(self): # test that selectinload does not occur because the parent # row cannot support it sess = fixture_session() def go(): eq_( sess.query(Person) .order_by(Person.person_id) .options(selectinload(Engineer.machines)) .all(), all_employees, ) count = {"": 14, "Polymorphic": 7}.get(self.select_type, 8) self.assert_sql_count(testing.db, go, count) def test_primary_eager_aliasing_three_reset_selectable(self): """test now related to #7262 See test_primary_eager_aliasing_three_dont_reset_selectable for the non-reset selectable version. """ # assert the JOINs don't over JOIN sess = fixture_session() # note selectable=None wp = with_polymorphic(Person, "*", None) def go(): eq_( sess.query(wp) .order_by(wp.person_id) .options(joinedload(wp.Engineer.machines))[1:3], all_employees[1:3], ) self.assert_sql_count(testing.db, go, 3) eq_( sess.scalar( select(func.count("*")).select_from( sess.query(wp) .options(joinedload(wp.Engineer.machines)) .order_by(wp.person_id) .limit(2) .offset(1) .subquery() ) ), 2, ) def test_get_one(self): """ For all mappers, ensure the primary key has been calculated as just the "person_id" column. """ sess = fixture_session() eq_( sess.get(Person, e1.person_id), Engineer(name="dilbert", primary_language="java"), ) def test_get_two(self): sess = fixture_session() eq_( sess.get(Engineer, e1.person_id), Engineer(name="dilbert", primary_language="java"), ) def test_get_three(self): sess = fixture_session() eq_( sess.get(Manager, b1.person_id), Boss(name="pointy haired boss", golf_swing="fore"), ) def test_lazyload_related_w_cache_check(self): sess = fixture_session() c1 = sess.get(Company, 1) c2 = sess.get(Company, 2) q1 = ( sess.query(Person) .filter(with_parent(c1, Company.employees)) .order_by(Person.person_id) ) eq_( q1.all(), [ Engineer(name="dilbert"), Engineer(name="wally"), Boss(name="pointy haired boss"), Manager(name="dogbert"), ], ) q2 = ( sess.query(Person) .filter(with_parent(c2, Company.employees)) .order_by(Person.person_id) ) eq_(q2.all(), [Engineer(name="vlad")]) def test_multi_join(self): sess = fixture_session() e = aliased(Person) c = aliased(Company) q = ( sess.query(Company, Person, c, e) .join(Person, Company.employees) .join(e, c.employees) .filter(Person.person_id != e.person_id) .filter(Person.name == "dilbert") .filter(e.name == "wally") ) eq_(q.count(), 1) eq_( q.all(), [ ( Company(company_id=1, name="MegaCorp, Inc."), Engineer( status="regular engineer", engineer_name="dilbert", name="dilbert", company_id=1, primary_language="java", person_id=1, type="engineer", ), Company(company_id=1, name="MegaCorp, Inc."), Engineer( status="regular engineer", engineer_name="wally", name="wally", company_id=1, primary_language="c++", person_id=2, type="engineer", ), ) ], ) def test_multi_join_future(self): sess = fixture_session(future=True) e = aliased(Person) c = aliased(Company) q = ( select(Company, Person, c, e) .join(Person, Company.employees) .join(e, c.employees) .filter(Person.person_id != e.person_id) .filter(Person.name == "dilbert") .filter(e.name == "wally") ) eq_( sess.execute( select(func.count()).select_from(q.subquery()) ).scalar(), 1, ) eq_( sess.execute(q).all(), [ ( Company(company_id=1, name="MegaCorp, Inc."), Engineer( status="regular engineer", engineer_name="dilbert", name="dilbert", company_id=1, primary_language="java", person_id=1, type="engineer", ), Company(company_id=1, name="MegaCorp, Inc."), Engineer( status="regular engineer", engineer_name="wally", name="wally", company_id=1, primary_language="c++", person_id=2, type="engineer", ), ) ], ) def test_filter_on_subclass_one(self): sess = fixture_session() eq_(sess.query(Engineer).all()[0], Engineer(name="dilbert")) def test_filter_on_subclass_one_future(self): sess = fixture_session(future=True) eq_( sess.execute(select(Engineer)).scalar(), Engineer(name="dilbert"), ) def test_filter_on_subclass_two(self): sess = fixture_session() eq_(sess.query(Engineer).first(), Engineer(name="dilbert")) def test_filter_on_subclass_three(self): sess = fixture_session() eq_( sess.query(Engineer) .filter(Engineer.person_id == e1.person_id) .first(), Engineer(name="dilbert"), ) def test_filter_on_subclass_four(self): sess = fixture_session() eq_( sess.query(Manager) .filter(Manager.person_id == m1.person_id) .one(), Manager(name="dogbert"), ) def test_filter_on_subclass_five(self): sess = fixture_session() eq_( sess.query(Manager) .filter(Manager.person_id == b1.person_id) .one(), Boss(name="pointy haired boss"), ) def test_filter_on_subclass_six(self): sess = fixture_session() eq_( sess.query(Boss).filter(Boss.person_id == b1.person_id).one(), Boss(name="pointy haired boss"), ) def test_join_from_polymorphic_nonaliased_one(self): sess = fixture_session() eq_( sess.query(Person) .join(Person.paperwork) .filter(Paperwork.description.like("%review%")) .all(), [b1, m1], ) def test_join_from_polymorphic_nonaliased_one_future(self): sess = fixture_session(future=True) eq_( sess.execute( select(Person) .join(Person.paperwork) .filter(Paperwork.description.like("%review%")) ) .unique() .scalars() .all(), [b1, m1], ) def test_join_from_polymorphic_nonaliased_two(self): sess = fixture_session() eq_( sess.query(Person) .order_by(Person.person_id) .join(Person.paperwork) .filter(Paperwork.description.like("%#2%")) .all(), [e1, m1], ) def test_join_from_polymorphic_nonaliased_three(self): sess = fixture_session() eq_( sess.query(Engineer) .order_by(Person.person_id) .join(Person.paperwork) .filter(Paperwork.description.like("%#2%")) .all(), [e1], ) def test_join_from_polymorphic_nonaliased_four(self): sess = fixture_session() eq_( sess.query(Person) .order_by(Person.person_id) .join(Person.paperwork) .filter(Person.name.like("%dog%")) .filter(Paperwork.description.like("%#2%")) .all(), [m1], ) def test_join_from_polymorphic_aliased_one_future(self): sess = fixture_session(future=True) pa = aliased(Paperwork) eq_( sess.execute( select(Person) .order_by(Person.person_id) .join(Person.paperwork.of_type(pa)) .filter(pa.description.like("%review%")) ) .unique() .scalars() .all(), [b1, m1], ) def test_join_from_polymorphic_explicit_aliased_one(self): sess = fixture_session() pa = aliased(Paperwork) eq_( sess.query(Person) .order_by(Person.person_id) .join(pa, Person.paperwork) .filter(pa.description.like("%review%")) .all(), [b1, m1], ) def test_join_from_polymorphic_explicit_aliased_two(self): sess = fixture_session() pa = aliased(Paperwork) eq_( sess.query(Person) .order_by(Person.person_id) .join(pa, Person.paperwork) .filter(pa.description.like("%#2%")) .all(), [e1, m1], ) def test_join_from_polymorphic_explicit_aliased_three(self): sess = fixture_session() pa = aliased(Paperwork) eq_( sess.query(Engineer) .order_by(Person.person_id) .join(pa, Person.paperwork) .filter(pa.description.like("%#2%")) .all(), [e1], ) def test_join_from_polymorphic_aliased_four(self): sess = fixture_session() pa = aliased(Paperwork) eq_( sess.query(Person) .order_by(Person.person_id) .join(pa, Person.paperwork) .filter(Person.name.like("%dog%")) .filter(pa.description.like("%#2%")) .all(), [m1], ) def test_join_from_with_polymorphic_nonaliased_one_future(self): sess = fixture_session(future=True) pm = with_polymorphic(Person, [Manager]) eq_( sess.execute( select(pm) .order_by(pm.person_id) .join(pm.paperwork) .filter(Paperwork.description.like("%review%")) ) .unique() .scalars() .all(), [b1, m1], ) def test_join_from_with_polymorphic_nonaliased_two_future(self): sess = fixture_session() wp = with_polymorphic(Person, [Manager, Engineer]) eq_( sess.query(wp) .order_by(wp.person_id) .join(wp.paperwork) .filter(Paperwork.description.like("%#2%")) .all(), [e1, m1], ) def test_join_from_with_polymorphic_nonaliased_three_future(self): sess = fixture_session() wp = with_polymorphic(Person, [Manager, Engineer]) eq_( sess.query(wp) .order_by(wp.person_id) .join(wp.paperwork) .filter(wp.name.like("%dog%")) .filter(Paperwork.description.like("%#2%")) .all(), [m1], ) def test_join_from_with_polymorphic_explicit_aliased_one_future(self): sess = fixture_session() pa = aliased(Paperwork) wp = with_polymorphic(Person, [Manager]) eq_( sess.query(wp) .join(pa, wp.paperwork) .filter(pa.description.like("%review%")) .all(), [b1, m1], ) def test_join_from_with_polymorphic_explicit_aliased_two_future(self): sess = fixture_session() pa = aliased(Paperwork) wp = with_polymorphic(Person, [Manager, Engineer]) eq_( sess.query(wp) .order_by(wp.person_id) .join(pa, wp.paperwork) .filter(pa.description.like("%#2%")) .all(), [e1, m1], ) def test_join_from_with_polymorphic_ot_explicit_aliased_two_future(self): sess = fixture_session() pa = aliased(Paperwork) wp = with_polymorphic(Person, [Manager, Engineer]) eq_( sess.query(wp) .order_by(wp.person_id) .join(wp.paperwork.of_type(pa)) .filter(pa.description.like("%#2%")) .all(), [e1, m1], ) def test_join_from_with_polymorphic_aliased_three_future(self): sess = fixture_session() pa = aliased(Paperwork) wp = with_polymorphic(Person, [Manager, Engineer]) eq_( sess.query(wp) .order_by(wp.person_id) .join(pa, wp.paperwork) .filter(wp.name.like("%dog%")) .filter(pa.description.like("%#2%")) .all(), [m1], ) def test_join_from_with_polymorphic_ot_aliased_three_future(self): sess = fixture_session() pa = aliased(Paperwork) wp = with_polymorphic(Person, [Manager, Engineer]) eq_( sess.query(wp) .order_by(wp.person_id) .join(wp.paperwork.of_type(pa)) .filter(wp.name.like("%dog%")) .filter(pa.description.like("%#2%")) .all(), [m1], ) def test_join_to_polymorphic_nonaliased(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .filter(Person.name == "vlad") .one(), c2, ) def test_join_to_polymorphic_explicit_aliased(self): sess = fixture_session() ea = aliased(Person) eq_( sess.query(Company) .join(ea, Company.employees) .filter(ea.name == "vlad") .one(), c2, ) def test_polymorphic_any_one(self): sess = fixture_session() any_ = Company.employees.any(Person.name == "vlad") eq_(sess.query(Company).filter(any_).all(), [c2]) def test_polymorphic_any_explicit_alias_two(self): sess = fixture_session() # test that the aliasing on "Person" does not bleed into the # EXISTS clause generated by any() any_ = Company.employees.any(Person.name == "wally") ea = aliased(Person) eq_( sess.query(Company) .join(ea, Company.employees) .filter(ea.name == "dilbert") .filter(any_) .all(), [c1], ) def test_polymorphic_any_three(self): sess = fixture_session() any_ = Company.employees.any(Person.name == "vlad") ea = aliased(Person) eq_( sess.query(Company) .join(ea, Company.employees) .filter(ea.name == "dilbert") .filter(any_) .all(), [], ) def test_polymorphic_any_eight(self): sess = fixture_session() any_ = Engineer.machines.any(Machine.name == "Commodore 64") eq_( sess.query(Person).order_by(Person.person_id).filter(any_).all(), [e2, e3], ) def test_polymorphic_any_nine(self): sess = fixture_session() any_ = Person.paperwork.any(Paperwork.description == "review #2") eq_( sess.query(Person).order_by(Person.person_id).filter(any_).all(), [m1], ) def test_join_from_columns_or_subclass_one(self): sess = fixture_session() expected = [("dogbert",), ("pointy haired boss",)] eq_(sess.query(Manager.name).order_by(Manager.name).all(), expected) def test_join_from_columns_or_subclass_two(self): sess = fixture_session() expected = [("dogbert",), ("dogbert",), ("pointy haired boss",)] eq_( sess.query(Manager.name) .join(Paperwork, Manager.paperwork) .order_by(Manager.name) .all(), expected, ) def test_join_from_columns_or_subclass_three(self): sess = fixture_session() expected = [ ("dilbert",), ("dilbert",), ("dogbert",), ("dogbert",), ("pointy haired boss",), ("vlad",), ("wally",), ("wally",), ] eq_( sess.query(Person.name) .join(Paperwork, Person.paperwork) .order_by(Person.name) .all(), expected, ) def test_join_from_columns_or_subclass_four(self): sess = fixture_session() # Load Person.name, joining from Person -> paperwork, get all # the people. expected = [ ("dilbert",), ("dilbert",), ("dogbert",), ("dogbert",), ("pointy haired boss",), ("vlad",), ("wally",), ("wally",), ] eq_( sess.query(Person.name) .join(paperwork, Person.person_id == paperwork.c.person_id) .order_by(Person.name) .all(), expected, ) def test_join_from_columns_or_subclass_five(self): sess = fixture_session() # same, on manager. get only managers. expected = [("dogbert",), ("dogbert",), ("pointy haired boss",)] eq_( sess.query(Manager.name) .join(paperwork, Manager.person_id == paperwork.c.person_id) .order_by(Person.name) .all(), expected, ) def test_join_from_columns_or_subclass_six(self): sess = fixture_session() if self.select_type == "": # this now raises, due to [ticket:1892]. Manager.person_id # is now the "person_id" column on Manager. SQL is incorrect. assert_raises( sa_exc.DBAPIError, sess.query(Person.name) .join(paperwork, Manager.person_id == paperwork.c.person_id) .order_by(Person.name) .all, ) elif self.select_type == "Unions": # with the union, not something anyone would really be using # here, it joins to the full result set. This is 0.6's # behavior and is more or less wrong. expected = [ ("dilbert",), ("dilbert",), ("dogbert",), ("dogbert",), ("pointy haired boss",), ("vlad",), ("wally",), ("wally",), ] eq_( sess.query(Person.name) .join(paperwork, Manager.person_id == paperwork.c.person_id) .order_by(Person.name) .all(), expected, ) else: # when a join is present and managers.person_id is available, # you get the managers. expected = [("dogbert",), ("dogbert",), ("pointy haired boss",)] eq_( sess.query(Person.name) .join(paperwork, Manager.person_id == paperwork.c.person_id) .order_by(Person.name) .all(), expected, ) def test_join_from_columns_or_subclass_seven(self): sess = fixture_session() eq_( sess.query(Manager) .join(Paperwork, Manager.paperwork) .order_by(Manager.name) .all(), [m1, b1], ) def test_join_from_columns_or_subclass_eight(self): sess = fixture_session() expected = [("dogbert",), ("dogbert",), ("pointy haired boss",)] eq_( sess.query(Manager.name) .join(paperwork, Manager.person_id == paperwork.c.person_id) .order_by(Manager.name) .all(), expected, ) def test_join_from_columns_or_subclass_nine(self): sess = fixture_session() eq_( sess.query(Manager.person_id) .join(paperwork, Manager.person_id == paperwork.c.person_id) .order_by(Manager.name) .all(), [(4,), (4,), (3,)], ) def test_join_from_columns_or_subclass_ten(self): sess = fixture_session() expected = [ ("pointy haired boss", "review #1"), ("dogbert", "review #2"), ("dogbert", "review #3"), ] eq_( sess.query(Manager.name, Paperwork.description) .join(Paperwork, Manager.person_id == Paperwork.person_id) .order_by(Paperwork.paperwork_id) .all(), expected, ) def test_join_from_columns_or_subclass_eleven(self): sess = fixture_session() expected = [("pointy haired boss",), ("dogbert",), ("dogbert",)] malias = aliased(Manager) eq_( sess.query(malias.name) .join(paperwork, malias.person_id == paperwork.c.person_id) .all(), expected, ) def test_subclass_option_pathing(self): sess = fixture_session() dilbert = ( sess.query(Person) .options(defaultload(Engineer.machines).defer(Machine.name)) .filter(Person.name == "dilbert") .first() ) m = dilbert.machines[0] assert "name" not in m.__dict__ eq_(m.name, "IBM ThinkPad") def test_expire(self): """ Test that individual column refresh doesn't get tripped up by the select_table mapper. """ sess = fixture_session() name = "dogbert" m1 = sess.query(Manager).filter(Manager.name == name).one() sess.expire(m1) assert m1.status == "regular manager" name = "pointy haired boss" m2 = sess.query(Manager).filter(Manager.name == name).one() sess.expire(m2, ["manager_name", "golf_swing"]) assert m2.golf_swing == "fore" def test_with_polymorphic_one_future(self): sess = fixture_session() def go(): wp = with_polymorphic(Person, [Engineer]) eq_( sess.query(wp) .filter(wp.Engineer.primary_language == "java") .all(), self._emps_wo_relationships_fixture()[0:1], ) self.assert_sql_count(testing.db, go, 1) def test_with_polymorphic_two_future_adhoc_wp(self): """test #7262 compare to test_with_polymorphic_two_future_default_wp """ sess = fixture_session() def go(): wp = with_polymorphic(Person, "*", selectable=None) eq_( sess.query(wp).order_by(wp.person_id).all(), self._emps_wo_relationships_fixture(), ) self.assert_sql_count(testing.db, go, 1) def test_with_polymorphic_three_future(self, nocache): sess = fixture_session() def go(): wp = with_polymorphic(Person, [Engineer]) eq_( sess.query(wp).order_by(wp.person_id).all(), self._emps_wo_relationships_fixture(), ) self.assert_sql_count(testing.db, go, 3) def test_with_polymorphic_four_future(self): sess = fixture_session() def go(): wp = with_polymorphic( Person, Engineer, selectable=people.outerjoin(engineers) ) eq_( sess.query(wp).order_by(wp.person_id).all(), self._emps_wo_relationships_fixture(), ) self.assert_sql_count(testing.db, go, 3) def test_with_polymorphic_five_future_override_selectable(self): """test part of #7262 this is kind of a hack though, people wouldn't know to do this this way. """ sess = fixture_session() def go(): # needs both [Person] and the selectable=None part # TODO: why do we need [Person] and can't send []? possible # bug wp = with_polymorphic(Person, [Person], selectable=None) # limit the polymorphic join down to just "Person", # overriding select_table eq_( sess.query(wp).all(), self._emps_wo_relationships_fixture(), ) self.assert_sql_count(testing.db, go, 6) def test_with_polymorphic_six_future(self): assert_raises( sa_exc.InvalidRequestError, with_polymorphic, Person, [Paperwork] ) assert_raises( sa_exc.InvalidRequestError, with_polymorphic, Engineer, [Boss] ) assert_raises( sa_exc.InvalidRequestError, with_polymorphic, Engineer, [Person] ) def test_with_polymorphic_seven_future(self): sess = fixture_session() # compare to entities without related collections to prevent # additional lazy SQL from firing on loaded entities wp = with_polymorphic(Person, "*") eq_( sess.query(wp).order_by(wp.person_id).all(), self._emps_wo_relationships_fixture(), ) def test_relationship_to_polymorphic_one(self): expected = self._company_with_emps_machines_fixture() sess = fixture_session() def go(): # test load Companies with lazy load to 'employees' eq_(sess.query(Company).all(), expected) count = {"": 10, "Polymorphic": 5}.get(self.select_type, 6) self.assert_sql_count(testing.db, go, count) def test_relationship_to_polymorphic_two(self): expected = self._company_with_emps_machines_fixture() sess = fixture_session() def go(): # with #2438, of_type() is recognized. This # overrides the with_polymorphic of the mapper # and we get a consistent 3 queries now. eq_( sess.query(Company) .options( joinedload(Company.employees.of_type(Engineer)).joinedload( Engineer.machines ) ) .all(), expected, ) # in the old case, we would get this # count = {'':7, 'Polymorphic':1}.get(self.select_type, 2) # query one is company->Person/Engineer->Machines # query two is managers + boss for row #3 # query three is managers for row #4 count = 3 self.assert_sql_count(testing.db, go, count) def test_relationship_to_polymorphic_three(self): expected = self._company_with_emps_machines_fixture() sess = fixture_session() sess = fixture_session() def go(): eq_( sess.query(Company) .options( subqueryload( Company.employees.of_type(Engineer) ).subqueryload(Engineer.machines) ) .all(), expected, ) # the old case where subqueryload_all # didn't work with of_tyoe # count = { '':8, 'Joins':4, 'Unions':4, 'Polymorphic':3, # 'AliasedJoins':4}[self.select_type] # query one is company->Person/Engineer->Machines # query two is Person/Engineer subq # query three is Machines subq # (however this test can't tell if the Q was a # lazyload or subqload ...) # query four is managers + boss for row #3 # query five is managers for row #4 count = 5 self.assert_sql_count(testing.db, go, count) def test_joinedload_on_subclass(self): sess = fixture_session() expected = [ Engineer( name="dilbert", engineer_name="dilbert", primary_language="java", status="regular engineer", machines=[ Machine(name="IBM ThinkPad"), Machine(name="IPhone"), ], ) ] def go(): # test load People with joinedload to engineers + machines wp = with_polymorphic(Person, "*") eq_( sess.query(wp) .options(joinedload(wp.Engineer.machines)) .filter(wp.name == "dilbert") .all(), expected, ) self.assert_sql_count(testing.db, go, 1) def test_subqueryload_on_subclass(self): sess = fixture_session() expected = [ Engineer( name="dilbert", engineer_name="dilbert", primary_language="java", status="regular engineer", machines=[ Machine(name="IBM ThinkPad"), Machine(name="IPhone"), ], ) ] def go(): wp = with_polymorphic(Person, "*") eq_( sess.query(wp) .options(subqueryload(wp.Engineer.machines)) .filter(wp.name == "dilbert") .all(), expected, ) self.assert_sql_count(testing.db, go, 2) def test_query_subclass_join_to_base_relationship(self): sess = fixture_session() # non-polymorphic eq_(sess.query(Engineer).join(Person.paperwork).all(), [e1, e2, e3]) def test_join_to_subclass_manual_alias(self): sess = fixture_session() target = aliased(Engineer, people.join(engineers)) eq_( sess.query(Company) .join(Company.employees.of_type(target)) .filter(target.primary_language == "java") .all(), [c1], ) def test_join_to_subclass_one(self): sess = fixture_session() eq_( sess.query(Company) .select_from(companies.join(people).join(engineers)) .filter(Engineer.primary_language == "java") .all(), [c1], ) def test_join_to_subclass_three(self): sess = fixture_session() ealias = aliased(Engineer) eq_( sess.query(Company) .join(ealias, Company.employees) .filter(ealias.primary_language == "java") .all(), [c1], ) def test_join_to_subclass_six(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees.of_type(Engineer)) .join(Engineer.machines) .all(), [c1, c2], ) def test_join_to_subclass_six_point_five(self): sess = fixture_session() q = ( sess.query(Company) .join(Company.employees.of_type(Engineer)) .join(Engineer.machines) .filter(Engineer.name == "dilbert") ) self.assert_compile( q, "SELECT companies.company_id AS companies_company_id, " "companies.name AS companies_name FROM companies JOIN " "(people JOIN engineers ON people.person_id = " "engineers.person_id) ON " "companies.company_id = people.company_id " "JOIN machines ON engineers.person_id = machines.engineer_id " "WHERE people.name = :name_1", ) eq_( q.all(), [c1], ) def test_join_to_subclass_eight(self): sess = fixture_session() eq_(sess.query(Person).join(Engineer.machines).all(), [e1, e2, e3]) def test_join_to_subclass_nine(self): sess = fixture_session() eq_( sess.query(Company) .select_from(companies.join(people).join(engineers)) .filter(Engineer.primary_language == "java") .all(), [c1], ) def test_join_to_subclass_ten(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .filter(Engineer.primary_language == "java") .all(), [c1], ) def test_join_to_subclass_eleven(self): sess = fixture_session() eq_( sess.query(Company) .select_from(companies.join(people).join(engineers)) .filter(Engineer.primary_language == "java") .all(), [c1], ) def test_join_to_subclass_twelve(self): sess = fixture_session() eq_(sess.query(Person).join(Engineer.machines).all(), [e1, e2, e3]) def test_join_to_subclass_thirteen(self): sess = fixture_session() eq_( sess.query(Person) .join(Engineer.machines) .filter(Machine.name.ilike("%ibm%")) .all(), [e1, e3], ) def test_join_to_subclass_fourteen(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .join(Engineer.machines) .all(), [c1, c2], ) def test_join_to_subclass_fifteen(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .join(Engineer.machines) .filter(Machine.name.ilike("%thinkpad%")) .all(), [c1], ) def test_join_to_subclass_sixteen(self): sess = fixture_session() # non-polymorphic eq_(sess.query(Engineer).join(Engineer.machines).all(), [e1, e2, e3]) def test_join_to_subclass_seventeen(self): sess = fixture_session() eq_( sess.query(Engineer) .join(Engineer.machines) .filter(Machine.name.ilike("%ibm%")) .all(), [e1, e3], ) def test_join_and_thru_polymorphic_nonaliased_one(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .join(Person.paperwork.and_(Paperwork.description.like("%#2%"))) .all(), [c1], ) def test_join_and_thru_polymorphic_aliased_one(self): sess = fixture_session() ea = aliased(Person) pa = aliased(Paperwork) eq_( sess.query(Company) .join(ea, Company.employees) .join(pa, ea.paperwork.and_(pa.description.like("%#2%"))) .all(), [c1], ) def test_join_through_polymorphic_nonaliased_one(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .join(Person.paperwork) .filter(Paperwork.description.like("%#2%")) .all(), [c1], ) def test_join_through_polymorphic_nonaliased_two(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .join(Person.paperwork) .filter(Paperwork.description.like("%#%")) .all(), [c1, c2], ) def test_join_through_polymorphic_nonaliased_three(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .join(Person.paperwork) .filter(Person.name.in_(["dilbert", "vlad"])) .filter(Paperwork.description.like("%#2%")) .all(), [c1], ) def test_join_through_polymorphic_nonaliased_four(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .join(Person.paperwork) .filter(Person.name.in_(["dilbert", "vlad"])) .filter(Paperwork.description.like("%#%")) .all(), [c1, c2], ) def test_join_through_polymorphic_nonaliased_five(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .filter(Person.name.in_(["dilbert", "vlad"])) .join(Person.paperwork) .filter(Paperwork.description.like("%#2%")) .all(), [c1], ) def test_join_through_polymorphic_nonaliased_six(self): sess = fixture_session() eq_( sess.query(Company) .join(Company.employees) .filter(Person.name.in_(["dilbert", "vlad"])) .join(Person.paperwork) .filter(Paperwork.description.like("%#%")) .all(), [c1, c2], ) def test_join_through_polymorphic_aliased_one(self): sess = fixture_session() ea = aliased(Person) pa = aliased(Paperwork) eq_( sess.query(Company) .join(ea, Company.employees) .join(pa, ea.paperwork) .filter(pa.description.like("%#2%")) .all(), [c1], ) def test_join_through_polymorphic_aliased_two(self): sess = fixture_session() ea = aliased(Person) pa = aliased(Paperwork) eq_( sess.query(Company) .join(ea, Company.employees) .join(pa, ea.paperwork) .filter(pa.description.like("%#%")) .all(), [c1, c2], ) def test_join_through_polymorphic_aliased_three(self): sess = fixture_session() ea = aliased(Person) pa = aliased(Paperwork) eq_( sess.query(Company) .join(ea, Company.employees) .join(pa, ea.paperwork) .filter(ea.name.in_(["dilbert", "vlad"])) .filter(pa.description.like("%#2%")) .all(), [c1], ) def test_join_through_polymorphic_aliased_four(self): sess = fixture_session() ea = aliased(Person) pa = aliased(Paperwork) eq_( sess.query(Company) .join(ea, Company.employees) .join(pa, ea.paperwork) # we can't use "paperwork" here? .filter(ea.name.in_(["dilbert", "vlad"])) .filter(pa.description.like("%#%")) .all(), [c1, c2], ) def test_join_through_polymorphic_aliased_five(self): sess = fixture_session() ea = aliased(Person) pa = aliased(Paperwork) eq_( sess.query(Company) .join(ea, Company.employees) .filter(ea.name.in_(["dilbert", "vlad"])) .join(pa, ea.paperwork) .filter(pa.description.like("%#2%")) .all(), [c1], ) def test_join_through_polymorphic_aliased_six(self): sess = fixture_session() pa = aliased(Paperwork) ea = aliased(Person) eq_( sess.query(Company) .join(ea, Company.employees) .filter(ea.name.in_(["dilbert", "vlad"])) .join(pa, ea.paperwork) .filter(pa.description.like("%#%")) .all(), [c1, c2], ) def test_explicit_polymorphic_join_one(self): sess = fixture_session() # join from Company to Engineer; join condition formulated by # ORMJoin using regular table foreign key connections. Engineer # is expressed as "(select * people join engineers) as anon_1" # so the join is contained. eq_( sess.query(Company) .join(Engineer) .filter(Engineer.engineer_name == "vlad") .one(), c2, ) def test_explicit_polymorphic_join_two(self): sess = fixture_session() # same, using explicit join condition. Query.join() must # adapt the on clause here to match the subquery wrapped around # "people join engineers". eq_( sess.query(Company) .join(Engineer, Company.company_id == Engineer.company_id) .filter(Engineer.engineer_name == "vlad") .one(), c2, ) def test_filter_on_baseclass(self): sess = fixture_session() eq_(sess.query(Person).order_by(Person.person_id).all(), all_employees) eq_( sess.query(Person).order_by(Person.person_id).first(), all_employees[0], ) eq_( sess.query(Person) .order_by(Person.person_id) .filter(Person.person_id == e2.person_id) .one(), e2, ) def test_from_alias(self): sess = fixture_session() palias = aliased(Person) eq_( sess.query(palias) .order_by(palias.person_id) .filter(palias.name.in_(["dilbert", "wally"])) .all(), [e1, e2], ) def test_self_referential_one(self): sess = fixture_session() palias = aliased(Person) expected = [(m1, e1), (m1, e2), (m1, b1)] eq_( sess.query(Person, palias) .filter(Person.company_id == palias.company_id) .filter(Person.name == "dogbert") .filter(Person.person_id > palias.person_id) .order_by(Person.person_id, palias.person_id) .all(), expected, ) def test_self_referential_two_future(self): # TODO: this is the SECOND test *EVER* of an aliased class of # an aliased class. sess = fixture_session(future=True) expected = [(m1, e1), (m1, e2), (m1, b1)] # not aliasing the first class p1 = Person p2 = aliased(Person) stmt = ( select(p1, p2) .filter(p1.company_id == p2.company_id) .filter(p1.name == "dogbert") .filter(p1.person_id > p2.person_id) ) subq = stmt.subquery() pa1 = aliased(p1, subq) pa2 = aliased(p2, subq) stmt2 = select(pa1, pa2).order_by(pa1.person_id, pa2.person_id) eq_( sess.execute(stmt2).unique().all(), expected, ) def test_self_referential_two_point_five_future(self): # TODO: this is the first test *EVER* of an aliased class of # an aliased class. we should add many more tests for this. # new case added in Id810f485c5f7ed971529489b84694e02a3356d6d sess = fixture_session(future=True) expected = [(m1, e1), (m1, e2), (m1, b1)] # aliasing the first class p1 = aliased(Person) p2 = aliased(Person) stmt = ( select(p1, p2) .filter(p1.company_id == p2.company_id) .filter(p1.name == "dogbert") .filter(p1.person_id > p2.person_id) ) subq = stmt.subquery() pa1 = aliased(p1, subq) pa2 = aliased(p2, subq) stmt2 = select(pa1, pa2).order_by(pa1.person_id, pa2.person_id) eq_( sess.execute(stmt2).unique().all(), expected, ) def test_nesting_queries(self): # query.statement places a flag "no_adapt" on the returned # statement. This prevents the polymorphic adaptation in the # second "filter" from hitting it, which would pollute the # subquery and usually results in recursion overflow errors # within the adaption. sess = fixture_session() subq = ( sess.query(engineers.c.person_id) .filter(Engineer.primary_language == "java") .statement.scalar_subquery() ) eq_(sess.query(Person).filter(Person.person_id.in_(subq)).one(), e1) def test_mixed_entities_one(self): sess = fixture_session() expected = [ ( Engineer( status="regular engineer", engineer_name="dilbert", name="dilbert", company_id=1, primary_language="java", person_id=1, type="engineer", ), "MegaCorp, Inc.", ), ( Engineer( status="regular engineer", engineer_name="wally", name="wally", company_id=1, primary_language="c++", person_id=2, type="engineer", ), "MegaCorp, Inc.", ), ( Engineer( status="elbonian engineer", engineer_name="vlad", name="vlad", company_id=2, primary_language="cobol", person_id=5, type="engineer", ), "Elbonia, Inc.", ), ] eq_( sess.query(Engineer, Company.name) .join(Company.employees) .order_by(Person.person_id) .filter(Person.type == "engineer") .all(), expected, ) def _join_to_poly_wp_one(self, sess): wp = with_polymorphic(self.classes.Person, "*") return ( sess.query(wp.name, self.classes.Company.name) .join(self.classes.Company.employees.of_type(wp)) .order_by(wp.person_id) ) def _join_to_poly_wp_two(self, sess): wp = with_polymorphic(self.classes.Person, "*", aliased=True) return ( sess.query(wp.name, self.classes.Company.name) .join(self.classes.Company.employees.of_type(wp)) .order_by(wp.person_id) ) def _join_to_poly_wp_three(self, sess): wp = with_polymorphic( self.classes.Person, "*", aliased=True, flat=True ) return ( sess.query(wp.name, self.classes.Company.name) .join(self.classes.Company.employees.of_type(wp)) .order_by(wp.person_id) ) @testing.combinations( lambda self, sess: ( sess.query(self.classes.Person.name, self.classes.Company.name) .join(self.classes.Company.employees) .order_by(self.classes.Person.person_id) ), _join_to_poly_wp_one, _join_to_poly_wp_two, _join_to_poly_wp_three, ) def test_mixed_entities_join_to_poly(self, q): sess = fixture_session() expected = [ ("dilbert", "MegaCorp, Inc."), ("wally", "MegaCorp, Inc."), ("pointy haired boss", "MegaCorp, Inc."), ("dogbert", "MegaCorp, Inc."), ("vlad", "Elbonia, Inc."), ] eq_( q(self, sess).all(), expected, ) def test_mixed_entities_two(self): sess = fixture_session() expected = [ ("java", "MegaCorp, Inc."), ("cobol", "Elbonia, Inc."), ("c++", "MegaCorp, Inc."), ] eq_( sess.query(Engineer.primary_language, Company.name) .join(Company.employees) .filter(Person.type == "engineer") .order_by(desc(Engineer.primary_language)) .all(), expected, ) def test_mixed_entities_three(self): sess = fixture_session() palias = aliased(Person) expected = [ ( Engineer( status="elbonian engineer", engineer_name="vlad", name="vlad", primary_language="cobol", ), "Elbonia, Inc.", Engineer( status="regular engineer", engineer_name="dilbert", name="dilbert", company_id=1, primary_language="java", person_id=1, type="engineer", ), ) ] eq_( sess.query(Person, Company.name, palias) .join(Company.employees) .filter(Company.name == "Elbonia, Inc.") .filter(palias.name == "dilbert") .filter(palias.person_id != Person.person_id) .all(), expected, ) def test_mixed_entities_four(self): sess = fixture_session() palias = aliased(Person) expected = [ ( Engineer( status="regular engineer", engineer_name="dilbert", name="dilbert", company_id=1, primary_language="java", person_id=1, type="engineer", ), "Elbonia, Inc.", Engineer( status="elbonian engineer", engineer_name="vlad", name="vlad", primary_language="cobol", ), ) ] eq_( sess.query(palias, Company.name, Person) .select_from(join(palias, Company, true())) .join(Company.employees) .filter(Company.name == "Elbonia, Inc.") .filter(palias.name == "dilbert") .all(), expected, ) def test_mixed_entities_five(self): sess = fixture_session() palias = aliased(Person) expected = [("vlad", "Elbonia, Inc.", "dilbert")] eq_( sess.query(Person.name, Company.name, palias.name) .join(Company.employees) .filter(Company.name == "Elbonia, Inc.") .filter(palias.name == "dilbert") .filter(palias.company_id != Person.company_id) .all(), expected, ) def test_mixed_entities_six(self): sess = fixture_session() palias = aliased(Person) expected = [ ("manager", "dogbert", "engineer", "dilbert"), ("manager", "dogbert", "engineer", "wally"), ("manager", "dogbert", "boss", "pointy haired boss"), ] eq_( sess.query(Person.type, Person.name, palias.type, palias.name) .filter(Person.company_id == palias.company_id) .filter(Person.name == "dogbert") .filter(Person.person_id > palias.person_id) .order_by(Person.person_id, palias.person_id) .all(), expected, ) def test_mixed_entities_seven(self): sess = fixture_session() expected = [ ("dilbert", "tps report #1"), ("dilbert", "tps report #2"), ("dogbert", "review #2"), ("dogbert", "review #3"), ("pointy haired boss", "review #1"), ("vlad", "elbonian missive #3"), ("wally", "tps report #3"), ("wally", "tps report #4"), ] eq_( sess.query(Person.name, Paperwork.description) .filter(Person.person_id == Paperwork.person_id) .order_by(Person.name, Paperwork.description) .all(), expected, ) def test_mixed_entities_eight(self): sess = fixture_session() eq_( sess.query(func.count(Person.person_id)) .filter(Engineer.primary_language == "java") .all(), [(1,)], ) def test_mixed_entities_nine(self): sess = fixture_session() expected = [("Elbonia, Inc.", 1), ("MegaCorp, Inc.", 4)] eq_( sess.query(Company.name, func.count(Person.person_id)) .filter(Company.company_id == Person.company_id) .group_by(Company.name) .order_by(Company.name) .all(), expected, ) def test_mixed_entities_ten(self): sess = fixture_session() expected = [("Elbonia, Inc.", 1), ("MegaCorp, Inc.", 4)] eq_( sess.query(Company.name, func.count(Person.person_id)) .join(Company.employees) .group_by(Company.name) .order_by(Company.name) .all(), expected, ) # def test_mixed_entities(self): # sess = fixture_session() # TODO: I think raise error on these for now. different # inheritance/loading schemes have different results here, # all incorrect # # eq_( # sess.query(Person.name, Engineer.primary_language).all(), # []) # def test_mixed_entities(self): # sess = fixture_session() # eq_(sess.query( # Person.name, # Engineer.primary_language, # Manager.manager_name) # .all(), # []) def test_mixed_entities_eleven(self): sess = fixture_session() expected = [("java",), ("c++",), ("cobol",)] eq_( sess.query(Engineer.primary_language) .filter(Person.type == "engineer") .all(), expected, ) def test_mixed_entities_twelve(self): sess = fixture_session() expected = [("vlad", "Elbonia, Inc.")] eq_( sess.query(Person.name, Company.name) .join(Company.employees) .filter(Company.name == "Elbonia, Inc.") .all(), expected, ) def test_mixed_entities_thirteen(self): sess = fixture_session() expected = [("pointy haired boss", "fore")] eq_(sess.query(Boss.name, Boss.golf_swing).all(), expected) def test_mixed_entities_fourteen(self): sess = fixture_session() expected = [("dilbert", "java"), ("wally", "c++"), ("vlad", "cobol")] eq_( sess.query(Engineer.name, Engineer.primary_language).all(), expected, ) def test_mixed_entities_fifteen(self): sess = fixture_session() expected = [ ( "Elbonia, Inc.", Engineer( status="elbonian engineer", engineer_name="vlad", name="vlad", primary_language="cobol", ), ) ] eq_( sess.query(Company.name, Person) .join(Company.employees) .filter(Company.name == "Elbonia, Inc.") .all(), expected, ) def test_mixed_entities_sixteen(self): sess = fixture_session() expected = [ ( Engineer( status="elbonian engineer", engineer_name="vlad", name="vlad", primary_language="cobol", ), "Elbonia, Inc.", ) ] eq_( sess.query(Person, Company.name) .join(Company.employees) .filter(Company.name == "Elbonia, Inc.") .all(), expected, ) def test_mixed_entities_seventeen(self): sess = fixture_session() expected = [("pointy haired boss",), ("dogbert",)] eq_(sess.query(Manager.name).all(), expected) def test_mixed_entities_eighteen(self): sess = fixture_session() expected = [("pointy haired boss foo",), ("dogbert foo",)] eq_(sess.query(Manager.name + " foo").all(), expected) def test_mixed_entities_nineteen(self): sess = fixture_session() row = ( sess.query(Engineer.name, Engineer.primary_language) .filter(Engineer.name == "dilbert") .first() ) assert row.name == "dilbert" assert row.primary_language == "java" def test_correlation_one(self): sess = fixture_session() # this for a long time did not work with PolymorphicAliased and # PolymorphicUnions, which was due to the no_replacement_traverse # annotation added to query.statement which then went into # scalar_subquery(). this is removed as of :ticket:`4304` so now # works. eq_( sess.query(Person.name) .filter( sess.query(Company.name) .filter(Company.company_id == Person.company_id) .correlate(Person) .scalar_subquery() == "Elbonia, Inc." ) .all(), [(e3.name,)], ) def test_correlation_two(self): sess = fixture_session() paliased = aliased(Person) eq_( sess.query(paliased.name) .filter( sess.query(Company.name) .filter(Company.company_id == paliased.company_id) .correlate(paliased) .scalar_subquery() == "Elbonia, Inc." ) .all(), [(e3.name,)], ) def test_correlation_three(self): sess = fixture_session() paliased = aliased(Person, flat=True) eq_( sess.query(paliased.name) .filter( sess.query(Company.name) .filter(Company.company_id == paliased.company_id) .correlate(paliased) .scalar_subquery() == "Elbonia, Inc." ) .all(), [(e3.name,)], ) class PolymorphicTest(_PolymorphicTestBase, _Polymorphic): def test_joined_aliasing_unrelated_subuqery(self): """test #8456""" inner = select(Engineer).where(Engineer.name == "vlad").subquery() crit = select(inner.c.person_id) outer = select(Engineer).where(Engineer.person_id.in_(crit)) # this query will not work at all for any "polymorphic" case # as it will adapt the inner query as well. for those cases, # aliased() has to be used for the inner entity to disambiguate it. self.assert_compile( outer, "SELECT engineers.person_id, people.person_id AS person_id_1, " "people.company_id, people.name, people.type, engineers.status, " "engineers.engineer_name, engineers.primary_language " "FROM people JOIN engineers " "ON people.person_id = engineers.person_id " "WHERE engineers.person_id IN " "(SELECT anon_1.person_id FROM " "(SELECT engineers.person_id AS person_id, " "people.person_id AS person_id_1, " "people.company_id AS company_id, people.name AS name, " "people.type AS type, engineers.status AS status, " "engineers.engineer_name AS engineer_name, " "engineers.primary_language AS primary_language FROM people " "JOIN engineers ON people.person_id = engineers.person_id " "WHERE people.name = :name_1) " "AS anon_1)", ) sess = fixture_session() eq_(sess.scalars(outer).all(), [Engineer(name="vlad")]) def test_primary_eager_aliasing_three_dont_reset_selectable(self): """test now related to #7262 See test_primary_eager_aliasing_three_reset_selectable for the reset selectable version. """ # assert the JOINs don't over JOIN sess = fixture_session() # selectable default is False wp = with_polymorphic(Person, "*") def go(): eq_( sess.query(wp) .order_by(wp.person_id) .options(joinedload(wp.Engineer.machines))[1:3], all_employees[1:3], ) self.assert_sql_count(testing.db, go, 3) eq_( sess.scalar( select(func.count("*")).select_from( sess.query(wp) .options(joinedload(wp.Engineer.machines)) .order_by(wp.person_id) .limit(2) .offset(1) .subquery() ) ), 2, ) def test_with_polymorphic_two_future_default_wp(self): """test #7262 compare to test_with_polymorphic_two_future_adhoc_wp """ sess = fixture_session() def go(): wp = with_polymorphic(Person, "*") eq_( sess.query(wp).order_by(wp.person_id).all(), self._emps_wo_relationships_fixture(), ) self.assert_sql_count(testing.db, go, 1) def test_join_to_subclass_four(self): sess = fixture_session() eq_( sess.query(Person) .select_from(people.join(engineers)) .join(Engineer.machines) .all(), [e1, e2, e3], ) def test_join_to_subclass_five(self): sess = fixture_session() eq_( sess.query(Person) .select_from(people.join(engineers)) .join(Engineer.machines) .filter(Machine.name.ilike("%ibm%")) .all(), [e1, e3], ) def test_correlation_w_polymorphic(self): sess = fixture_session() p_poly = with_polymorphic(Person, "*") eq_( sess.query(p_poly.name) .filter( sess.query(Company.name) .filter(Company.company_id == p_poly.company_id) .correlate(p_poly) .scalar_subquery() == "Elbonia, Inc." ) .all(), [(e3.name,)], ) def test_correlation_w_polymorphic_flat(self): sess = fixture_session() p_poly = with_polymorphic(Person, "*", flat=True) eq_( sess.query(p_poly.name) .filter( sess.query(Company.name) .filter(Company.company_id == p_poly.company_id) .correlate(p_poly) .scalar_subquery() == "Elbonia, Inc." ) .all(), [(e3.name,)], ) def test_join_to_subclass_ten(self): pass def test_mixed_entities_one(self): pass def test_mixed_entities_two(self): pass def test_mixed_entities_eight(self): pass def test_polymorphic_any_eight(self): pass class PolymorphicPolymorphicTest( _PolymorphicTestBase, _PolymorphicPolymorphic ): __dialect__ = "default" def test_with_polymorphic_two_future_default_wp(self): """test #7262 compare to test_with_polymorphic_two_future_adhoc_wp """ sess = fixture_session() def go(): wp = with_polymorphic(Person, "*") eq_( sess.query(wp).order_by(wp.person_id).all(), self._emps_wo_relationships_fixture(), ) self.assert_sql_count(testing.db, go, 1) def test_aliased_not_polluted_by_join(self): # aliased(polymorphic) will normally do the old-school # "(SELECT * FROM a JOIN b ...) AS anon_1" thing. # this is the safest sess = fixture_session() palias = aliased(Person) self.assert_compile( sess.query(palias, Company.name) .order_by(palias.person_id) .join(Person, Company.employees) .filter(palias.name == "dilbert"), "SELECT anon_1.people_person_id AS anon_1_people_person_id, " "anon_1.people_company_id AS anon_1_people_company_id, " "anon_1.people_name AS anon_1_people_name, " "anon_1.people_type AS anon_1_people_type, " "anon_1.engineers_person_id AS anon_1_engineers_person_id, " "anon_1.engineers_status AS anon_1_engineers_status, " "anon_1.engineers_engineer_name AS anon_1_engineers_engineer_name, " # noqa "anon_1.engineers_primary_language AS " "anon_1_engineers_primary_language, " "anon_1.managers_person_id AS anon_1_managers_person_id, " "anon_1.managers_status AS anon_1_managers_status, " "anon_1.managers_manager_name AS anon_1_managers_manager_name, " "anon_1.boss_boss_id AS anon_1_boss_boss_id, " "anon_1.boss_golf_swing AS anon_1_boss_golf_swing, " "companies.name AS companies_name " "FROM companies JOIN " "(people LEFT OUTER JOIN engineers " "ON people.person_id = engineers.person_id " "LEFT OUTER JOIN managers " "ON people.person_id = managers.person_id " "LEFT OUTER JOIN boss ON managers.person_id = boss.boss_id) " "ON companies.company_id = people.company_id, " "(SELECT people.person_id AS people_person_id, " "people.company_id AS people_company_id, " "people.name AS people_name, people.type AS people_type, " "engineers.person_id AS engineers_person_id, " "engineers.status AS engineers_status, " "engineers.engineer_name AS engineers_engineer_name, " "engineers.primary_language AS engineers_primary_language, " "managers.person_id AS managers_person_id, " "managers.status AS managers_status, " "managers.manager_name AS managers_manager_name, " "boss.boss_id AS boss_boss_id, " "boss.golf_swing AS boss_golf_swing " "FROM people LEFT OUTER JOIN engineers " "ON people.person_id = engineers.person_id " "LEFT OUTER JOIN managers " "ON people.person_id = managers.person_id LEFT OUTER JOIN boss " "ON managers.person_id = boss.boss_id) AS anon_1 " "WHERE anon_1.people_name = :people_name_1 " "ORDER BY anon_1.people_person_id", ) def test_flat_aliased_w_select_from(self): sess = fixture_session() palias = aliased(Person, flat=True) self.assert_compile( sess.query(palias, Company.name) .select_from(palias) .order_by(palias.person_id) .join(Person, Company.employees) .filter(palias.name == "dilbert"), "SELECT people_1.person_id AS people_1_person_id, " "people_1.company_id AS people_1_company_id, " "people_1.name AS people_1_name, people_1.type AS people_1_type, " "engineers_1.person_id AS engineers_1_person_id, " "engineers_1.status AS engineers_1_status, " "engineers_1.engineer_name AS engineers_1_engineer_name, " "engineers_1.primary_language AS engineers_1_primary_language, " "managers_1.person_id AS managers_1_person_id, " "managers_1.status AS managers_1_status, " "managers_1.manager_name AS managers_1_manager_name, " "boss_1.boss_id AS boss_1_boss_id, " "boss_1.golf_swing AS boss_1_golf_swing, " "companies.name AS companies_name " "FROM people AS people_1 " "LEFT OUTER JOIN engineers AS engineers_1 " "ON people_1.person_id = engineers_1.person_id " "LEFT OUTER JOIN managers AS managers_1 " "ON people_1.person_id = managers_1.person_id " "LEFT OUTER JOIN boss AS boss_1 " "ON managers_1.person_id = boss_1.boss_id, " "companies JOIN (people LEFT OUTER JOIN engineers " "ON people.person_id = engineers.person_id " "LEFT OUTER JOIN managers " "ON people.person_id = managers.person_id " "LEFT OUTER JOIN boss ON managers.person_id = boss.boss_id) " "ON companies.company_id = people.company_id " "WHERE people_1.name = :name_1 ORDER BY people_1.person_id", ) class PolymorphicUnionsTest(_PolymorphicTestBase, _PolymorphicUnions): @testing.skip_if( lambda: True, "join condition doesn't work w/ this mapping" ) def test_lazyload_related_w_cache_check(self): pass def test_with_polymorphic_two_future_default_wp(self): """test #7262 compare to test_with_polymorphic_two_future_adhoc_wp """ sess = fixture_session() def go(): wp = with_polymorphic(Person, "*") eq_( sess.query(wp).order_by(wp.person_id).all(), self._emps_wo_relationships_fixture(), ) self.assert_sql_count(testing.db, go, 2) def test_subqueryload_on_subclass_uses_path_correctly(self): sess = fixture_session() expected = [ Engineer( name="dilbert", engineer_name="dilbert", primary_language="java", status="regular engineer", machines=[ Machine(name="IBM ThinkPad"), Machine(name="IPhone"), ], ) ] with self.sql_execution_asserter(testing.db) as asserter: wp = with_polymorphic(Person, "*") eq_( sess.query(wp) .options(subqueryload(wp.Engineer.machines)) .filter(wp.name == "dilbert") .all(), expected, ) asserter.assert_( CompiledSQL( "SELECT pjoin.person_id AS pjoin_person_id, " "pjoin.company_id AS pjoin_company_id, " "pjoin.name AS pjoin_name, pjoin.type AS pjoin_type, " "pjoin.status AS pjoin_status, " "pjoin.engineer_name AS pjoin_engineer_name, " "pjoin.primary_language AS pjoin_primary_language, " "pjoin.manager_name AS pjoin_manager_name " "FROM (SELECT engineers.person_id AS person_id, " "people.company_id AS company_id, people.name AS name, " "people.type AS type, engineers.status AS status, " "engineers.engineer_name AS engineer_name, " "engineers.primary_language AS primary_language, " "CAST(NULL AS VARCHAR(50)) AS manager_name " "FROM people JOIN engineers ON people.person_id = " "engineers.person_id UNION ALL SELECT managers.person_id " "AS person_id, people.company_id AS company_id, people.name " "AS name, people.type AS type, managers.status AS status, " "CAST(NULL AS VARCHAR(50)) AS engineer_name, " "CAST(NULL AS VARCHAR(50)) AS primary_language, " "managers.manager_name AS manager_name FROM people " "JOIN managers ON people.person_id = managers.person_id) " "AS pjoin WHERE pjoin.name = :name_1", params=[{"name_1": "dilbert"}], ), CompiledSQL( "SELECT machines.machine_id AS machines_machine_id, " "machines.name AS machines_name, machines.engineer_id " "AS machines_engineer_id, anon_1.pjoin_person_id AS " "anon_1_pjoin_person_id FROM " "(SELECT pjoin.person_id AS pjoin_person_id FROM " "(SELECT engineers.person_id AS person_id, people.company_id " "AS company_id, people.name AS name, " "people.type AS type, engineers.status AS status, " "engineers.engineer_name AS engineer_name, " "engineers.primary_language AS primary_language, " "CAST(NULL AS VARCHAR(50)) AS manager_name FROM people " "JOIN engineers ON people.person_id = engineers.person_id " "UNION ALL SELECT managers.person_id AS person_id, " "people.company_id AS company_id, people.name AS name, " "people.type AS type, managers.status AS status, " "CAST(NULL AS VARCHAR(50)) AS engineer_name, " "CAST(NULL AS VARCHAR(50)) AS primary_language, " "managers.manager_name AS manager_name FROM people " "JOIN managers ON people.person_id = managers.person_id) " "AS pjoin WHERE pjoin.name = :name_1) AS anon_1 JOIN " "machines ON anon_1.pjoin_person_id = machines.engineer_id " "ORDER BY machines.machine_id", params=[{"name_1": "dilbert"}], ), ) class PolymorphicAliasedJoinsTest( _PolymorphicTestBase, _PolymorphicAliasedJoins ): @testing.skip_if( lambda: True, "join condition doesn't work w/ this mapping" ) def test_lazyload_related_w_cache_check(self): pass def test_with_polymorphic_two_future_default_wp(self): """test #7262 compare to test_with_polymorphic_two_future_adhoc_wp """ sess = fixture_session() def go(): wp = with_polymorphic(Person, "*") eq_( sess.query(wp).order_by(wp.person_id).all(), self._emps_wo_relationships_fixture(), ) self.assert_sql_count(testing.db, go, 2) class PolymorphicJoinsTest(_PolymorphicTestBase, _PolymorphicJoins): def test_with_polymorphic_two_future_default_wp(self): """test #7262 compare to test_with_polymorphic_two_future_adhoc_wp """ sess = fixture_session() def go(): wp = with_polymorphic(Person, "*") eq_( sess.query(wp).order_by(wp.person_id).all(), self._emps_wo_relationships_fixture(), ) self.assert_sql_count(testing.db, go, 2) def test_having_group_by(self): sess = fixture_session() eq_( sess.query(Person.name) .group_by(Person.name) .having(Person.name == "dilbert") .all(), [("dilbert",)], )