Extract the environment set up and tear down for tests

Using trial there is no setUpClass neither tearDownClass, the setting up
of the environment needs to be in an external class to be able to call
it from inhereted classes.

2 files changed, 10 insertions, 1 deletions

diff --git a/changes/feature-test_env b/changes/feature-test_env
new file mode 100644
index 0000000..77614d7
--- /dev/null
+++ b/changes/feature-test_env
@@ -0,0 +1 @@
+- Extract the environment set up and tear down for tests
diff --git a/src/leap/common/testing/basetest.py b/src/leap/common/testing/basetest.py
index 54826d5..3c6fc29 100644
--- a/src/leap/common/testing/basetest.py
+++ b/src/leap/common/testing/basetest.py
@@ -40,6 +40,14 @@ class BaseLeapTest(unittest.TestCase):
 
     @classmethod
     def setUpClass(cls):
+        cls.setUpEnv()
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.tearDownEnv()
+
+    @classmethod
+    def setUpEnv(cls):
         """
         Sets up common facilities for testing this TestCase:
         - custom PATH and HOME environmental variables
@@ -57,7 +65,7 @@ class BaseLeapTest(unittest.TestCase):
         os.environ["HOME"] = cls.tempdir
 
     @classmethod
-    def tearDownClass(cls):
+    def tearDownEnv(cls):
         """
         Cleanup common facilities used for testing this TestCase:
         - restores the default PATH and HOME variables