91 lines
4.0 KiB
Python
91 lines
4.0 KiB
Python
# ***************************************************************************
|
|
# * Copyright (c) 2005 Werner Mayer <werner.wm.mayer@gmx.de> *
|
|
# * *
|
|
# * 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 2005 *
|
|
# ***************************************************************************/
|
|
|
|
# Menu test module
|
|
|
|
import FreeCAD, os, unittest, FreeCADGui
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# define the functions to test the FreeCAD base code
|
|
# ---------------------------------------------------------------------------
|
|
|
|
# def suite():
|
|
# suite = unittest.TestSuite()
|
|
# suite.addTest(DocTestCase("DocumentProperties"))
|
|
# suite.addTest(DocTestCase("DocumentLabels"))
|
|
# suite.addTest(DocTestCase("DocumentSaveAndRestore"))
|
|
# return suite
|
|
|
|
|
|
class MenuCreateCases(unittest.TestCase):
|
|
def setUp(self):
|
|
FreeCADGui.activateWorkbench("TestWorkbench")
|
|
FreeCAD.Console.PrintLog("Setup Test menu...\n")
|
|
list = ["Test_TestAll", "Test_TestDoc", "Test_TestBase", "Test_TestWork"]
|
|
w = FreeCADGui.getWorkbench("TestWorkbench")
|
|
w.appendMenu("TestMenu", list)
|
|
|
|
def testMenu(self):
|
|
# check menu for items
|
|
FreeCAD.Console.PrintLog("Checking Test menu...\n")
|
|
w = FreeCADGui.getWorkbench("TestWorkbench")
|
|
list = w.listMenus()
|
|
self.b = False
|
|
for i in list:
|
|
if i == "TestMenu":
|
|
self.b = True
|
|
self.assertTrue(self.b, "Test menu not found")
|
|
|
|
def tearDown(self):
|
|
if self.b:
|
|
FreeCAD.Console.PrintLog("Test menu successfully added\n")
|
|
else:
|
|
FreeCAD.Console.PrintLog("Adding Test menu failed\n")
|
|
|
|
|
|
class MenuDeleteCases(unittest.TestCase):
|
|
def setUp(self):
|
|
FreeCADGui.activateWorkbench("TestWorkbench")
|
|
FreeCAD.Console.PrintLog("Remove Test menu...\n")
|
|
w = FreeCADGui.getWorkbench("TestWorkbench")
|
|
w.removeMenu("TestMenu")
|
|
|
|
def testMenu(self):
|
|
# check menu for items
|
|
FreeCAD.Console.PrintLog("Checking Test menu...\n")
|
|
w = FreeCADGui.getWorkbench("TestWorkbench")
|
|
list = w.listMenus()
|
|
self.b = True
|
|
for i in list:
|
|
if i == "TestMenu":
|
|
self.b = False
|
|
self.assertTrue(self.b == True, "Test menu still added")
|
|
|
|
def tearDown(self):
|
|
if self.b:
|
|
FreeCAD.Console.PrintLog("Test menu successfully removed\n")
|
|
else:
|
|
FreeCAD.Console.PrintLog("Removing Test menu failed\n")
|