summaryrefslogtreecommitdiff
path: root/test/orm/test_attributes.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2010-09-14 20:43:48 -0400
committerMike Bayer <mike_mp@zzzcomputing.com>2010-09-14 20:43:48 -0400
commit3d389b19b70b65cb76226c3f3aa4c5d926e1f12b (patch)
treebdc597ebf2786f0514c8029dd2dbe2385bb9c2df /test/orm/test_attributes.py
parent60b82d6e13246a3d88e7288e863a5231b7572c6a (diff)
downloadsqlalchemy-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.py134
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())