File size: 1,703 Bytes
985c397 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | #!/usr/bin/env python3
# basic test script for TechDraw module
# creates a page and 1 view
import FreeCAD
import unittest
from .TechDrawTestUtilities import createPageWithSVGTemplate
from PySide import QtCore
class DrawViewPartTest(unittest.TestCase):
def setUp(self):
"""Creates a page"""
FreeCAD.newDocument("TDPart")
FreeCAD.setActiveDocument("TDPart")
FreeCAD.ActiveDocument = FreeCAD.getDocument("TDPart")
FreeCAD.ActiveDocument.addObject("Part::Box", "Box")
self.page = createPageWithSVGTemplate()
self.page.Scale = 5.0
# page.ViewObject.show() # unit tests run in console mode
print("DrawViewPart test: page created")
def tearDown(self):
print("DrawViewPart test finished")
FreeCAD.closeDocument("TDPart")
def testMakeDrawViewPart(self):
"""Tests if a view can be added to page"""
print("testing DrawViewPart")
view = FreeCAD.ActiveDocument.addObject("TechDraw::DrawViewPart", "View")
self.page.addView(view)
FreeCAD.ActiveDocument.View.Source = [FreeCAD.ActiveDocument.Box]
FreeCAD.ActiveDocument.recompute()
#wait for threads to complete before checking result
loop = QtCore.QEventLoop()
timer = QtCore.QTimer()
timer.setSingleShot(True)
timer.timeout.connect(loop.quit)
timer.start(2000) #2 second delay
loop.exec_()
edges = view.getVisibleEdges()
self.assertEqual(len(edges), 4, "DrawViewPart has wrong number of edges")
self.assertTrue("Up-to-date" in view.State, "DrawViewPart is not Up-to-date")
if __name__ == "__main__":
unittest.main()
|