diff options
author | Mike Bayer <mike_mp@zzzcomputing.com> | 2012-08-05 15:14:51 -0400 |
---|---|---|
committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2012-08-05 15:14:51 -0400 |
commit | 6bd46945ccd585c494eb7550a0dfea22f17727c0 (patch) | |
tree | 4412d12bb42af58a1c30f7f961f77acb2fa35386 /test/ext/declarative/test_basic.py | |
parent | a4f2db890322a225e6c9754b711f5c16d04f377c (diff) | |
download | sqlalchemy-6bd46945ccd585c494eb7550a0dfea22f17727c0.tar.gz |
- reorganization of declarative such that file sizes are managable again.
the vast majority of file lines are spent on documentation, which moves
into package __init__. The core declarative idea lives in base and
is back down to its originally low size of under 500 lines. The various
helpers and such move into api.py, and the full span of string lookup
moves into a new module clsregistry. the rest of declarative only
refers to two functions in clsregistry in three places inside of base.
- [feature] Declarative now maintains a registry
of classes by string name as well as by full
module-qualified name. Multiple classes with the
same name can now be looked up based on a module-qualified
string within relationship(). Simple class name
lookups where more than one class shares the same
name now raises an informative error message.
[ticket:2338]
- lots of tests to ensure the new weak referencing memory management
is maintained by the new class registry system. this ticket was
served very well by waiting to do #2526 first, else this would
have needed to be rewritten anyway.
Diffstat (limited to 'test/ext/declarative/test_basic.py')
-rw-r--r-- | test/ext/declarative/test_basic.py | 1460 |
1 files changed, 1460 insertions, 0 deletions
diff --git a/test/ext/declarative/test_basic.py b/test/ext/declarative/test_basic.py new file mode 100644 index 000000000..b580558af --- /dev/null +++ b/test/ext/declarative/test_basic.py @@ -0,0 +1,1460 @@ + +from test.lib.testing import eq_, assert_raises, \ + assert_raises_message, is_ +from sqlalchemy.ext import declarative as decl +from sqlalchemy import exc +import sqlalchemy as sa +from test.lib import testing +from sqlalchemy import MetaData, Integer, String, ForeignKey, \ + ForeignKeyConstraint, Index +from test.lib.schema import Table, Column +from sqlalchemy.orm import relationship, create_session, class_mapper, \ + joinedload, configure_mappers, backref, clear_mappers, \ + deferred, column_property, composite,\ + Session +from test.lib.testing import eq_ +from sqlalchemy.util import classproperty +from sqlalchemy.ext.declarative import declared_attr, AbstractConcreteBase, ConcreteBase +from test.lib import fixtures +from test.lib.util import gc_collect + +Base = None + +class DeclarativeTestBase(fixtures.TestBase, testing.AssertsExecutionResults): + def setup(self): + global Base + Base = decl.declarative_base(testing.db) + + def teardown(self): + Session.close_all() + clear_mappers() + Base.metadata.drop_all() + +class DeclarativeTest(DeclarativeTestBase): + def test_basic(self): + class User(Base, fixtures.ComparableEntity): + __tablename__ = 'users' + + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + addresses = relationship("Address", backref="user") + + class Address(Base, fixtures.ComparableEntity): + __tablename__ = 'addresses' + + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column(String(50), key='_email') + user_id = Column('user_id', Integer, ForeignKey('users.id'), + key='_user_id') + + Base.metadata.create_all() + + eq_(Address.__table__.c['id'].name, 'id') + eq_(Address.__table__.c['_email'].name, 'email') + eq_(Address.__table__.c['_user_id'].name, 'user_id') + + u1 = User(name='u1', addresses=[ + Address(email='one'), + Address(email='two'), + ]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + + eq_(sess.query(User).all(), [User(name='u1', addresses=[ + Address(email='one'), + Address(email='two'), + ])]) + + a1 = sess.query(Address).filter(Address.email == 'two').one() + eq_(a1, Address(email='two')) + eq_(a1.user, User(name='u1')) + + def test_no_table(self): + def go(): + class User(Base): + id = Column('id', Integer, primary_key=True) + + assert_raises_message(sa.exc.InvalidRequestError, + 'does not have a __table__', go) + + def test_table_args_empty_dict(self): + + class MyModel(Base): + __tablename__ = 'test' + id = Column(Integer, primary_key=True) + __table_args__ = {} + + def test_table_args_empty_tuple(self): + + class MyModel(Base): + __tablename__ = 'test' + id = Column(Integer, primary_key=True) + __table_args__ = () + + def test_cant_add_columns(self): + t = Table('t', Base.metadata, Column('id', Integer, + primary_key=True), Column('data', String)) + + def go(): + class User(Base): + __table__ = t + foo = Column(Integer, primary_key=True) + + # can't specify new columns not already in the table + + assert_raises_message(sa.exc.ArgumentError, + "Can't add additional column 'foo' when " + "specifying __table__", go) + + # regular re-mapping works tho + + class Bar(Base): + __table__ = t + some_data = t.c.data + + assert class_mapper(Bar).get_property('some_data').columns[0] \ + is t.c.data + + def test_difficult_class(self): + """test no getattr() errors with a customized class""" + + # metaclass to mock the way zope.interface breaks getattr() + class BrokenMeta(type): + def __getattribute__(self, attr): + if attr == 'xyzzy': + raise AttributeError, 'xyzzy' + else: + return object.__getattribute__(self,attr) + + # even though this class has an xyzzy attribute, getattr(cls,"xyzzy") + # fails + class BrokenParent(object): + __metaclass__ = BrokenMeta + xyzzy = "magic" + + # _as_declarative() inspects obj.__class__.__bases__ + class User(BrokenParent,fixtures.ComparableEntity): + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + + decl.instrument_declarative(User,{},Base.metadata) + + def test_reserved_identifiers(self): + def go1(): + class User1(Base): + __tablename__ = 'user1' + id = Column(Integer, primary_key=True) + metadata = Column(Integer) + + def go2(): + class User2(Base): + __tablename__ = 'user2' + id = Column(Integer, primary_key=True) + metadata = relationship("Address") + + for go in (go1, go2): + assert_raises_message( + exc.InvalidRequestError, + "Attribute name 'metadata' is reserved " + "for the MetaData instance when using a " + "declarative base class.", + go + ) + + def test_undefer_column_name(self): + # TODO: not sure if there was an explicit + # test for this elsewhere + foo = Column(Integer) + eq_(str(foo), '(no name)') + eq_(foo.key, None) + eq_(foo.name, None) + decl.base._undefer_column_name('foo', foo) + eq_(str(foo), 'foo') + eq_(foo.key, 'foo') + eq_(foo.name, 'foo') + + def test_recompile_on_othermapper(self): + """declarative version of the same test in mappers.py""" + + from sqlalchemy.orm import mapperlib + + class User(Base): + __tablename__ = 'users' + + id = Column('id', Integer, primary_key=True) + name = Column('name', String(50)) + + class Address(Base): + __tablename__ = 'addresses' + + id = Column('id', Integer, primary_key=True) + email = Column('email', String(50)) + user_id = Column('user_id', Integer, ForeignKey('users.id')) + user = relationship("User", primaryjoin=user_id == User.id, + backref="addresses") + + assert mapperlib._new_mappers is True + u = User() + assert User.addresses + assert mapperlib._new_mappers is False + + def test_string_dependency_resolution(self): + from sqlalchemy.sql import desc + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column(String(50)) + addresses = relationship('Address', + order_by='desc(Address.email)', + primaryjoin='User.id==Address.user_id', + foreign_keys='[Address.user_id]', + backref=backref('user', + primaryjoin='User.id==Address.user_id', + foreign_keys='[Address.user_id]')) + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column(String(50)) + user_id = Column(Integer) # note no foreign key + + Base.metadata.create_all() + sess = create_session() + u1 = User(name='ed', addresses=[Address(email='abc'), + Address(email='def'), Address(email='xyz')]) + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).filter(User.name == 'ed').one(), + User(name='ed', addresses=[Address(email='xyz'), + Address(email='def'), Address(email='abc')])) + + class Foo(Base, fixtures.ComparableEntity): + + __tablename__ = 'foo' + id = Column(Integer, primary_key=True) + rel = relationship('User', + primaryjoin='User.addresses==Foo.id') + + assert_raises_message(exc.InvalidRequestError, + "'addresses' is not an instance of " + "ColumnProperty", configure_mappers) + + def test_string_dependency_resolution_two(self): + + class User(Base, fixtures.ComparableEntity): + __tablename__ = 'users' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column(String(50)) + + class Bar(Base, fixtures.ComparableEntity): + __tablename__ = 'bar' + id = Column(Integer, primary_key=True) + rel = relationship('User', + primaryjoin='User.id==Bar.__table__.id') + + assert_raises_message(exc.InvalidRequestError, + "does not have a mapped column named " + "'__table__'", configure_mappers) + + def test_string_w_pj_annotations(self): + + class User(Base, fixtures.ComparableEntity): + __tablename__ = 'users' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column(String(50)) + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column(String(50)) + user_id = Column(Integer) + user = relationship("User", + primaryjoin="remote(User.id)==foreign(Address.user_id)" + ) + + eq_( + Address.user.property._join_condition.local_remote_pairs, + [(Address.__table__.c.user_id, User.__table__.c.id)] + ) + + def test_string_dependency_resolution_no_magic(self): + """test that full tinkery expressions work as written""" + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column(Integer, primary_key=True) + addresses = relationship('Address', + primaryjoin='User.id==Address.user_id.prop.columns[' + '0]') + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True) + user_id = Column(Integer, ForeignKey('users.id')) + + configure_mappers() + eq_(str(User.addresses.prop.primaryjoin), + 'users.id = addresses.user_id') + + def test_string_dependency_resolution_module_qualified(self): + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column(Integer, primary_key=True) + addresses = relationship('%s.Address' % __name__, + primaryjoin='%s.User.id==%s.Address.user_id.prop.columns[' + '0]' % (__name__, __name__)) + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True) + user_id = Column(Integer, ForeignKey('users.id')) + + configure_mappers() + eq_(str(User.addresses.prop.primaryjoin), + 'users.id = addresses.user_id') + + def test_string_dependency_resolution_in_backref(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column(Integer, primary_key=True) + name = Column(String(50)) + addresses = relationship('Address', + primaryjoin='User.id==Address.user_id', + backref='user') + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True) + email = Column(String(50)) + user_id = Column(Integer, ForeignKey('users.id')) + + configure_mappers() + eq_(str(User.addresses.property.primaryjoin), + str(Address.user.property.primaryjoin)) + + def test_string_dependency_resolution_tables(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column(Integer, primary_key=True) + name = Column(String(50)) + props = relationship('Prop', secondary='user_to_prop', + primaryjoin='User.id==user_to_prop.c.u' + 'ser_id', + secondaryjoin='user_to_prop.c.prop_id=' + '=Prop.id', backref='users') + + class Prop(Base, fixtures.ComparableEntity): + + __tablename__ = 'props' + id = Column(Integer, primary_key=True) + name = Column(String(50)) + + user_to_prop = Table('user_to_prop', Base.metadata, + Column('user_id', Integer, + ForeignKey('users.id')), Column('prop_id', + Integer, ForeignKey('props.id'))) + configure_mappers() + assert class_mapper(User).get_property('props').secondary \ + is user_to_prop + + def test_string_dependency_resolution_schemas(self): + Base = decl.declarative_base() + + class User(Base): + + __tablename__ = 'users' + __table_args__ = {'schema':'fooschema'} + + id = Column(Integer, primary_key=True) + name = Column(String(50)) + props = relationship('Prop', secondary='fooschema.user_to_prop', + primaryjoin='User.id==fooschema.user_to_prop.c.user_id', + secondaryjoin='fooschema.user_to_prop.c.prop_id==Prop.id', + backref='users') + + class Prop(Base): + + __tablename__ = 'props' + __table_args__ = {'schema':'fooschema'} + + id = Column(Integer, primary_key=True) + name = Column(String(50)) + + user_to_prop = Table('user_to_prop', Base.metadata, + Column('user_id', Integer, ForeignKey('fooschema.users.id')), + Column('prop_id',Integer, ForeignKey('fooschema.props.id')), + schema='fooschema') + configure_mappers() + + assert class_mapper(User).get_property('props').secondary \ + is user_to_prop + + def test_string_dependency_resolution_annotations(self): + Base = decl.declarative_base() + + class Parent(Base): + __tablename__ = 'parent' + id = Column(Integer, primary_key=True) + name = Column(String) + children = relationship("Child", + primaryjoin="Parent.name==remote(foreign(func.lower(Child.name_upper)))" + ) + + class Child(Base): + __tablename__ = 'child' + id = Column(Integer, primary_key=True) + name_upper = Column(String) + + configure_mappers() + eq_( + Parent.children.property._calculated_foreign_keys, + set([Child.name_upper.property.columns[0]]) + ) + + def test_shared_class_registry(self): + reg = {} + Base1 = decl.declarative_base(testing.db, class_registry=reg) + Base2 = decl.declarative_base(testing.db, class_registry=reg) + + class A(Base1): + __tablename__ = 'a' + id = Column(Integer, primary_key=True) + + class B(Base2): + __tablename__ = 'b' + id = Column(Integer, primary_key=True) + aid = Column(Integer, ForeignKey(A.id)) + as_ = relationship("A") + + assert B.as_.property.mapper.class_ is A + + def test_uncompiled_attributes_in_relationship(self): + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column(String(50)) + user_id = Column(Integer, ForeignKey('users.id')) + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column(String(50)) + addresses = relationship('Address', order_by=Address.email, + foreign_keys=Address.user_id, + remote_side=Address.user_id) + + # get the mapper for User. User mapper will compile, + # "addresses" relationship will call upon Address.user_id for + # its clause element. Address.user_id is a _CompileOnAttr, + # which then calls class_mapper(Address). But ! We're already + # "in compilation", but class_mapper(Address) needs to + # initialize regardless, or COA's assertion fails and things + # generally go downhill from there. + + class_mapper(User) + Base.metadata.create_all() + sess = create_session() + u1 = User(name='ed', addresses=[Address(email='abc'), + Address(email='xyz'), Address(email='def')]) + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).filter(User.name == 'ed').one(), + User(name='ed', addresses=[Address(email='abc'), + Address(email='def'), Address(email='xyz')])) + + def test_nice_dependency_error(self): + + class User(Base): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True) + addresses = relationship('Address') + + class Address(Base): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True) + foo = sa.orm.column_property(User.id == 5) + + # this used to raise an error when accessing User.id but that's + # no longer the case since we got rid of _CompileOnAttr. + + assert_raises(sa.exc.ArgumentError, configure_mappers) + + def test_nice_dependency_error_works_with_hasattr(self): + + class User(Base): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True) + addresses = relationship('Address') + + # hasattr() on a compile-loaded attribute + try: + hasattr(User.addresses, 'property') + except exc.InvalidRequestError: + assert sa.util.compat.py32 + + # the exception is preserved. Remains the + # same through repeated calls. + for i in range(3): + assert_raises_message(sa.exc.InvalidRequestError, + "^One or more mappers failed to initialize - " + "can't proceed with initialization of other " + "mappers. Original exception was: When initializing.*", + configure_mappers) + + def test_custom_base(self): + class MyBase(object): + def foobar(self): + return "foobar" + Base = decl.declarative_base(cls=MyBase) + assert hasattr(Base, 'metadata') + assert Base().foobar() == "foobar" + + def test_uses_get_on_class_col_fk(self): + + # test [ticket:1492] + + class Master(Base): + + __tablename__ = 'master' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + + class Detail(Base): + + __tablename__ = 'detail' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + master_id = Column(None, ForeignKey(Master.id)) + master = relationship(Master) + + Base.metadata.create_all() + configure_mappers() + assert class_mapper(Detail).get_property('master' + ).strategy.use_get + m1 = Master() + d1 = Detail(master=m1) + sess = create_session() + sess.add(d1) + sess.flush() + sess.expunge_all() + d1 = sess.query(Detail).first() + m1 = sess.query(Master).first() + + def go(): + assert d1.master + + self.assert_sql_count(testing.db, go, 0) + + def test_index_doesnt_compile(self): + class User(Base): + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True) + name = Column('name', String(50)) + error = relationship("Address") + + i = Index('my_index', User.name) + + # compile fails due to the nonexistent Addresses relationship + assert_raises(sa.exc.InvalidRequestError, configure_mappers) + + # index configured + assert i in User.__table__.indexes + assert User.__table__.c.id not in set(i.columns) + assert User.__table__.c.name in set(i.columns) + + # tables create fine + Base.metadata.create_all() + + def test_add_prop(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + + User.name = Column('name', String(50)) + User.addresses = relationship('Address', backref='user') + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + + Address.email = Column(String(50), key='_email') + Address.user_id = Column('user_id', Integer, + ForeignKey('users.id'), key='_user_id') + Base.metadata.create_all() + eq_(Address.__table__.c['id'].name, 'id') + eq_(Address.__table__.c['_email'].name, 'email') + eq_(Address.__table__.c['_user_id'].name, 'user_id') + u1 = User(name='u1', addresses=[Address(email='one'), + Address(email='two')]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).all(), [User(name='u1', + addresses=[Address(email='one'), Address(email='two')])]) + a1 = sess.query(Address).filter(Address.email == 'two').one() + eq_(a1, Address(email='two')) + eq_(a1.user, User(name='u1')) + + def test_eager_order_by(self): + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column('email', String(50)) + user_id = Column('user_id', Integer, ForeignKey('users.id')) + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + addresses = relationship('Address', order_by=Address.email) + + Base.metadata.create_all() + u1 = User(name='u1', addresses=[Address(email='two'), + Address(email='one')]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).options(joinedload(User.addresses)).all(), + [User(name='u1', addresses=[Address(email='one'), + Address(email='two')])]) + + def test_order_by_multi(self): + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column('email', String(50)) + user_id = Column('user_id', Integer, ForeignKey('users.id')) + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + addresses = relationship('Address', + order_by=(Address.email, Address.id)) + + Base.metadata.create_all() + u1 = User(name='u1', addresses=[Address(email='two'), + Address(email='one')]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + u = sess.query(User).filter(User.name == 'u1').one() + a = u.addresses + + def test_as_declarative(self): + + class User(fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + addresses = relationship('Address', backref='user') + + class Address(fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column('email', String(50)) + user_id = Column('user_id', Integer, ForeignKey('users.id')) + + reg = {} + decl.instrument_declarative(User, reg, Base.metadata) + decl.instrument_declarative(Address, reg, Base.metadata) + Base.metadata.create_all() + u1 = User(name='u1', addresses=[Address(email='one'), + Address(email='two')]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).all(), [User(name='u1', + addresses=[Address(email='one'), Address(email='two')])]) + + def test_custom_mapper_attribute(self): + + def mymapper(cls, tbl, **kwargs): + m = sa.orm.mapper(cls, tbl, **kwargs) + m.CHECK = True + return m + + base = decl.declarative_base() + + class Foo(base): + __tablename__ = 'foo' + __mapper_cls__ = mymapper + id = Column(Integer, primary_key=True) + + eq_(Foo.__mapper__.CHECK, True) + + def test_custom_mapper_argument(self): + + def mymapper(cls, tbl, **kwargs): + m = sa.orm.mapper(cls, tbl, **kwargs) + m.CHECK = True + return m + + base = decl.declarative_base(mapper=mymapper) + + class Foo(base): + __tablename__ = 'foo' + id = Column(Integer, primary_key=True) + + eq_(Foo.__mapper__.CHECK, True) + + @testing.emits_warning('Ignoring declarative-like tuple value of ' + 'attribute id') + def test_oops(self): + + def define(): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True), + name = Column('name', String(50)) + + assert False + + assert_raises_message(sa.exc.ArgumentError, + 'Mapper Mapper|User|users could not ' + 'assemble any primary key', define) + + def test_table_args_no_dict(self): + + class Foo1(Base): + + __tablename__ = 'foo' + __table_args__ = ForeignKeyConstraint(['id'], ['foo.bar']), + id = Column('id', Integer, primary_key=True) + bar = Column('bar', Integer) + + assert Foo1.__table__.c.id.references(Foo1.__table__.c.bar) + + def test_table_args_type(self): + def err(): + class Foo1(Base): + + __tablename__ = 'foo' + __table_args__ = ForeignKeyConstraint(['id'], ['foo.id' + ]) + id = Column('id', Integer, primary_key=True) + assert_raises_message(sa.exc.ArgumentError, + '__table_args__ value must be a tuple, ', err) + + def test_table_args_none(self): + + class Foo2(Base): + + __tablename__ = 'foo' + __table_args__ = None + id = Column('id', Integer, primary_key=True) + + assert Foo2.__table__.kwargs == {} + + def test_table_args_dict_format(self): + + class Foo2(Base): + + __tablename__ = 'foo' + __table_args__ = {'mysql_engine': 'InnoDB'} + id = Column('id', Integer, primary_key=True) + + assert Foo2.__table__.kwargs['mysql_engine'] == 'InnoDB' + + def test_table_args_tuple_format(self): + class Foo2(Base): + + __tablename__ = 'foo' + __table_args__ = {'mysql_engine': 'InnoDB'} + id = Column('id', Integer, primary_key=True) + + class Bar(Base): + + __tablename__ = 'bar' + __table_args__ = ForeignKeyConstraint(['id'], ['foo.id']), \ + {'mysql_engine': 'InnoDB'} + id = Column('id', Integer, primary_key=True) + + assert Bar.__table__.c.id.references(Foo2.__table__.c.id) + assert Bar.__table__.kwargs['mysql_engine'] == 'InnoDB' + + def test_expression(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + addresses = relationship('Address', backref='user') + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column('email', String(50)) + user_id = Column('user_id', Integer, ForeignKey('users.id')) + + User.address_count = \ + sa.orm.column_property(sa.select([sa.func.count(Address.id)]). + where(Address.user_id + == User.id).as_scalar()) + Base.metadata.create_all() + u1 = User(name='u1', addresses=[Address(email='one'), + Address(email='two')]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).all(), [User(name='u1', address_count=2, + addresses=[Address(email='one'), Address(email='two')])]) + + def test_useless_declared_attr(self): + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column('email', String(50)) + user_id = Column('user_id', Integer, ForeignKey('users.id')) + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + addresses = relationship('Address', backref='user') + + @declared_attr + def address_count(cls): + # this doesn't really gain us anything. but if + # one is used, lets have it function as expected... + return sa.orm.column_property(sa.select([sa.func.count(Address.id)]). + where(Address.user_id == cls.id)) + + Base.metadata.create_all() + u1 = User(name='u1', addresses=[Address(email='one'), + Address(email='two')]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).all(), [User(name='u1', address_count=2, + addresses=[Address(email='one'), Address(email='two')])]) + + def test_useless_declared_attr_warns_on_subclass(self): + def go(): + class MyBase(Base): + __tablename__ = 'foo' + id = Column(Integer, primary_key=True) + @declared_attr + def somecol(cls): + return Column(Integer) + + class MyClass(MyBase): + __tablename__ = 'bar' + assert_raises_message( + sa.exc.SAWarning, + r"Regular \(i.e. not __special__\) attribute 'MyBase.somecol' " + "uses @declared_attr, but owning class " + "<class 'test.ext.declarative.test_basic.MyBase'> is " + "mapped - not applying to subclass <class " + "'test.ext.declarative.test_basic.MyClass'>.", + go + ) + + def test_column(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + + User.a = Column('a', String(10)) + User.b = Column(String(10)) + Base.metadata.create_all() + u1 = User(name='u1', a='a', b='b') + eq_(u1.a, 'a') + eq_(User.a.get_history(u1), (['a'], (), ())) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).all(), [User(name='u1', a='a', b='b')]) + + def test_column_properties(self): + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column(String(50)) + user_id = Column(Integer, ForeignKey('users.id')) + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + + adr_count = \ + sa.orm.column_property( + sa.select([sa.func.count(Address.id)], + Address.user_id == id).as_scalar()) + addresses = relationship(Address) + + Base.metadata.create_all() + u1 = User(name='u1', addresses=[Address(email='one'), + Address(email='two')]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).all(), [User(name='u1', adr_count=2, + addresses=[Address(email='one'), Address(email='two')])]) + + def test_column_properties_2(self): + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True) + email = Column(String(50)) + user_id = Column(Integer, ForeignKey('users.id')) + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True) + name = Column('name', String(50)) + + # this is not "valid" but we want to test that Address.id + # doesnt get stuck into user's table + + adr_count = Address.id + + eq_(set(User.__table__.c.keys()), set(['id', 'name'])) + eq_(set(Address.__table__.c.keys()), set(['id', 'email', + 'user_id'])) + + def test_deferred(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + name = sa.orm.deferred(Column(String(50))) + + Base.metadata.create_all() + sess = create_session() + sess.add(User(name='u1')) + sess.flush() + sess.expunge_all() + u1 = sess.query(User).filter(User.name == 'u1').one() + assert 'name' not in u1.__dict__ + + def go(): + eq_(u1.name, 'u1') + + self.assert_sql_count(testing.db, go, 1) + + def test_composite_inline(self): + class AddressComposite(fixtures.ComparableEntity): + def __init__(self, street, state): + self.street = street + self.state = state + def __composite_values__(self): + return [self.street, self.state] + + class User(Base, fixtures.ComparableEntity): + __tablename__ = 'user' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + address = composite(AddressComposite, + Column('street', String(50)), + Column('state', String(2)), + ) + + Base.metadata.create_all() + sess = Session() + sess.add(User( + address=AddressComposite('123 anywhere street', + 'MD') + )) + sess.commit() + eq_( + sess.query(User).all(), + [User(address=AddressComposite('123 anywhere street', + 'MD'))] + ) + + def test_composite_separate(self): + class AddressComposite(fixtures.ComparableEntity): + def __init__(self, street, state): + self.street = street + self.state = state + def __composite_values__(self): + return [self.street, self.state] + + class User(Base, fixtures.ComparableEntity): + __tablename__ = 'user' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + street = Column(String(50)) + state = Column(String(2)) + address = composite(AddressComposite, + street, state) + + Base.metadata.create_all() + sess = Session() + sess.add(User( + address=AddressComposite('123 anywhere street', + 'MD') + )) + sess.commit() + eq_( + sess.query(User).all(), + [User(address=AddressComposite('123 anywhere street', + 'MD'))] + ) + + def test_mapping_to_join(self): + users = Table('users', Base.metadata, + Column('id', Integer, primary_key=True) + ) + addresses = Table('addresses', Base.metadata, + Column('id', Integer, primary_key=True), + Column('user_id', Integer, ForeignKey('users.id')) + ) + usersaddresses = sa.join(users, addresses, users.c.id + == addresses.c.user_id) + class User(Base): + __table__ = usersaddresses + __table_args__ = {'primary_key':[users.c.id]} + + # need to use column_property for now + user_id = column_property(users.c.id, addresses.c.user_id) + address_id = addresses.c.id + + assert User.__mapper__.get_property('user_id').columns[0] \ + is users.c.id + assert User.__mapper__.get_property('user_id').columns[1] \ + is addresses.c.user_id + + def test_synonym_inline(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + _name = Column('name', String(50)) + + def _set_name(self, name): + self._name = 'SOMENAME ' + name + + def _get_name(self): + return self._name + + name = sa.orm.synonym('_name', + descriptor=property(_get_name, + _set_name)) + + Base.metadata.create_all() + sess = create_session() + u1 = User(name='someuser') + eq_(u1.name, 'SOMENAME someuser') + sess.add(u1) + sess.flush() + eq_(sess.query(User).filter(User.name == 'SOMENAME someuser' + ).one(), u1) + + def test_synonym_no_descriptor(self): + from sqlalchemy.orm.properties import ColumnProperty + + class CustomCompare(ColumnProperty.Comparator): + + __hash__ = None + + def __eq__(self, other): + return self.__clause_element__() == other + ' FOO' + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + _name = Column('name', String(50)) + name = sa.orm.synonym('_name', + comparator_factory=CustomCompare) + + Base.metadata.create_all() + sess = create_session() + u1 = User(name='someuser FOO') + sess.add(u1) + sess.flush() + eq_(sess.query(User).filter(User.name == 'someuser').one(), u1) + + def test_synonym_added(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + _name = Column('name', String(50)) + + def _set_name(self, name): + self._name = 'SOMENAME ' + name + + def _get_name(self): + return self._name + + name = property(_get_name, _set_name) + + User.name = sa.orm.synonym('_name', descriptor=User.name) + Base.metadata.create_all() + sess = create_session() + u1 = User(name='someuser') + eq_(u1.name, 'SOMENAME someuser') + sess.add(u1) + sess.flush() + eq_(sess.query(User).filter(User.name == 'SOMENAME someuser' + ).one(), u1) + + def test_reentrant_compile_via_foreignkey(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + addresses = relationship('Address', backref='user') + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column('email', String(50)) + user_id = Column('user_id', Integer, ForeignKey(User.id)) + + # previous versions would force a re-entrant mapper compile via + # the User.id inside the ForeignKey but this is no longer the + # case + + sa.orm.configure_mappers() + eq_(str(list(Address.user_id.property.columns[0].foreign_keys)[0]), + "ForeignKey('users.id')") + Base.metadata.create_all() + u1 = User(name='u1', addresses=[Address(email='one'), + Address(email='two')]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).all(), [User(name='u1', + addresses=[Address(email='one'), Address(email='two')])]) + + def test_relationship_reference(self): + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column('email', String(50)) + user_id = Column('user_id', Integer, ForeignKey('users.id')) + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + addresses = relationship('Address', backref='user', + primaryjoin=id == Address.user_id) + + User.address_count = \ + sa.orm.column_property(sa.select([sa.func.count(Address.id)]). + where(Address.user_id + == User.id).as_scalar()) + Base.metadata.create_all() + u1 = User(name='u1', addresses=[Address(email='one'), + Address(email='two')]) + sess = create_session() + sess.add(u1) + sess.flush() + sess.expunge_all() + eq_(sess.query(User).all(), [User(name='u1', address_count=2, + addresses=[Address(email='one'), Address(email='two')])]) + + def test_pk_with_fk_init(self): + + class Bar(Base): + + __tablename__ = 'bar' + id = sa.Column(sa.Integer, sa.ForeignKey('foo.id'), + primary_key=True) + ex = sa.Column(sa.Integer, primary_key=True) + + class Foo(Base): + + __tablename__ = 'foo' + id = sa.Column(sa.Integer, primary_key=True) + bars = sa.orm.relationship(Bar) + + assert Bar.__mapper__.primary_key[0] is Bar.__table__.c.id + assert Bar.__mapper__.primary_key[1] is Bar.__table__.c.ex + + def test_with_explicit_autoloaded(self): + meta = MetaData(testing.db) + t1 = Table('t1', meta, Column('id', String(50), + primary_key=True, test_needs_autoincrement=True), + Column('data', String(50))) + meta.create_all() + try: + + class MyObj(Base): + + __table__ = Table('t1', Base.metadata, autoload=True) + + sess = create_session() + m = MyObj(id='someid', data='somedata') + sess.add(m) + sess.flush() + eq_(t1.select().execute().fetchall(), [('someid', 'somedata' + )]) + finally: + meta.drop_all() + + def test_synonym_for(self): + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + + @decl.synonym_for('name') + @property + def namesyn(self): + return self.name + + Base.metadata.create_all() + sess = create_session() + u1 = User(name='someuser') + eq_(u1.name, 'someuser') + eq_(u1.namesyn, 'someuser') + sess.add(u1) + sess.flush() + rt = sess.query(User).filter(User.namesyn == 'someuser').one() + eq_(rt, u1) + + def test_comparable_using(self): + + class NameComparator(sa.orm.PropComparator): + + @property + def upperself(self): + cls = self.prop.parent.class_ + col = getattr(cls, 'name') + return sa.func.upper(col) + + def operate( + self, + op, + other, + **kw + ): + return op(self.upperself, other, **kw) + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column('id', Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column('name', String(50)) + + @decl.comparable_using(NameComparator) + @property + def uc_name(self): + return self.name is not None and self.name.upper() \ + or None + + Base.metadata.create_all() + sess = create_session() + u1 = User(name='someuser') + eq_(u1.name, 'someuser', u1.name) + eq_(u1.uc_name, 'SOMEUSER', u1.uc_name) + sess.add(u1) + sess.flush() + sess.expunge_all() + rt = sess.query(User).filter(User.uc_name == 'SOMEUSER').one() + eq_(rt, u1) + sess.expunge_all() + rt = sess.query(User).filter(User.uc_name.startswith('SOMEUSE' + )).one() + eq_(rt, u1) + + def test_duplicate_classes_in_base(self): + + class Test(Base): + __tablename__ = 'a' + id = Column(Integer, primary_key=True) + + assert_raises_message( + sa.exc.SAWarning, + "This declarative base already contains a class with ", + lambda: type(Base)("Test", (Base,), dict( + __tablename__='b', + id=Column(Integer, primary_key=True) + )) + ) + + + +def _produce_test(inline, stringbased): + + class ExplicitJoinTest(fixtures.MappedTest): + + @classmethod + def define_tables(cls, metadata): + global User, Address + Base = decl.declarative_base(metadata=metadata) + + class User(Base, fixtures.ComparableEntity): + + __tablename__ = 'users' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + name = Column(String(50)) + + class Address(Base, fixtures.ComparableEntity): + + __tablename__ = 'addresses' + id = Column(Integer, primary_key=True, + test_needs_autoincrement=True) + email = Column(String(50)) + user_id = Column(Integer, ForeignKey('users.id')) + if inline: + if stringbased: + user = relationship('User', + primaryjoin='User.id==Address.user_id', + backref='addresses') + else: + user = relationship(User, primaryjoin=User.id + == user_id, backref='addresses') + + if not inline: + configure_mappers() + if stringbased: + Address.user = relationship('User', + primaryjoin='User.id==Address.user_id', + backref='addresses') + else: + Address.user = relationship(User, + primaryjoin=User.id == Address.user_id, + backref='addresses') + + @classmethod + def insert_data(cls): + params = [dict(zip(('id', 'name'), column_values)) + for column_values in [(7, 'jack'), (8, 'ed'), (9, + 'fred'), (10, 'chuck')]] + User.__table__.insert().execute(params) + Address.__table__.insert().execute([dict(zip(('id', + 'user_id', 'email'), column_values)) + for column_values in [(1, 7, 'jack@bean.com'), (2, + 8, 'ed@wood.com'), (3, 8, 'ed@bettyboop.com'), (4, + 8, 'ed@lala.com'), (5, 9, 'fred@fred.com')]]) + + def test_aliased_join(self): + + # this query will screw up if the aliasing enabled in + # query.join() gets applied to the right half of the join + # condition inside the any(). the join condition inside of + # any() comes from the "primaryjoin" of the relationship, + # and should not be annotated with _orm_adapt. + # PropertyLoader.Comparator will annotate the left side with + # _orm_adapt, though. + + sess = create_session() + eq_(sess.query(User).join(User.addresses, + aliased=True).filter(Address.email == 'ed@wood.com' + ).filter(User.addresses.any(Address.email + == 'jack@bean.com')).all(), []) + + ExplicitJoinTest.__name__ = 'ExplicitJoinTest%s%s' % (inline + and 'Inline' or 'Separate', stringbased and 'String' + or 'Literal') + return ExplicitJoinTest + +for inline in True, False: + for stringbased in True, False: + testclass = _produce_test(inline, stringbased) + exec '%s = testclass' % testclass.__name__ + del testclass + |