mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-06-04 06:48:27 -04:00
Mike Bayer
4cd1902796
features keynames which prefer mapped attribute names over column keys, column keys over column names, i.e. Query(Class.foo, Class.bar) will have names "foo" and "bar" even if those are not the names of the underlying Column objects. Direct Column objects such as Query(table.c.col) will return the "key" attribute of the Column.
203 lines
7.4 KiB
Python
203 lines
7.4 KiB
Python
import testenv; testenv.configure_for_tests()
|
|
from sqlalchemy import *
|
|
from sqlalchemy.orm import *
|
|
from testlib import *
|
|
from testlib.fixtures import Base
|
|
from orm._base import MappedTest, ComparableEntity
|
|
|
|
class SingleInheritanceTest(MappedTest):
|
|
def define_tables(self, metadata):
|
|
global employees_table
|
|
employees_table = Table('employees', metadata,
|
|
Column('employee_id', Integer, primary_key=True),
|
|
Column('name', String(50)),
|
|
Column('manager_data', String(50)),
|
|
Column('engineer_info', String(50)),
|
|
Column('type', String(20))
|
|
)
|
|
|
|
def setup_classes(self):
|
|
class Employee(ComparableEntity):
|
|
pass
|
|
class Manager(Employee):
|
|
pass
|
|
class Engineer(Employee):
|
|
pass
|
|
class JuniorEngineer(Engineer):
|
|
pass
|
|
|
|
@testing.resolve_artifact_names
|
|
def setup_mappers(self):
|
|
mapper(Employee, employees_table, polymorphic_on=employees_table.c.type)
|
|
mapper(Manager, inherits=Employee, polymorphic_identity='manager')
|
|
mapper(Engineer, inherits=Employee, polymorphic_identity='engineer')
|
|
mapper(JuniorEngineer, inherits=Engineer, polymorphic_identity='juniorengineer')
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_single_inheritance(self):
|
|
|
|
session = create_session()
|
|
|
|
m1 = Manager(name='Tom', manager_data='knows how to manage things')
|
|
e1 = Engineer(name='Kurt', engineer_info='knows how to hack')
|
|
e2 = JuniorEngineer(name='Ed', engineer_info='oh that ed')
|
|
session.add_all([m1, e1, e2])
|
|
session.flush()
|
|
|
|
assert session.query(Employee).all() == [m1, e1, e2]
|
|
assert session.query(Engineer).all() == [e1, e2]
|
|
assert session.query(Manager).all() == [m1]
|
|
assert session.query(JuniorEngineer).all() == [e2]
|
|
|
|
m1 = session.query(Manager).one()
|
|
session.expire(m1, ['manager_data'])
|
|
self.assertEquals(m1.manager_data, "knows how to manage things")
|
|
|
|
row = session.query(Engineer.name, Engineer.employee_id).filter(Engineer.name=='Kurt').first()
|
|
assert row.name == 'Kurt'
|
|
assert row.employee_id == e1.employee_id
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_multi_qualification(self):
|
|
session = create_session()
|
|
|
|
m1 = Manager(name='Tom', manager_data='knows how to manage things')
|
|
e1 = Engineer(name='Kurt', engineer_info='knows how to hack')
|
|
e2 = JuniorEngineer(name='Ed', engineer_info='oh that ed')
|
|
|
|
session.add_all([m1, e1, e2])
|
|
session.flush()
|
|
|
|
ealias = aliased(Engineer)
|
|
self.assertEquals(
|
|
session.query(Manager, ealias).all(),
|
|
[(m1, e1), (m1, e2)]
|
|
)
|
|
|
|
self.assertEquals(
|
|
session.query(Manager.name).all(),
|
|
[("Tom",)]
|
|
)
|
|
|
|
self.assertEquals(
|
|
session.query(Manager.name, ealias.name).all(),
|
|
[("Tom", "Kurt"), ("Tom", "Ed")]
|
|
)
|
|
|
|
self.assertEquals(
|
|
session.query(func.upper(Manager.name), func.upper(ealias.name)).all(),
|
|
[("TOM", "KURT"), ("TOM", "ED")]
|
|
)
|
|
|
|
self.assertEquals(
|
|
session.query(Manager).add_entity(ealias).all(),
|
|
[(m1, e1), (m1, e2)]
|
|
)
|
|
|
|
self.assertEquals(
|
|
session.query(Manager.name).add_column(ealias.name).all(),
|
|
[("Tom", "Kurt"), ("Tom", "Ed")]
|
|
)
|
|
|
|
# TODO: I think raise error on this for now
|
|
# self.assertEquals(
|
|
# session.query(Employee.name, Manager.manager_data, Engineer.engineer_info).all(),
|
|
# []
|
|
# )
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_select_from(self):
|
|
sess = create_session()
|
|
m1 = Manager(name='Tom', manager_data='data1')
|
|
m2 = Manager(name='Tom2', manager_data='data2')
|
|
e1 = Engineer(name='Kurt', engineer_info='knows how to hack')
|
|
e2 = JuniorEngineer(name='Ed', engineer_info='oh that ed')
|
|
sess.add_all([m1, m2, e1, e2])
|
|
sess.flush()
|
|
|
|
self.assertEquals(
|
|
sess.query(Manager).select_from(employees_table.select().limit(10)).all(),
|
|
[m1, m2]
|
|
)
|
|
|
|
@testing.resolve_artifact_names
|
|
def test_count(self):
|
|
sess = create_session()
|
|
m1 = Manager(name='Tom', manager_data='data1')
|
|
m2 = Manager(name='Tom2', manager_data='data2')
|
|
e1 = Engineer(name='Kurt', engineer_info='data3')
|
|
e2 = JuniorEngineer(name='marvin', engineer_info='data4')
|
|
sess.add_all([m1, m2, e1, e2])
|
|
sess.flush()
|
|
|
|
self.assertEquals(sess.query(Manager).count(), 2)
|
|
self.assertEquals(sess.query(Engineer).count(), 2)
|
|
self.assertEquals(sess.query(Employee).count(), 4)
|
|
|
|
self.assertEquals(sess.query(Manager).filter(Manager.name.like('%m%')).count(), 2)
|
|
self.assertEquals(sess.query(Employee).filter(Employee.name.like('%m%')).count(), 3)
|
|
|
|
class SingleOnJoinedTest(ORMTest):
|
|
def define_tables(self, metadata):
|
|
global persons_table, employees_table
|
|
|
|
persons_table = Table('persons', metadata,
|
|
Column('person_id', Integer, primary_key=True),
|
|
Column('name', String(50)),
|
|
Column('type', String(20), nullable=False)
|
|
)
|
|
|
|
employees_table = Table('employees', metadata,
|
|
Column('person_id', Integer, ForeignKey('persons.person_id'),primary_key=True),
|
|
Column('employee_data', String(50)),
|
|
Column('manager_data', String(50)),
|
|
)
|
|
|
|
def test_single_on_joined(self):
|
|
class Person(Base):
|
|
pass
|
|
class Employee(Person):
|
|
pass
|
|
class Manager(Employee):
|
|
pass
|
|
|
|
mapper(Person, persons_table, polymorphic_on=persons_table.c.type, polymorphic_identity='person')
|
|
mapper(Employee, employees_table, inherits=Person,polymorphic_identity='engineer')
|
|
mapper(Manager, inherits=Employee,polymorphic_identity='manager')
|
|
|
|
sess = create_session()
|
|
sess.save(Person(name='p1'))
|
|
sess.save(Employee(name='e1', employee_data='ed1'))
|
|
sess.save(Manager(name='m1', employee_data='ed2', manager_data='md1'))
|
|
sess.flush()
|
|
sess.clear()
|
|
|
|
self.assertEquals(sess.query(Person).order_by(Person.person_id).all(), [
|
|
Person(name='p1'),
|
|
Employee(name='e1', employee_data='ed1'),
|
|
Manager(name='m1', employee_data='ed2', manager_data='md1')
|
|
])
|
|
sess.clear()
|
|
|
|
self.assertEquals(sess.query(Employee).order_by(Person.person_id).all(), [
|
|
Employee(name='e1', employee_data='ed1'),
|
|
Manager(name='m1', employee_data='ed2', manager_data='md1')
|
|
])
|
|
sess.clear()
|
|
|
|
self.assertEquals(sess.query(Manager).order_by(Person.person_id).all(), [
|
|
Manager(name='m1', employee_data='ed2', manager_data='md1')
|
|
])
|
|
sess.clear()
|
|
|
|
def go():
|
|
self.assertEquals(sess.query(Person).with_polymorphic('*').order_by(Person.person_id).all(), [
|
|
Person(name='p1'),
|
|
Employee(name='e1', employee_data='ed1'),
|
|
Manager(name='m1', employee_data='ed2', manager_data='md1')
|
|
])
|
|
self.assert_sql_count(testing.db, go, 1)
|
|
|
|
if __name__ == '__main__':
|
|
testenv.main()
|