From 831c69e7eb9ea7c81272ab5db5ef5a85eac2820c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stefan=20Tr=C3=B6ger?= Date: Sat, 3 Dec 2016 11:21:15 +0100 Subject: [PATCH] Extensions: Fix ViewProvider extension test --- src/Mod/Test/Document.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/Mod/Test/Document.py b/src/Mod/Test/Document.py index 4c90cc77a7..b960dac7a8 100644 --- a/src/Mod/Test/Document.py +++ b/src/Mod/Test/Document.py @@ -237,22 +237,23 @@ class DocumentBasicCases(unittest.TestCase): def testExtensionBugViewProvider(self): class Layer(): - def __init__(self, obj): - ''' Add the properties ''' - obj.addExtension("App::GroupExtensionPython", self) + def __init__(self, obj): + obj.addExtension("App::GroupExtensionPython", self) class LayerViewProvider(): def __init__(self, obj): - ''' Add the properties ''' obj.addExtension("Gui::ViewProviderGroupExtensionPython", self) obj.Proxy = self - obj = FreeCAD.ActiveDocument.addObject("App::FeaturePython","Layer") + obj = self.Doc.addObject("App::FeaturePython","Layer") Layer(obj) LayerViewProvider(obj.ViewObject) self.failUnless(obj.hasExtension("App::GroupExtension")) self.failUnless(obj.ViewObject.hasExtension("Gui::ViewProviderGroupExtension")) self.failUnless(obj.ViewObject.hasExtension("Gui::ViewProviderGroupExtensionPython")) + + self.Doc.removeObject(obj.Name) + del obj def tearDown(self): #closing doc