Merge pull request #9921 from wwmayer/test_for_pr_9905

Part: add a unit test for PR #9905

src/Mod/Part/CMakeLists.txt

@@ -66,6 +66,7 @@ set(Part_tests
     parttests/part_test_objects.py
     parttests/regression_tests.py
     parttests/TopoShapeListTest.py
+    parttests/ColorPerFaceTest.py
 )
 
 add_custom_target(PartScripts ALL SOURCES

src/Mod/Part/TestPartGui.py

@@ -43,6 +43,7 @@ def findDockWidget(name):
 # define the test cases to test the FreeCAD Part module
 #---------------------------------------------------------------------------
 """
+from parttests.ColorPerFaceTest import ColorPerFaceTest
 
 
 #class PartGuiTestCases(unittest.TestCase):

src/Mod/Part/parttests/ColorPerFaceTest.py

@@ -0,0 +1,86 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# test to check color per face when after restore
+
+import FreeCAD as App
+import Part
+import os
+import tempfile
+import unittest
+from pivy import coin
+
+class ColorPerFaceTest(unittest.TestCase):
+    def setUp(self):
+        TempPath = tempfile.gettempdir()
+        self.fileName = TempPath + os.sep + "ColorPerFaceTest.FCStd"
+        self.doc = App.newDocument()
+
+    def tearDown(self):
+        App.closeDocument(self.doc.Name)
+
+    def testBox(self):
+        box = self.doc.addObject("Part::Box","Box")
+        self.doc.recompute()
+
+        box.ViewObject.DiffuseColor = [(1.,0.,0.,0.),
+                                       (1.,0.,0.,0.),
+                                       (1.,0.,0.,0.),
+                                       (1.,0.,0.,0.),
+                                       (1.,1.,0.,0.),
+                                       (1.,1.,0.,0.)]
+
+        box.Visibility = False
+        self.doc.recompute()
+
+        self.doc.saveAs(self.fileName)
+        App.closeDocument(self.doc.Name)
+
+        self.doc = App.openDocument(self.fileName)
+        box = self.doc.Box
+        box.Visibility = True
+        self.assertEqual(len(box.ViewObject.DiffuseColor), 6)
+
+        sa = coin.SoSearchAction()
+        sa.setType(coin.SoMaterial.getClassTypeId())
+        # We need an easier way to access nodes of a display mode
+        sa.setInterest(coin.SoSearchAction.ALL)
+        sa.apply(box.ViewObject.RootNode)
+        paths = sa.getPaths()
+
+        mat = paths.get(2).getTail()
+        self.assertEqual(mat.diffuseColor.getNum(), 6)
+
+    def testBoxAndLink(self):
+        box = self.doc.addObject("Part::Box","Box")
+        self.doc.recompute()
+
+        box.ViewObject.DiffuseColor = [(1.,0.,0.,0.),
+                                       (1.,0.,0.,0.),
+                                       (1.,0.,0.,0.),
+                                       (1.,0.,0.,0.),
+                                       (1.,1.,0.,0.),
+                                       (1.,1.,0.,0.)]
+
+        link = self.doc.addObject('App::Link','Link')
+        link.setLink(box)
+        box.Visibility = False
+        self.doc.recompute()
+
+        self.doc.saveAs(self.fileName)
+        App.closeDocument(self.doc.Name)
+
+        self.doc = App.openDocument(self.fileName)
+        box = self.doc.Box
+        box.Visibility = True
+        self.assertEqual(len(box.ViewObject.DiffuseColor), 6)
+
+        sa = coin.SoSearchAction()
+        sa.setType(coin.SoMaterial.getClassTypeId())
+        # We need an easier way to access nodes of a display mode
+        sa.setInterest(coin.SoSearchAction.ALL)
+        sa.apply(box.ViewObject.RootNode)
+        paths = sa.getPaths()
+
+        mat = paths.get(2).getTail()
+        self.assertEqual(mat.diffuseColor.getNum(), 6)