def test_setdefault(self):
data, wim = self._fixture()
o1 = self.Data()
oid1 = id(o1)
assert o1 not in wim
res1 = wim.setdefault(o1, 123)
assert o1 in wim
assert oid1 in wim.by_id
eq_(res1, 123)
id_keys = set(wim.by_id.keys())
res2 = wim.setdefault(o1, 456)
assert o1 in wim
assert oid1 in wim.by_id
eq_(res2, 123)
assert set(wim.by_id.keys()) == id_keys
del wim[o1]
assert o1 not in wim
assert oid1 not in wim.by_id
ne_(set(wim.by_id.keys()), id_keys)
res3 = wim.setdefault(o1, 789)
assert o1 in wim
assert oid1 in wim.by_id
eq_(res3, 789)
eq_(set(wim.by_id.keys()), id_keys)
def test_transient_exception(self):
"""An object that goes from a pk value to transient/pending
doesn't count as a "pk" switch.
"""
mapper(User, users)
mapper(Address, addresses, properties={'user': relationship(User)})
sess = create_session()
u1 = User(id=5, name='u1')
ad1 = Address(email_address='e1', user=u1)
sess.add_all([u1, ad1])
sess.flush()
make_transient(u1)
u1.id = None
u1.username = '******'
sess.add(u1)
sess.flush()
eq_(ad1.user_id, 5)
sess.expire_all()
eq_(ad1.user_id, 5)
ne_(u1.id, 5)
ne_(u1.id, None)
eq_(sess.query(User).count(), 2)
def test_setdefault(self):
data, wim = self._fixture()
o1 = self.Data()
oid1 = id(o1)
assert o1 not in wim
res1 = wim.setdefault(o1, 123)
assert o1 in wim
assert oid1 in wim.by_id
eq_(res1, 123)
id_keys = set(wim.by_id.keys())
res2 = wim.setdefault(o1, 456)
assert o1 in wim
assert oid1 in wim.by_id
eq_(res2, 123)
assert set(wim.by_id.keys()) == id_keys
del wim[o1]
assert o1 not in wim
assert oid1 not in wim.by_id
ne_(set(wim.by_id.keys()), id_keys)
res3 = wim.setdefault(o1, 789)
assert o1 in wim
assert oid1 in wim.by_id
eq_(res3, 789)
eq_(set(wim.by_id.keys()), id_keys)
def test_transient_exception(self):
"""An object that goes from a pk value to transient/pending
doesn't count as a "pk" switch.
"""
mapper(User, users)
mapper(Address, addresses, properties={'user':relationship(User)})
sess = create_session()
u1 = User(id=5, name='u1')
ad1 = Address(email_address='e1', user=u1)
sess.add_all([u1, ad1])
sess.flush()
make_transient(u1)
u1.id = None
u1.username='******'
sess.add(u1)
sess.flush()
eq_(ad1.user_id, 5)
sess.expire_all()
eq_(ad1.user_id, 5)
ne_(u1.id, 5)
ne_(u1.id, None)
eq_(sess.query(User).count(), 2)
def test_flush(self):
mapper(User, users, properties={
'addresses':dynamic_loader(mapper(Address, addresses))
})
sess = create_session()
u1 = User(name='jack')
u2 = User(name='ed')
u2.addresses.append(Address(email_address='*****@*****.**'))
u1.addresses.append(Address(email_address='*****@*****.**'))
sess.add_all((u1, u2))
sess.flush()
from sqlalchemy.orm import attributes
eq_(attributes.get_history(u1, 'addresses'), ([], [Address(email_address='*****@*****.**')], []))
sess.expunge_all()
# test the test fixture a little bit
ne_(User(name='jack', addresses=[Address(email_address='wrong')]),
sess.query(User).first())
eq_(User(name='jack', addresses=[Address(email_address='*****@*****.**')]),
sess.query(User).first())
eq_([
User(name='jack', addresses=[Address(email_address='*****@*****.**')]),
User(name='ed', addresses=[Address(email_address='*****@*****.**')])
],
sess.query(User).all())
def register(self, cls, canary):
original_init = cls.__init__
attributes.register_class(cls)
ne_(cls.__init__, original_init)
manager = attributes.manager_of_class(cls)
def on_init(state, instance, args, kwargs):
canary.append((cls, 'on_init', type(instance)))
manager.events.add_listener('on_init', on_init)
def register(self, cls, canary):
original_init = cls.__init__
attributes.register_class(cls)
ne_(cls.__init__, original_init)
manager = attributes.manager_of_class(cls)
def on_init(state, instance, args, kwargs):
canary.append((cls, 'on_init', type(instance)))
manager.events.add_listener('on_init', on_init)
def test_memoized_instancemethod(self):
val = [20]
class Foo(object):
@util.memoized_instancemethod
def bar(self):
v = val[0]
val[0] += 1
return v
ne_(Foo.bar, None)
f1 = Foo()
assert 'bar' not in f1.__dict__
eq_(f1.bar(), 20)
eq_(f1.bar(), 20)
eq_(val[0], 21)
def test_memoized_instancemethod(self):
val = [20]
class Foo(object):
@util.memoized_instancemethod
def bar(self):
v = val[0]
val[0] += 1
return v
ne_(Foo.bar, None)
f1 = Foo()
assert 'bar' not in f1.__dict__
eq_(f1.bar(), 20)
eq_(f1.bar(), 20)
eq_(val[0], 21)
def test_memoized_property(self):
val = [20]
class Foo(object):
@util.memoized_property
def bar(self):
v = val[0]
val[0] += 1
return v
ne_(Foo.bar, None)
f1 = Foo()
assert 'bar' not in f1.__dict__
eq_(f1.bar, 20)
eq_(f1.bar, 20)
eq_(val[0], 21)
eq_(f1.__dict__['bar'] , 20)
def test_memoized_property(self):
val = [20]
class Foo(object):
@util.memoized_property
def bar(self):
v = val[0]
val[0] += 1
return v
ne_(Foo.bar, None)
f1 = Foo()
assert 'bar' not in f1.__dict__
eq_(f1.bar, 20)
eq_(f1.bar, 20)
eq_(val[0], 21)
eq_(f1.__dict__['bar'], 20)
def test_pop_default(self):
data, wim = self._fixture()
needle = data[-1]
value = wim[needle]
x = wim.pop(needle, 123)
ne_(x, 123)
eq_(x, value)
assert needle not in wim
assert id(needle) not in wim.by_id
eq_(len(data), (len(wim) + 1))
n2 = self.Data()
y = wim.pop(n2, 456)
eq_(y, 456)
assert n2 not in wim
assert id(n2) not in wim.by_id
eq_(len(data), (len(wim) + 1))
def test_pop_default(self):
data, wim = self._fixture()
needle = data[-1]
value = wim[needle]
x = wim.pop(needle, 123)
ne_(x, 123)
eq_(x, value)
assert needle not in wim
assert id(needle) not in wim.by_id
eq_(len(data), (len(wim) + 1))
n2 = self.Data()
y = wim.pop(n2, 456)
eq_(y, 456)
assert n2 not in wim
assert id(n2) not in wim.by_id
eq_(len(data), (len(wim) + 1))
def test_nativeext_interfaceexact(self):
class A(object):
__sa_instrumentation_manager__ = sa.orm.interfaces.InstrumentationManager
attributes.register_class(A)
ne_(type(attributes.manager_of_class(A)), attributes.ClassManager)
def test_basic_sanity(self):
IdentitySet = util.IdentitySet
o1, o2, o3 = object(), object(), object()
ids = IdentitySet([o1])
ids.discard(o1)
ids.discard(o1)
ids.add(o1)
ids.remove(o1)
assert_raises(KeyError, ids.remove, o1)
eq_(ids.copy(), ids)
# explicit __eq__ and __ne__ tests
assert ids != None
assert not(ids == None)
ne_(ids, IdentitySet([o1,o2,o3]))
ids.clear()
assert o1 not in ids
ids.add(o2)
assert o2 in ids
eq_(ids.pop(), o2)
ids.add(o1)
eq_(len(ids), 1)
isuper = IdentitySet([o1,o2])
assert ids < isuper
assert ids.issubset(isuper)
assert isuper.issuperset(ids)
assert isuper > ids
eq_(ids.union(isuper), isuper)
eq_(ids | isuper, isuper)
eq_(isuper - ids, IdentitySet([o2]))
eq_(isuper.difference(ids), IdentitySet([o2]))
eq_(ids.intersection(isuper), IdentitySet([o1]))
eq_(ids & isuper, IdentitySet([o1]))
eq_(ids.symmetric_difference(isuper), IdentitySet([o2]))
eq_(ids ^ isuper, IdentitySet([o2]))
ids.update(isuper)
ids |= isuper
ids.difference_update(isuper)
ids -= isuper
ids.intersection_update(isuper)
ids &= isuper
ids.symmetric_difference_update(isuper)
ids ^= isuper
ids.update('foobar')
try:
ids |= 'foobar'
assert False
except TypeError:
assert True
try:
s = set([o1,o2])
s |= ids
assert False
except TypeError:
assert True
assert_raises(TypeError, util.cmp, ids)
assert_raises(TypeError, hash, ids)
def test_basic_sanity(self):
IdentitySet = util.IdentitySet
o1, o2, o3 = object(), object(), object()
ids = IdentitySet([o1])
ids.discard(o1)
ids.discard(o1)
ids.add(o1)
ids.remove(o1)
assert_raises(KeyError, ids.remove, o1)
eq_(ids.copy(), ids)
# explicit __eq__ and __ne__ tests
assert ids != None
assert not (ids == None)
ne_(ids, IdentitySet([o1, o2, o3]))
ids.clear()
assert o1 not in ids
ids.add(o2)
assert o2 in ids
eq_(ids.pop(), o2)
ids.add(o1)
eq_(len(ids), 1)
isuper = IdentitySet([o1, o2])
assert ids < isuper
assert ids.issubset(isuper)
assert isuper.issuperset(ids)
assert isuper > ids
eq_(ids.union(isuper), isuper)
eq_(ids | isuper, isuper)
eq_(isuper - ids, IdentitySet([o2]))
eq_(isuper.difference(ids), IdentitySet([o2]))
eq_(ids.intersection(isuper), IdentitySet([o1]))
eq_(ids & isuper, IdentitySet([o1]))
eq_(ids.symmetric_difference(isuper), IdentitySet([o2]))
eq_(ids ^ isuper, IdentitySet([o2]))
ids.update(isuper)
ids |= isuper
ids.difference_update(isuper)
ids -= isuper
ids.intersection_update(isuper)
ids &= isuper
ids.symmetric_difference_update(isuper)
ids ^= isuper
ids.update('foobar')
try:
ids |= 'foobar'
assert False
except TypeError:
assert True
try:
s = set([o1, o2])
s |= ids
assert False
except TypeError:
assert True
assert_raises(TypeError, util.cmp, ids)
assert_raises(TypeError, hash, ids)
def test_nativeext_interfaceexact(self):
class A(object):
__sa_instrumentation_manager__ = sa.orm.interfaces.InstrumentationManager
attributes.register_class(A)
ne_(type(attributes.manager_of_class(A)), attributes.ClassManager)
