src/ModelAPI/Test/TestUndoRedo.py

   1 from ModelAPI import *\r
   2 aSession = ModelAPI_Session.get()\r
   3 aDoc = aSession.moduleDocument()\r
   4 assert(not aSession.canUndo())\r
   5 assert(not aSession.canRedo())\r
   6 \r
   7 aSession.startOperation()\r
   8 aFeature = aDoc.addFeature("Point")\r
   9 # Since validators are introduced we have to initialize all\r
  10 # the feature's attributes\r
  11 aFeature.real("x").setValue(1.)\r
  12 aFeature.real("y").setValue(-1.)\r
  13 aFeature.real("z").setValue(0.)\r
  14 aFeatureName = aFeature.name()\r
  15 assert(aFeatureName == "Point_1")\r
  16 \r
  17 aFeature.execute()\r
  18 aSession.finishOperation()\r
  19 \r
  20 assert(aDoc.size("Construction") == 1)\r
  21 assert(aSession.canUndo())\r
  22 assert(not aSession.canRedo())\r
  23 \r
  24 aSession.undo()\r
  25 assert(aDoc.size("Construction") == 0)\r
  26 assert(not aSession.canUndo())\r
  27 assert(aSession.canRedo())\r
  28 \r
  29 aSession.redo()\r
  30 assert(aDoc.size("Construction") == 1)\r
  31 assert(aSession.canUndo())\r
  32 assert(not aSession.canRedo())\r