diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2010-09-14 20:43:48 -0400 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2010-09-14 20:43:48 -0400 |
| commit | 3d389b19b70b65cb76226c3f3aa4c5d926e1f12b (patch) | |
| tree | bdc597ebf2786f0514c8029dd2dbe2385bb9c2df /test/orm/test_attributes.py | |
| parent | 60b82d6e13246a3d88e7288e863a5231b7572c6a (diff) | |
| download | sqlalchemy-3d389b19b70b65cb76226c3f3aa4c5d926e1f12b.tar.gz | |
- reorganization
- attrbutes.py splits into attribtes.py and instrumentation.py
- all the various Event subclasses go into events.py modules
- some ideas for orm events
- move *Extension out to deprecated_interfaces
Diffstat (limited to 'test/orm/test_attributes.py')
| -rw-r--r-- | test/orm/test_attributes.py | 134 |
1 files changed, 67 insertions, 67 deletions
diff --git a/test/orm/test_attributes.py b/test/orm/test_attributes.py index 742e9d874..60f9c6b9e 100644 --- a/test/orm/test_attributes.py +++ b/test/orm/test_attributes.py @@ -1,5 +1,5 @@ import pickle -import sqlalchemy.orm.attributes as attributes +from sqlalchemy.orm import attributes, instrumentation from sqlalchemy.orm.collections import collection from sqlalchemy.orm.interfaces import AttributeExtension from sqlalchemy import exc as sa_exc @@ -27,7 +27,7 @@ class AttributesTest(_base.ORMTest): def test_basic(self): class User(object):pass - attributes.register_class(User) + instrumentation.register_class(User) attributes.register_attribute(User, 'user_id', uselist=False, useobject=False) attributes.register_attribute(User, 'user_name', uselist=False, useobject=False) attributes.register_attribute(User, 'email_address', uselist=False, useobject=False) @@ -46,8 +46,8 @@ class AttributesTest(_base.ORMTest): self.assert_(u.user_id == 7 and u.user_name == 'heythere' and u.email_address == 'foo@bar.com') def test_pickleness(self): - attributes.register_class(MyTest) - attributes.register_class(MyTest2) + instrumentation.register_class(MyTest) + instrumentation.register_class(MyTest2) attributes.register_attribute(MyTest, 'user_id', uselist=False, useobject=False) attributes.register_attribute(MyTest, 'user_name', uselist=False, useobject=False) attributes.register_attribute(MyTest, 'email_address', uselist=False, useobject=False) @@ -116,7 +116,7 @@ class AttributesTest(_base.ORMTest): class Foo(object): pass - attributes.register_class(Foo) + instrumentation.register_class(Foo) f = Foo() state = attributes.instance_state(f) f.bar = "foo" @@ -135,7 +135,7 @@ class AttributesTest(_base.ORMTest): state.dict[k] = data[k] return attributes.ATTR_WAS_SET - attributes.register_class(Foo) + instrumentation.register_class(Foo) manager = attributes.manager_of_class(Foo) manager.deferred_scalar_loader = loader attributes.register_attribute(Foo, 'a', uselist=False, useobject=False) @@ -175,7 +175,7 @@ class AttributesTest(_base.ORMTest): state.dict[k] = data[k] return attributes.ATTR_WAS_SET - attributes.register_class(MyTest) + instrumentation.register_class(MyTest) manager = attributes.manager_of_class(MyTest) manager.deferred_scalar_loader=loader attributes.register_attribute(MyTest, 'a', uselist=False, useobject=False) @@ -193,8 +193,8 @@ class AttributesTest(_base.ORMTest): class User(object):pass class Address(object):pass - attributes.register_class(User) - attributes.register_class(Address) + instrumentation.register_class(User) + instrumentation.register_class(Address) attributes.register_attribute(User, 'user_id', uselist=False, useobject=False) attributes.register_attribute(User, 'user_name', uselist=False, useobject=False) attributes.register_attribute(User, 'addresses', uselist = True, useobject=True) @@ -253,8 +253,8 @@ class AttributesTest(_base.ORMTest): state.commit_all(state.dict) return child - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) b1, b2, b3, b4 = Bar(id='b1'), Bar(id='b2'), Bar(id='b3'), Bar(id='b4') @@ -336,8 +336,8 @@ class AttributesTest(_base.ORMTest): def set(self, state, child, oldchild, initiator): return child - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) bar1, bar2, bar3 = [Bar(id=1), Bar(id=2), Bar(id=3)] def func1(**kw): @@ -377,7 +377,7 @@ class AttributesTest(_base.ORMTest): results.append(("set", state.obj(), child, oldchild)) return child - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'x', uselist=False, mutable_scalars=False, useobject=False, extension=ReceiveEvents()) attributes.register_attribute(Foo, 'y', uselist=False, mutable_scalars=True, useobject=False, copy_function=lambda x:x, extension=ReceiveEvents()) @@ -407,8 +407,8 @@ class AttributesTest(_base.ORMTest): class Post(object):pass class Blog(object):pass - attributes.register_class(Post) - attributes.register_class(Blog) + instrumentation.register_class(Post) + instrumentation.register_class(Blog) # set up instrumented attributes with backrefs attributes.register_attribute(Post, 'blog', uselist=False, @@ -448,8 +448,8 @@ class AttributesTest(_base.ORMTest): class Bar(Foo):pass - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) def func1(**kw): return "this is the foo attr" @@ -482,8 +482,8 @@ class AttributesTest(_base.ORMTest): Foo.__init__(self) - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) b = Bar() eq_(len(states), 1) @@ -499,8 +499,8 @@ class AttributesTest(_base.ORMTest): class Element(object): _state = True - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) attributes.register_attribute(Foo, 'element', uselist=False, useobject=True) el = Element() x = Bar() @@ -520,8 +520,8 @@ class AttributesTest(_base.ORMTest): class Bar(_base.BasicEntity): pass - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) bar1, bar2, bar3, bar4 = [Bar(id=1), Bar(id=2), Bar(id=3), Bar(id=4)] def func1(**kw): @@ -542,8 +542,8 @@ class AttributesTest(_base.ORMTest): class Foo(object):pass class Bar(object):pass - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) attributes.register_attribute(Foo, 'element', uselist=False, trackparent=True, useobject=True) attributes.register_attribute(Bar, 'element', uselist=False, trackparent=True, useobject=True) @@ -576,7 +576,7 @@ class AttributesTest(_base.ORMTest): """test detection of changes on mutable scalar items""" class Foo(object):pass - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'element', uselist=False, copy_function=lambda x:[y for y in x], mutable_scalars=True, useobject=False) x = Foo() x.element = ['one', 'two', 'three'] @@ -584,9 +584,9 @@ class AttributesTest(_base.ORMTest): x.element[1] = 'five' assert attributes.instance_state(x).modified - attributes.unregister_class(Foo) + instrumentation.unregister_class(Foo) - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'element', uselist=False, useobject=False) x = Foo() x.element = ['one', 'two', 'three'] @@ -607,13 +607,13 @@ class AttributesTest(_base.ORMTest): class Foo(object): A = des() - attributes.register_class(Foo) - attributes.unregister_class(Foo) + instrumentation.register_class(Foo) + instrumentation.unregister_class(Foo) def test_collectionclasses(self): class Foo(object):pass - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, "collection", uselist=True, typecallable=set, useobject=True) assert attributes.manager_of_class(Foo).is_instrumented("collection") @@ -672,8 +672,8 @@ class UtilTest(_base.ORMTest): class Bar(object): pass - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) attributes.register_attribute(Foo, "coll", uselist=True, useobject=True) f1 = Foo() @@ -697,8 +697,8 @@ class BackrefTest(_base.ORMTest): class Student(object):pass class Course(object):pass - attributes.register_class(Student) - attributes.register_class(Course) + instrumentation.register_class(Student) + instrumentation.register_class(Course) attributes.register_attribute(Student, 'courses', uselist=True, extension=attributes.GenericBackrefExtension('students' ), useobject=True) @@ -725,8 +725,8 @@ class BackrefTest(_base.ORMTest): class Post(object):pass class Blog(object):pass - attributes.register_class(Post) - attributes.register_class(Blog) + instrumentation.register_class(Post) + instrumentation.register_class(Blog) attributes.register_attribute(Post, 'blog', uselist=False, extension=attributes.GenericBackrefExtension('posts'), trackparent=True, useobject=True) @@ -759,8 +759,8 @@ class BackrefTest(_base.ORMTest): def test_o2o(self): class Port(object):pass class Jack(object):pass - attributes.register_class(Port) - attributes.register_class(Jack) + instrumentation.register_class(Port) + instrumentation.register_class(Jack) attributes.register_attribute(Port, 'jack', uselist=False, extension=attributes.GenericBackrefExtension('port'), useobject=True) @@ -793,9 +793,9 @@ class BackrefTest(_base.ORMTest): p_token = object() c_token = object() - attributes.register_class(Parent) - attributes.register_class(Child) - attributes.register_class(SubChild) + instrumentation.register_class(Parent) + instrumentation.register_class(Child) + instrumentation.register_class(SubChild) attributes.register_attribute(Parent, 'child', uselist=False, extension=attributes.GenericBackrefExtension('parent'), parent_token = p_token, @@ -828,9 +828,9 @@ class BackrefTest(_base.ORMTest): p_token = object() c_token = object() - attributes.register_class(Parent) - attributes.register_class(SubParent) - attributes.register_class(Child) + instrumentation.register_class(Parent) + instrumentation.register_class(SubParent) + instrumentation.register_class(Child) attributes.register_attribute(Parent, 'children', uselist=True, extension=attributes.GenericBackrefExtension('parent'), parent_token = p_token, @@ -896,8 +896,8 @@ class PendingBackrefTest(_base.ORMTest): return attributes.PASSIVE_NO_RESULT return load - attributes.register_class(Post) - attributes.register_class(Blog) + instrumentation.register_class(Post) + instrumentation.register_class(Blog) attributes.register_attribute(Post, 'blog', uselist=False, extension=attributes.GenericBackrefExtension('posts'), trackparent=True, useobject=True) attributes.register_attribute(Blog, 'posts', uselist=True, extension=attributes.GenericBackrefExtension('blog'), callable_=lazy_posts, trackparent=True, useobject=True) @@ -993,7 +993,7 @@ class HistoryTest(_base.ORMTest): class Foo(_base.BasicEntity): pass - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'someattr', uselist=False, useobject=False) f = Foo() @@ -1013,7 +1013,7 @@ class HistoryTest(_base.ORMTest): class Foo(_base.BasicEntity): pass - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'someattr', uselist=False, useobject=False) # case 1. new object @@ -1073,7 +1073,7 @@ class HistoryTest(_base.ORMTest): class Foo(_base.BasicEntity): pass - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'someattr', uselist=False, useobject=False, mutable_scalars=True, copy_function=dict) # case 1. new object @@ -1121,7 +1121,7 @@ class HistoryTest(_base.ORMTest): new = Bar(name='new') old = Bar(name='old') - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'someattr', uselist=False, useobject=True) # case 1. new object @@ -1184,7 +1184,7 @@ class HistoryTest(_base.ORMTest): def __nonzero__(self): assert False - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'someattr', uselist=True, useobject=True) hi = Bar(name='hi') @@ -1236,7 +1236,7 @@ class HistoryTest(_base.ORMTest): from sqlalchemy.orm.collections import attribute_mapped_collection - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'someattr', uselist=True, useobject=True, typecallable=attribute_mapped_collection('name')) hi = Bar(name='hi') @@ -1262,7 +1262,7 @@ class HistoryTest(_base.ORMTest): class Bar(_base.BasicEntity): pass - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'someattr', uselist=True, useobject=True) attributes.register_attribute(Foo, 'id', uselist=False, useobject=False) @@ -1352,8 +1352,8 @@ class HistoryTest(_base.ORMTest): class Bar(_base.BasicEntity): pass - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) attributes.register_attribute(Foo, 'bars', uselist=True, extension=attributes.GenericBackrefExtension('foo'), trackparent=True, useobject=True) attributes.register_attribute(Bar, 'foo', uselist=False, extension=attributes.GenericBackrefExtension('bars'), trackparent=True, useobject=True) @@ -1385,8 +1385,8 @@ class HistoryTest(_base.ORMTest): return lazy_load return load - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) attributes.register_attribute(Foo, 'bars', uselist=True, extension=attributes.GenericBackrefExtension('foo'), trackparent=True, callable_=lazyload, useobject=True) attributes.register_attribute(Bar, 'foo', uselist=False, extension=attributes.GenericBackrefExtension('bars'), trackparent=True, useobject=True) @@ -1420,8 +1420,8 @@ class HistoryTest(_base.ORMTest): return lazy_load return load - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) attributes.register_attribute(Foo, 'bars', uselist=True, callable_=lazyload, trackparent=True, useobject=True) bar1, bar2, bar3, bar4 = [Bar(id=1), Bar(id=2), Bar(id=3), Bar(id=4)] @@ -1460,7 +1460,7 @@ class HistoryTest(_base.ORMTest): return lazy_load return load - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'bar', uselist=False, callable_=lazyload, useobject=False) lazy_load = "hi" @@ -1498,7 +1498,7 @@ class HistoryTest(_base.ORMTest): return lazy_load return load - attributes.register_class(Foo) + instrumentation.register_class(Foo) attributes.register_attribute(Foo, 'bar', uselist=False, callable_=lazyload, useobject=False, active_history=True) lazy_load = "hi" @@ -1538,8 +1538,8 @@ class HistoryTest(_base.ORMTest): return lazy_load return load - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) attributes.register_attribute(Foo, 'bar', uselist=False, callable_=lazyload, trackparent=True, useobject=True) bar1, bar2 = [Bar(id=1), Bar(id=2)] lazy_load = bar1 @@ -1589,8 +1589,8 @@ class ListenerTest(_base.ORMTest): def set(self, state, value, oldvalue, initiator): return value + " modified" - attributes.register_class(Foo) - attributes.register_class(Bar) + instrumentation.register_class(Foo) + instrumentation.register_class(Bar) attributes.register_attribute(Foo, 'data', uselist=False, useobject=False, extension=AlteringListener()) attributes.register_attribute(Foo, 'barlist', uselist=True, useobject=True, extension=AlteringListener()) attributes.register_attribute(Foo, 'barset', typecallable=set, uselist=True, useobject=True, extension=AlteringListener()) |
