def test_query_with_joinedload(self, session):
query = session.query(Foo).options(joinedload(Foo.bar))
entities = get_query_models(query)
# Bar is not added to the query since the joinedload is transparent
assert {'Foo': Foo} == entities
def test_query_with_multiple_joins(self, session):
query = session.query(Foo).join(Bar).join(Qux, Bar.id == Qux.id)
entities = get_query_models(query)
assert {'Foo': Foo, 'Bar': Bar, 'Qux': Qux} == entities
def test_query_with_aggregate_func(self, session):
query = session.query(func.count(Foo.id))
entities = get_query_models(query)
assert {'Foo': Foo} == entities
def test_query_with_join(self, session):
query = session.query(Foo).join(Bar)
entities = get_query_models(query)
assert {'Foo': Foo, 'Bar': Bar} == entities
def test_query_with_one_field(self, session):
query = session.query(Foo.id)
entities = get_query_models(query)
assert {'Foo': Foo} == entities
def test_query_with_multiple_fields(self, session):
query = session.query(Foo.id, Bar.id, Bar.name)
entities = get_query_models(query)
assert {'Foo': Foo, 'Bar': Bar} == entities
def test_query_with_duplicated_models(self, session):
query = session.query(Bar, Qux, Bar)
entities = get_query_models(query)
assert {'Bar': Bar, 'Qux': Qux} == entities
def test_query_with_multiple_models(self, session):
query = session.query(Bar, Qux)
entities = get_query_models(query)
assert {'Bar': Bar, 'Qux': Qux} == entities
def test_query_with_one_model(self, session):
query = session.query(Bar)
entities = get_query_models(query)
assert {'Bar': Bar} == entities
def test_query_with_no_models(self, session):
query = session.query()
entities = get_query_models(query)
assert {} == entities
def test_query_with_select_from_model(self, session):
query = session.query().select_from(Bar)
entities = get_query_models(query)
assert {'Bar': Bar} == entities
