def testUndoAndRedoAreAssociated(self):
status = Status()
def inc_status(x):
status.count += x
def dec_status(x):
status.count -= x
def make_status_string(s):
status.count = s
def make_status_int(i):
status.count = i
self.assertEqual(status.count, 0)
# create sequence
undoer = undo.Sequence()
inc_status(1)
undoer.do(inc_status, 1, dec_status, 1)
make_status_string("foo")
undoer.do(make_status_string, "foo", make_status_int, 1)
self.assertEqual(status.count, "foo")
undoer.undo()
self.assertEqual(status.count, 1)
undoer.undo()
self.assertEqual(status.count, 0)
def testDoRemovesRedoStack(self):
status = Status()
def inc_status(x):
status.count += x
def dec_status(x):
status.count -= x
self.assertEqual(status.count, 0)
# perform 3 actions, undo 2, do 1 again
undoer = undo.Sequence()
inc_status(1)
undoer.do(inc_status, 1, dec_status, 1)
inc_status(1)
undoer.do(inc_status, 1, dec_status, 1)
inc_status(1)
undoer.do(inc_status, 1, dec_status, 1)
undoer.undo()
undoer.undo()
self.assertEqual(status.count, 1)
inc_status(1)
undoer.do(inc_status, 1, dec_status, 1)
self.assertEqual(len(undoer.history), 2)
self.assertEqual(undoer.can_redo(), False)
self.assertEqual(status.count, 2)
# undo again, check we're back where we started
undoer.undo()
self.assertEqual(status.count, 1)
def testCreateUndoRedo(self):
status = Status()
self.undo_cb_status = Status()
self.redo_cb_status = Status()
def inc_status(x):
status.count += x
def dec_status(x):
status.count -= x
def set_undoable(sequence, state):
# print "undo_status: %s" % state
self.undo_cb_status.count = state
def set_redoable(sequence, state):
# print "redo_status: %s" % state
self.redo_cb_status.count = state
self.assertEqual(status.count, 0)
self.assertEqual(self.undo_cb_status.count, 0)
self.assertEqual(self.redo_cb_status.count, 0)
# create sequence
undoer = undo.Sequence()
undoer.connect('can-undo', set_undoable)
undoer.connect('can-redo', set_redoable)
self.assertEqual(undoer.can_undo(), False)
self.assertEqual(undoer.can_redo(), False)
self.assertExternalAndInternalStatusMatch(undoer)
self.assertEqual(undoer.pos, 0)
self.assertEqual(len(undoer.history), 0)
# perform an action
inc_status(1)
undoer.do(inc_status, 1, dec_status, 1)
self.assertEqual(status.count, 1)
self.assertUndoStatus(undoer, True, False)
self.assertEqual(len(undoer.history), 1)
self.assertEqual(undoer.pos, 1)
# undo it
undoer.undo()
# check externals
self.assertEqual(status.count, 0)
self.assertUndoStatus(undoer, False, True)
# check internals
self.assertEqual(len(undoer.history), 1)
self.assertEqual(undoer.pos, 0)
# redo it
undoer.redo()
# check externals
self.assertEqual(status.count, 1)
self.assertUndoStatus(undoer, True, False)
# check internals
self.assertEqual(len(undoer.history), 1)
self.assertEqual(undoer.pos, 1)
def testInvalidOperationsThrow(self):
undoer = undo.Sequence()
self.assertRaises(ValueError, undoer.undo)
self.assertRaises(ValueError, undoer.redo)
def testCreateUndoRedo(self):
class Status1(Status):
def set_enabled(self, state):
self.count = state
status = Status1()
self.undo_cb_status = Status1()
self.redo_cb_status = Status1()
def inc_status(x):
status.count += x
def dec_status(x):
status.count -= x
self.assertEqual(status.count, 0)
self.assertEqual(self.undo_cb_status.count, 0)
self.assertEqual(self.redo_cb_status.count, 0)
# create sequence
undoer = undo.Sequence()
undoer.register_callbacks(self.redo_cb_status.set_enabled,
self.undo_cb_status.set_enabled)
self.assertEqual(undoer.can_undo(), False)
self.assertEqual(undoer.can_redo(), False)
self.assertExternalAndInternalStatusMatch(undoer)
self.assertEqual(undoer.pos, 0)
self.assertEqual(len(undoer.history), 0)
# perform an action
inc_status(1)
undoer.do(inc_status, 1, dec_status, 1)
self.assertEqual(status.count, 1)
self.assertUndoStatus(undoer, True, False)
self.assertEqual(len(undoer.history), 1)
self.assertEqual(undoer.pos, 1)
# undo it
undoer.undo()
# check externals
self.assertEqual(status.count, 0)
self.assertUndoStatus(undoer, False, True)
# check internals
self.assertEqual(len(undoer.history), 1)
self.assertEqual(undoer.pos, 0)
# redo it
undoer.redo()
# check externals
self.assertEqual(status.count, 1)
self.assertUndoStatus(undoer, True, False)
# check internals
self.assertEqual(len(undoer.history), 1)
self.assertEqual(undoer.pos, 1)