def test_element_factory_undo(self):
from gaphor.UML.element import Element
ef = self.element_factory
ef.flush()
undo_manager = UndoManager()
undo_manager.init(Application)
undo_manager.begin_transaction()
p = ef.create(Element)
assert undo_manager._current_transaction
assert undo_manager._current_transaction._actions
assert undo_manager.can_undo()
undo_manager.commit_transaction()
assert undo_manager.can_undo()
assert ef.size() == 1
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
assert ef.size() == 0
undo_manager.redo_transaction()
assert undo_manager.can_undo()
assert not undo_manager.can_redo()
assert ef.size() == 1
assert ef.lselect()[0] is p
undo_manager.shutdown()
def test_element_factory_undo(self):
from gaphor.UML.element import Element
ef = self.element_factory
ef.flush()
undo_manager = UndoManager()
undo_manager.init(Application)
undo_manager.begin_transaction()
p = ef.create(Element)
assert undo_manager._current_transaction
assert undo_manager._current_transaction._actions
assert undo_manager.can_undo()
undo_manager.commit_transaction()
assert undo_manager.can_undo()
assert ef.size() == 1
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
assert ef.size() == 0
undo_manager.redo_transaction()
assert undo_manager.can_undo()
assert not undo_manager.can_redo()
assert ef.size() == 1
assert ef.lselect()[0] is p
undo_manager.shutdown()
def test_element_factory_undo(self):
from gaphor.core.modeling import Element
event_manager = EventManager()
undo_manager = UndoManager(event_manager)
element_factory = ElementFactory(event_manager)
undo_manager.begin_transaction()
p = element_factory.create(Element)
assert undo_manager._current_transaction
assert undo_manager._current_transaction._actions
assert undo_manager.can_undo()
undo_manager.commit_transaction()
assert undo_manager.can_undo()
assert element_factory.size() == 1
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
assert element_factory.size() == 0
undo_manager.redo_transaction()
assert undo_manager.can_undo()
assert not undo_manager.can_redo()
assert element_factory.size() == 1
assert element_factory.lselect()[0] is p
undo_manager.shutdown()
def test_uml_associations(self):
from gaphor.core.modeling.event import AssociationUpdated
from gaphor.core.modeling.properties import association, derivedunion
from gaphor.UML import Element
event_manager = EventManager()
undo_manager = UndoManager(event_manager)
element_factory = ElementFactory(event_manager)
class A(Element):
is_unlinked = False
def unlink(self):
self.is_unlinked = True
Element.unlink(self)
A.a1 = association("a1", A, upper=1)
A.a2 = association("a2", A, upper=1)
A.b1 = association("b1", A, upper="*")
A.b2 = association("b2", A, upper="*")
A.b3 = association("b3", A, upper=1)
A.derived_a = derivedunion("derived_a", A, 0, 1, A.a1, A.a2)
A.derived_b = derivedunion("derived_b", A, 0, "*", A.b1, A.b2, A.b3)
events = []
@event_handler(AssociationUpdated)
def handler(event, events=events):
events.append(event)
event_manager.subscribe(handler)
try:
a = element_factory.create(A)
undo_manager.begin_transaction()
a.a1 = element_factory.create(A)
undo_manager.commit_transaction()
assert (
len(events) == 2
), events # both AssociationSet and DerivedSet events
assert events[0].property is A.a1
assert undo_manager.can_undo()
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
assert len(events) == 4, events
assert events[2].property is A.a1
finally:
event_manager.unsubscribe(handler)
undo_manager.shutdown()
def test_uml_associations(self):
from zope import component
from gaphor.UML.interfaces import IAssociationChangeEvent
from gaphor.UML.properties import association, derivedunion
from gaphor.UML import Element
class A(Element):
is_unlinked = False
def unlink(self):
self.is_unlinked = True
Element.unlink(self)
A.a1 = association('a1', A, upper=1)
A.a2 = association('a2', A, upper=1)
A.b1 = association('b1', A, upper='*')
A.b2 = association('b2', A, upper='*')
A.b3 = association('b3', A, upper=1)
A.derived_a = derivedunion('derived_a', 0, 1, A.a1, A.a2)
A.derived_b = derivedunion('derived_b', 0, '*', A.b1, A.b2, A.b3)
events = []
@component.adapter(IAssociationChangeEvent)
def handler(event, events=events):
events.append(event)
compreg = Application.get_service('component_registry')
compreg.register_handler(handler)
try:
a = A()
undo_manager = UndoManager()
undo_manager.init(Application)
undo_manager.begin_transaction()
a.a1 = A()
undo_manager.commit_transaction()
assert len(events) == 1, events
assert events[0].property is A.a1
assert undo_manager.can_undo()
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
assert len(events) == 2, events
assert events[1].property is A.a1
finally:
compreg.unregister_handler(handler)
undo_manager.shutdown()
def test_uml_associations(self):
from gaphor.UML.event import AssociationChangeEvent
from gaphor.UML.properties import association, derivedunion
from gaphor.UML import Element
class A(Element):
is_unlinked = False
def unlink(self):
self.is_unlinked = True
Element.unlink(self)
A.a1 = association("a1", A, upper=1)
A.a2 = association("a2", A, upper=1)
A.b1 = association("b1", A, upper="*")
A.b2 = association("b2", A, upper="*")
A.b3 = association("b3", A, upper=1)
A.derived_a = derivedunion("derived_a", 0, 1, A.a1, A.a2)
A.derived_b = derivedunion("derived_b", 0, "*", A.b1, A.b2, A.b3)
events = []
@event_handler(AssociationChangeEvent)
def handler(event, events=events):
events.append(event)
compreg = Application.get_service("component_registry")
compreg.register_handler(handler)
try:
a = A(factory=self.element_factory)
undo_manager = UndoManager()
undo_manager.init(Application)
undo_manager.begin_transaction()
a.a1 = A(factory=self.element_factory)
undo_manager.commit_transaction()
assert len(events) == 1, events
assert events[0].property is A.a1
assert undo_manager.can_undo()
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
assert len(events) == 2, events
assert events[1].property is A.a1
finally:
compreg.unregister_handler(handler)
undo_manager.shutdown()
def test_uml_associations(self):
from gaphor.UML.event import AssociationChangeEvent
from gaphor.UML.properties import association, derivedunion
from gaphor.UML import Element
class A(Element):
is_unlinked = False
def unlink(self):
self.is_unlinked = True
Element.unlink(self)
A.a1 = association("a1", A, upper=1)
A.a2 = association("a2", A, upper=1)
A.b1 = association("b1", A, upper="*")
A.b2 = association("b2", A, upper="*")
A.b3 = association("b3", A, upper=1)
A.derived_a = derivedunion("derived_a", 0, 1, A.a1, A.a2)
A.derived_b = derivedunion("derived_b", 0, "*", A.b1, A.b2, A.b3)
events = []
@event_handler(AssociationChangeEvent)
def handler(event, events=events):
events.append(event)
em = Application.get_service("event_manager")
em.subscribe(handler)
try:
a = A(factory=self.element_factory)
undo_manager = UndoManager()
undo_manager.init(Application)
undo_manager.begin_transaction()
a.a1 = A(factory=self.element_factory)
undo_manager.commit_transaction()
assert len(events) == 1, events
assert events[0].property is A.a1
assert undo_manager.can_undo()
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
assert len(events) == 2, events
assert events[1].property is A.a1
finally:
em.unsubscribe(handler)
undo_manager.shutdown()
def test_uml_associations(self):
from zope import component
from gaphor.UML.interfaces import IAssociationChangeEvent
from gaphor.UML.properties import association, derivedunion
from gaphor.UML import Element
class A(Element):
is_unlinked = False
def unlink(self):
self.is_unlinked = True
Element.unlink(self)
A.a1 = association('a1', A, upper=1)
A.a2 = association('a2', A, upper=1)
A.b1 = association('b1', A, upper='*')
A.b2 = association('b2', A, upper='*')
A.b3 = association('b3', A, upper=1)
A.derived_a = derivedunion('derived_a', 0, 1, A.a1, A.a2)
A.derived_b = derivedunion('derived_b', 0, '*', A.b1, A.b2, A.b3)
events = []
@component.adapter(IAssociationChangeEvent)
def handler(event, events=events):
events.append(event)
compreg = Application.get_service('component_registry')
compreg.register_handler(handler)
try:
a = A()
undo_manager = UndoManager()
undo_manager.init(Application)
undo_manager.begin_transaction()
a.a1 = A()
undo_manager.commit_transaction()
assert len(events) == 1, events
assert events[0].property is A.a1
assert undo_manager.can_undo()
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
assert len(events) == 2, events
assert events[1].property is A.a1
finally:
compreg.unregister_handler(handler)
undo_manager.shutdown()
def test_element_factory_rollback(self):
from gaphor.UML.elementfactory import ElementFactory
from gaphor.UML.element import Element
undo_manager = UndoManager()
undo_manager.init(Application)
undo_manager.begin_transaction()
ef = ElementFactory()
ef.init(Application)
p = ef.create(Element)
assert undo_manager._current_transaction
assert undo_manager._current_transaction._actions
assert undo_manager.can_undo()
undo_manager.rollback_transaction()
assert not undo_manager.can_undo()
assert ef.size() == 0
undo_manager.shutdown()
def test_element_factory_rollback(self):
from gaphor.core.modeling import Element
event_manager = EventManager()
undo_manager = UndoManager(event_manager)
element_factory = ElementFactory(event_manager)
undo_manager.begin_transaction()
element_factory.create(Element)
assert undo_manager._current_transaction
assert undo_manager._current_transaction._actions
assert undo_manager.can_undo()
undo_manager.rollback_transaction()
assert not undo_manager.can_undo()
assert element_factory.size() == 0
undo_manager.shutdown()
def test_redo_stack(self):
from gaphor.core.modeling import Element
event_manager = EventManager()
undo_manager = UndoManager(event_manager)
element_factory = ElementFactory(event_manager)
undo_manager.begin_transaction()
p = element_factory.create(Element)
assert undo_manager._current_transaction
assert undo_manager._current_transaction._actions
assert undo_manager.can_undo()
undo_manager.commit_transaction()
assert undo_manager.can_undo()
assert element_factory.size() == 1, element_factory.size()
with Transaction(event_manager):
element_factory.create(Element)
assert undo_manager.can_undo()
assert not undo_manager.can_redo()
assert element_factory.size() == 2
undo_manager.undo_transaction()
assert undo_manager.can_undo()
assert 1 == len(undo_manager._undo_stack)
assert 1 == len(undo_manager._redo_stack)
assert undo_manager.can_redo()
assert element_factory.size() == 1
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
assert 0 == len(undo_manager._undo_stack)
assert 2 == len(undo_manager._redo_stack)
# assert element_factory.size() == 0
undo_manager.redo_transaction()
assert 1 == len(undo_manager._undo_stack)
assert 1 == len(undo_manager._redo_stack)
assert undo_manager.can_undo()
assert undo_manager.can_redo()
assert element_factory.size() == 1
undo_manager.redo_transaction()
assert undo_manager.can_undo()
assert not undo_manager.can_redo()
assert element_factory.size() == 2
assert p in element_factory.lselect()
undo_manager.shutdown()
def test_redo_stack(self):
from gaphor.UML.element import Element
undo_manager = UndoManager()
undo_manager.init(Application)
undo_manager.begin_transaction()
ef = self.element_factory
ef.flush()
p = ef.create(Element)
assert undo_manager._current_transaction
assert undo_manager._current_transaction._actions
assert undo_manager.can_undo()
undo_manager.commit_transaction()
assert undo_manager.can_undo()
assert ef.size() == 1, ef.size()
with Transaction():
q = ef.create(Element)
assert undo_manager.can_undo()
assert not undo_manager.can_redo()
assert ef.size() == 2
undo_manager.undo_transaction()
assert undo_manager.can_undo()
self.assertEqual(1, len(undo_manager._undo_stack))
self.assertEqual(1, len(undo_manager._redo_stack))
assert undo_manager.can_redo()
assert ef.size() == 1
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
self.assertEqual(0, len(undo_manager._undo_stack))
self.assertEqual(2, len(undo_manager._redo_stack))
# assert ef.size() == 0
undo_manager.redo_transaction()
self.assertEqual(1, len(undo_manager._undo_stack))
self.assertEqual(1, len(undo_manager._redo_stack))
assert undo_manager.can_undo()
assert undo_manager.can_redo()
assert ef.size() == 1
undo_manager.redo_transaction()
assert undo_manager.can_undo()
assert not undo_manager.can_redo()
assert ef.size() == 2
assert p in ef.lselect()
undo_manager.shutdown()
def test_actions(self):
undone = [0]
def undo_action(undone=undone):
#print 'undo_action called'
undone[0] = 1
undo_manager.add_undo_action(redo_action)
def redo_action(undone=undone):
#print 'redo_action called'
undone[0] = -1
undo_manager.add_undo_action(undo_action)
undo_manager = UndoManager()
undo_manager.init(Application)
#undo_manager.begin_transaction()
tx = Transaction()
undo_manager.add_undo_action(undo_action)
assert undo_manager._current_transaction
assert undo_manager.can_undo()
assert len(undo_manager._current_transaction._actions) == 1
#undo_manager.commit_transaction()
tx.commit()
undo_manager.undo_transaction()
assert not undo_manager.can_undo(), undo_manager._undo_stack
assert undone[0] == 1, undone
undone[0] = 0
assert undo_manager.can_redo(), undo_manager._redo_stack
undo_manager.redo_transaction()
assert not undo_manager.can_redo()
assert undo_manager.can_undo()
assert undone[0] == -1, undone
undo_manager.shutdown()
def test_actions(self):
undone = [0]
def undo_action(undone=undone):
# print 'undo_action called'
undone[0] = 1
undo_manager.add_undo_action(redo_action)
def redo_action(undone=undone):
# print 'redo_action called'
undone[0] = -1
undo_manager.add_undo_action(undo_action)
undo_manager = UndoManager()
undo_manager.init(Application)
# undo_manager.begin_transaction()
tx = Transaction()
undo_manager.add_undo_action(undo_action)
assert undo_manager._current_transaction
assert undo_manager.can_undo()
assert len(undo_manager._current_transaction._actions) == 1
# undo_manager.commit_transaction()
tx.commit()
undo_manager.undo_transaction()
assert not undo_manager.can_undo(), undo_manager._undo_stack
assert undone[0] == 1, undone
undone[0] = 0
assert undo_manager.can_redo(), undo_manager._redo_stack
undo_manager.redo_transaction()
assert not undo_manager.can_redo()
assert undo_manager.can_undo()
assert undone[0] == -1, undone
undo_manager.shutdown()
def test_redo_stack(self):
from gaphor.UML.element import Element
undo_manager = UndoManager()
undo_manager.init(Application)
undo_manager.begin_transaction()
ef = self.element_factory
ef.flush()
p = ef.create(Element)
assert undo_manager._current_transaction
assert undo_manager._current_transaction._actions
assert undo_manager.can_undo()
undo_manager.commit_transaction()
assert undo_manager.can_undo()
assert ef.size() == 1, ef.size()
with Transaction():
q = ef.create(Element)
assert undo_manager.can_undo()
assert not undo_manager.can_redo()
assert ef.size() == 2
undo_manager.undo_transaction()
assert undo_manager.can_undo()
self.assertEqual(1, len(undo_manager._undo_stack))
self.assertEqual(1, len(undo_manager._redo_stack))
assert undo_manager.can_redo()
assert ef.size() == 1
undo_manager.undo_transaction()
assert not undo_manager.can_undo()
assert undo_manager.can_redo()
self.assertEqual(0, len(undo_manager._undo_stack))
self.assertEqual(2, len(undo_manager._redo_stack))
# assert ef.size() == 0
undo_manager.redo_transaction()
self.assertEqual(1, len(undo_manager._undo_stack))
self.assertEqual(1, len(undo_manager._redo_stack))
assert undo_manager.can_undo()
assert undo_manager.can_redo()
assert ef.size() == 1
undo_manager.redo_transaction()
assert undo_manager.can_undo()
assert not undo_manager.can_redo()
assert ef.size() == 2
assert p in ef.lselect()
undo_manager.shutdown()
def test_not_in_transaction(self):
event_manager = EventManager()
undo_manager = UndoManager(event_manager)
action = object()
undo_manager.add_undo_action(action)
assert undo_manager._current_transaction is None
undo_manager.begin_transaction()
undo_manager.add_undo_action(action)
assert undo_manager._current_transaction
assert undo_manager.can_undo()
assert len(undo_manager._current_transaction._actions) == 1
undo_manager.shutdown()
def test_actions(self):
undone = [0]
def undo_action(undone=undone):
undone[0] = 1
undo_manager.add_undo_action(redo_action)
def redo_action(undone=undone):
undone[0] = -1
undo_manager.add_undo_action(undo_action)
event_manager = EventManager()
undo_manager = UndoManager(event_manager)
tx = Transaction(event_manager)
undo_manager.add_undo_action(undo_action)
assert undo_manager._current_transaction
assert undo_manager.can_undo()
assert len(undo_manager._current_transaction._actions) == 1
tx.commit()
undo_manager.undo_transaction()
assert not undo_manager.can_undo(), undo_manager._undo_stack
assert undone[0] == 1, undone
undone[0] = 0
assert undo_manager.can_redo(), undo_manager._redo_stack
undo_manager.redo_transaction()
assert not undo_manager.can_redo()
assert undo_manager.can_undo()
assert undone[0] == -1, undone
undo_manager.shutdown()
def test_not_in_transaction(self):
undo_manager = UndoManager()
undo_manager.init(Application)
action = object()
undo_manager.add_undo_action(action)
assert undo_manager._current_transaction is None
undo_manager.begin_transaction()
undo_manager.add_undo_action(action)
assert undo_manager._current_transaction
assert undo_manager.can_undo()
assert len(undo_manager._current_transaction._actions) == 1
undo_manager.shutdown()
