From 97054ee4be1f4b6168ee8d980fd892ca652e5f73 Mon Sep 17 00:00:00 2001
From: Yorik van Havre <yorik@uncreated.net>
Date: Sun, 21 Jul 2019 19:01:39 -0300
Subject: [PATCH] Test: Allow WBs to know they are being run under test
 environment

---
 src/Mod/Test/Workbench.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/Mod/Test/Workbench.py b/src/Mod/Test/Workbench.py
index 92478175e1..2e7a40c2c5 100755
--- a/src/Mod/Test/Workbench.py
+++ b/src/Mod/Test/Workbench.py
@@ -35,13 +35,18 @@ class WorkbenchTestCase(unittest.TestCase):
         
     def testActivate(self):
         wbs=FreeCADGui.listWorkbenches()
+        # this gives workbenches a possibility to detect that we're under test environment
+        FreeCAD.TestEnvironment = True
         try:
             for i in wbs:
                 success = FreeCADGui.activateWorkbench(i)
                 FreeCAD.Console.PrintLog("Active: "+FreeCADGui.activeWorkbench().name()+ " Expected: "+i+"\n")
                 self.assertTrue(success, "Test on activating workbench {0} failed".format(i))
         except Exception as e:
+            del FreeCAD.TestEnvironment
             self.fail("Loading of workbench '{0}' failed: {1}".format(i, e))
+        else:
+            del FreeCAD.TestEnvironment
         
     def testHandler(self):
         import __main__