X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FModelAPI%2FTest%2FTestUndoRedo.py;h=b9701a1b9639fe0e3a627af6be6dc563bcfd0d36;hb=549138c2302de10c21a72b8c3ac4da9694874da9;hp=74a1315cd9bc5b72ca61fbcf31ac5bf8467de6e2;hpb=151a5f74aff7af71db009da99ef5bf112ad1de91;p=modules%2Fshaper.git

diff --git a/src/ModelAPI/Test/TestUndoRedo.py b/src/ModelAPI/Test/TestUndoRedo.py
index 74a1315cd..b9701a1b9 100644
--- a/src/ModelAPI/Test/TestUndoRedo.py
+++ b/src/ModelAPI/Test/TestUndoRedo.py
@@ -1,26 +1,34 @@
 from ModelAPI import *
-plugin_manager = ModelAPI_PluginManager.get()
-doc = plugin_manager.rootDocument()
-assert(not doc.canUndo())
-assert(not doc.canRedo()) 
+aSession = ModelAPI_Session.get()
+aDoc = aSession.moduleDocument()
+assert(not aSession.canUndo())
+assert(not aSession.canRedo())
 
-doc.startOperation()
-feature = doc.addFeature("Point")
-feature_name = feature.data().name()
-assert(feature_name == "Point_1")
+aSession.startOperation()
+aFeature = aDoc.addFeature("Point")
+# Since validators are introduced we have to initialize all
+# the feature's attributes
+aFeature.string("creation_method").setValue("by_xyz")
+aFeature.real("x").setValue(1.)
+aFeature.real("y").setValue(-1.)
+aFeature.real("z").setValue(0.)
+aFeatureName = aFeature.name()
+# "2" is because Origin is the first point
+assert(aFeatureName == "Point_2")
 
-feature.execute()
-doc.finishOperation();
-assert(doc.size("Construction") == 1)
-assert(doc.canUndo())
-assert(not doc.canRedo()) 
+aFeature.execute()
+aSession.finishOperation()
 
-doc.undo()
-assert(doc.size("Construction") == 0)
-assert(not doc.canUndo())
-assert(doc.canRedo())
+assert(aDoc.size("Construction") == 8)
+assert(aSession.canUndo())
+assert(not aSession.canRedo())
 
-doc.redo()
-assert(doc.size("Construction") == 1)
-assert(doc.canUndo())
-assert(not doc.canRedo())
+aSession.undo()
+assert(aDoc.size("Construction") == 7)
+assert(not aSession.canUndo())
+assert(aSession.canRedo())
+
+aSession.redo()
+assert(aDoc.size("Construction") == 8)
+assert(aSession.canUndo())
+assert(not aSession.canRedo())