FreeCAD/src/Mod/Test/Workbench.py
2014-02-25 15:24:10 +01:00

72 lines
3.7 KiB
Python
Executable File

# Workbench test module
# (c) 2006 Werner Mayer
#
#***************************************************************************
#* (c) Werner Mayer <werner.wm.mayer@gmx.de> 2006 *
#* *
#* This file is part of the FreeCAD CAx development system. *
#* *
#* This program is free software; you can redistribute it and/or modify *
#* it under the terms of the GNU Lesser General Public License (LGPL) *
#* as published by the Free Software Foundation; either version 2 of *
#* the License, or (at your option) any later version. *
#* for detail see the LICENCE text file. *
#* *
#* FreeCAD is distributed in the hope that it will be useful, *
#* but WITHOUT ANY WARRANTY; without even the implied warranty of *
#* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
#* GNU Library General Public License for more details. *
#* *
#* You should have received a copy of the GNU Library General Public *
#* License along with FreeCAD; if not, write to the Free Software *
#* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
#* USA *
#* *
#* Werner Mayer 2006 *
#***************************************************************************/
import FreeCAD, FreeCADGui, os, unittest
class WorkbenchTestCase(unittest.TestCase):
def setUp(self):
self.Active = FreeCADGui.activeWorkbench()
FreeCAD.Console.PrintLog(FreeCADGui.activeWorkbench().name())
def testActivate(self):
list=FreeCADGui.listWorkbenches()
try:
for i in list:
FreeCADGui.activateWorkbench(i)
FreeCADGui.updateGui()
FreeCAD.Console.PrintLog("Active: "+FreeCADGui.activeWorkbench().name()+ " Expected: "+i+"\n")
FreeCADGui.updateGui()
self.failUnless(FreeCADGui.activeWorkbench().name()==i, "Test on activating workbench failed")
except Exception, e:
self.failUnless(False, "Loading of workbench '%s' failed: %s" % (i, e.message))
def testHandler(self):
import __main__
class UnitWorkbench(__main__.Workbench):
MenuText = "Unittest"
ToolTip = "Unittest"
def Initialize(self):
list = ["Test_Test"]
self.appendToolbar("My Unittest",list)
def GetClassName(self):
return "Gui::PythonWorkbench"
FreeCADGui.addWorkbench(UnitWorkbench())
list=FreeCADGui.listWorkbenches()
self.failUnless(list.has_key("UnitWorkbench")==True, "Test on adding workbench handler failed")
FreeCADGui.activateWorkbench("UnitWorkbench")
FreeCADGui.updateGui()
self.failUnless(FreeCADGui.activeWorkbench().name()=="UnitWorkbench", "Test on loading workbench 'Unittest' failed")
FreeCADGui.removeWorkbench("UnitWorkbench")
list=FreeCADGui.listWorkbenches()
self.failUnless(list.has_key("UnitWorkbench")==False, "Test on removing workbench handler failed")
def tearDown(self):
FreeCADGui.activateWorkbench(self.Active.name())
FreeCAD.Console.PrintLog(self.Active.name())