from sqlobject import *
from sqlobject.tests.dbtest import *
########################################
## Lazy updates
########################################
class Lazy(SQLObject):
class sqlmeta:
lazyUpdate = True
name = StringCol()
other = StringCol(default='nothing')
third = StringCol(default='third')
class TestLazyTest:
def setup_method(self, meth):
# All this stuff is so that we can track when the connection
# does an actual update; we put in a new _SO_update method
# that calls the original and sets an instance variable that
# we can later check.
setupClass(Lazy)
self.conn = Lazy._connection
self.conn.didUpdate = False
self._oldUpdate = self.conn._SO_update
newUpdate = (
lambda so, values, s=self, c=self.conn, o=self._oldUpdate:
self._alternateUpdate(so, values, c, o))
self.conn._SO_update = newUpdate
def teardown_method(self, meth):
self.conn._SO_update = self._oldUpdate
del self._oldUpdate
def _alternateUpdate(self, so, values, conn, oldUpdate):
conn.didUpdate = True
return oldUpdate(so, values)
def test_lazy(self):
assert not self.conn.didUpdate
obj = Lazy(name='tim')
# We just did an insert, but not an update:
assert not self.conn.didUpdate
obj.set(name='joe')
assert obj.dirty
assert obj.name == 'joe'
assert not self.conn.didUpdate
obj.syncUpdate()
assert obj.name == 'joe'
assert self.conn.didUpdate
assert not obj.dirty
assert obj.name == 'joe'
self.conn.didUpdate = False
obj = Lazy(name='frank')
obj.name = 'joe'
assert not self.conn.didUpdate
assert obj.dirty
assert obj.name == 'joe'
obj.name = 'joe2'
assert not self.conn.didUpdate
assert obj.dirty
assert obj.name == 'joe2'
obj.syncUpdate()
assert obj.name == 'joe2'
assert not obj.dirty
assert self.conn.didUpdate
self.conn.didUpdate = False
obj = Lazy(name='loaded')
assert not obj.dirty
assert not self.conn.didUpdate
assert obj.name == 'loaded'
obj.name = 'unloaded'
assert obj.dirty
assert obj.name == 'unloaded'
assert not self.conn.didUpdate
obj.sync()
assert not obj.dirty
assert obj.name == 'unloaded'
assert self.conn.didUpdate
self.conn.didUpdate = False
obj.name = 'whatever'
assert obj.dirty
assert obj.name == 'whatever'
assert not self.conn.didUpdate
obj._SO_loadValue('name')
assert obj.dirty
assert obj.name == 'whatever'
assert not self.conn.didUpdate
obj._SO_loadValue('other')
assert obj.name == 'whatever'
assert not self.conn.didUpdate
obj.syncUpdate()
assert self.conn.didUpdate
self.conn.didUpdate = False
# Now, check that get() doesn't screw
# cached objects' validator state.
obj_id = obj.id
old_state = obj._SO_validatorState
obj = Lazy.get(obj_id)
assert not obj.dirty
assert not self.conn.didUpdate
assert obj._SO_validatorState is old_state
assert obj.name == 'whatever'
obj.name = 'unloaded'
assert obj.name == 'unloaded'
assert obj.dirty
assert not self.conn.didUpdate
# Fetch the object again with get() and
# make sure dirty is still set, as the
# object should come from the cache.
obj = Lazy.get(obj_id)
assert obj.dirty
assert not self.conn.didUpdate
assert obj.name == 'unloaded'
obj.syncUpdate()
assert self.conn.didUpdate
assert not obj.dirty
self.conn.didUpdate = False
# Then clear the cache, and try a get()
# again, to make sure stuf like _SO_createdValues
# is properly initialized.
self.conn.cache.clear()
obj = Lazy.get(obj_id)
assert not obj.dirty
assert not self.conn.didUpdate
assert obj.name == 'unloaded'
obj.name = 'spongebob'
assert obj.name == 'spongebob'
assert obj.dirty
assert not self.conn.didUpdate
obj.syncUpdate()
assert self.conn.didUpdate
assert not obj.dirty
self.conn.didUpdate = False
obj = Lazy(name='last')
assert not obj.dirty
obj.syncUpdate()
assert not self.conn.didUpdate
assert not obj.dirty
# Check that setting multiple values
# actually works. This was broken
# and just worked because we were testing
# only one value at a time, so 'name'
# had the right value after the for loop *wink*
# Also, check that passing a name that is not
# a valid column doesn't break, but instead
# just does a plain setattr.
obj.set(name='first', other='who', third='yes')
assert obj.name == 'first'
assert obj.other == 'who'
assert obj.third == 'yes'
assert obj.dirty
assert not self.conn.didUpdate
obj.syncUpdate()
assert self.conn.didUpdate
assert not obj.dirty
|