Test corrected according to changes in the API

[modules/shaper.git] / src / ModelAPI / Test / TestUndoRedo.py

diff --git a/src/ModelAPI/Test/TestUndoRedo.py b/src/ModelAPI/Test/TestUndoRedo.py
index 74a1315cd9bc5b72ca61fbcf31ac5bf8467de6e2..1d0a37209b80cba0da6721496cb0f3d4a349a71b 100644 (file)
--- a/src/ModelAPI/Test/TestUndoRedo.py
+++ b/src/ModelAPI/Test/TestUndoRedo.py
@@ -1,26 +1,33 @@
 from ModelAPI import *\r
-plugin_manager = ModelAPI_PluginManager.get()\r
-doc = plugin_manager.rootDocument()\r
-assert(not doc.canUndo())\r
-assert(not doc.canRedo()) \r
+aSession = ModelAPI_Session.get()\r
+aDoc = aSession.moduleDocument()\r
+assert(not aSession.canUndo())\r
+assert(not aSession.canRedo())\r
 \r
-doc.startOperation()\r
-feature = doc.addFeature("Point")\r
-feature_name = feature.data().name()\r
-assert(feature_name == "Point_1")\r
+aSession.startOperation()\r
+aFeature = aDoc.addFeature("Point")\r
+aFeatureData = aFeature.data()\r
+# Since validators are introduced we have to initialize all\r
+# the feature's attributes\r
+aFeatureData.real("x").setValue(1.)\r
+aFeatureData.real("y").setValue(-1.)\r
+aFeatureData.real("z").setValue(0.)\r
+aFeatureName = aFeatureData.name()\r
+assert(aFeatureName == "Point_1")\r
 \r
-feature.execute()\r
-doc.finishOperation();\r
-assert(doc.size("Construction") == 1)\r
-assert(doc.canUndo())\r
-assert(not doc.canRedo()) \r
+aFeature.execute()\r
+aSession.finishOperation()\r
 \r
-doc.undo()\r
-assert(doc.size("Construction") == 0)\r
-assert(not doc.canUndo())\r
-assert(doc.canRedo())\r
+assert(aDoc.size("Construction") == 1)\r
+assert(aSession.canUndo())\r
+assert(not aSession.canRedo())\r
 \r
-doc.redo()\r
-assert(doc.size("Construction") == 1)\r
-assert(doc.canUndo())\r
-assert(not doc.canRedo())\r
+aSession.undo()\r
+assert(aDoc.size("Construction") == 0)\r
+assert(not aSession.canUndo())\r
+assert(aSession.canRedo())\r
+\r
+aSession.redo()\r
+assert(aDoc.size("Construction") == 1)\r
+assert(aSession.canUndo())\r
+assert(not aSession.canRedo())\r