src/ModelAPI/Test/TestUndoRedo.py

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