def test_add_entry_undo_precommit(session):
entry = DatabaseEntry()
cmd = AddEntry(session, entry)
cmd()
cmd.undo()
session.commit()
assert session.query(DatabaseEntry).count() == 0
def test_add_entry_undo_precommit(session):
entry = DatabaseEntry()
cmd = AddEntry(session, entry)
cmd()
cmd.undo()
session.commit()
assert session.query(DatabaseEntry).count() == 0
def test_add_entry_undo(session):
entry = DatabaseEntry()
cmd = AddEntry(session, entry)
cmd()
assert session.query(DatabaseEntry).count() == 1
assert entry.id == 1
cmd.undo()
assert entry in session.deleted
assert session.query(DatabaseEntry).count() == 0
def test_add_entry_undo(session):
entry = DatabaseEntry()
cmd = AddEntry(session, entry)
cmd()
assert session.query(DatabaseEntry).count() == 1
assert entry.id == 1
cmd.undo()
assert entry in session.deleted
assert session.query(DatabaseEntry).count() == 0
def test_add_removed_entry(session):
entry = DatabaseEntry()
AddEntry(session, entry)()
session.commit()
RemoveEntry(session, entry)()
session.commit()
AddEntry(session, entry)()
session.commit()
assert not session.new
assert entry.id == 1
def test_cmd_manager_redo_stack_empty_after_call(session, command_manager):
command_manager.do(AddEntry(session, DatabaseEntry()))
command_manager.do(AddEntry(session, DatabaseEntry()))
assert len(command_manager.undo_commands) == 2
session.commit()
command_manager.undo(2)
assert not command_manager.undo_commands
assert len(command_manager.redo_commands) == 2
command_manager.do(AddEntry(session, DatabaseEntry()))
assert not command_manager.redo_commands
def test_cmd_manager_redo(session, command_manager):
assert command_manager.undo_commands == []
assert command_manager.redo_commands == []
command_manager.do(AddEntry(session, DatabaseEntry()))
command_manager.do(AddEntry(session, DatabaseEntry()))
assert len(command_manager.undo_commands) == 2
assert command_manager.redo_commands == []
session.commit()
command_manager.undo(2)
assert command_manager.undo_commands == []
assert len(command_manager.redo_commands) == 2
command_manager.redo(2)
assert len(command_manager.undo_commands) == 2
assert command_manager.redo_commands == []
def test_undo_redo_multiple_cmds_at_once(session, command_manager):
assert command_manager.undo_commands == []
command_manager.do(CompositeOperation([
AddEntry(session, DatabaseEntry()),
AddEntry(session, DatabaseEntry()),
AddEntry(session, DatabaseEntry())]))
assert len(command_manager.undo_commands) == 1
assert session.query(DatabaseEntry).count() == 3
command_manager.undo()
assert command_manager.undo_commands == []
assert session.query(DatabaseEntry).count() == 0
command_manager.redo()
assert command_manager.redo_commands == []
assert session.query(DatabaseEntry).count() == 3
def test_add_entry_repr(session):
entry = DatabaseEntry(id=5)
repr_result = repr(AddEntry(session, entry))
expected_repr_result = (
'<AddEntry('
'session <sqlalchemy.orm.session.Session object at *>, '
'entry id 5)>'.format(id(session)))
assert fnmatch.fnmatch(repr_result, expected_repr_result)
def test_add_entry(session):
assert not session.new
entry = DatabaseEntry()
AddEntry(session, entry)()
assert len(session.new) == 1
assert entry.id is None
session.commit()
assert not session.new
assert entry.id == 1
def test_cmd_manager_pop_redo_cmd_empty_stack(session, command_manager):
cmd = AddEntry(session, DatabaseEntry())
command_manager.do(cmd)
command_manager.undo()
popped_cmd = command_manager.pop_redo_command()
assert popped_cmd == cmd
def test_cmd_manager_pop_undo_cmd(session, command_manager):
cmd = AddEntry(session, DatabaseEntry())
command_manager.do(cmd)
popped_cmd = command_manager.pop_undo_command()
assert popped_cmd == cmd
