From 2b7f0b03d41516748ecbb823883ce8e4d79089c0 Mon Sep 17 00:00:00 2001
From: Antoine Nguyen <tonio@ngyn.org>
Date: Sun, 23 Oct 2016 15:04:04 +0200
Subject: [PATCH] Make sure test extensions are not used elsewhere.

---
 modoboa/core/tests/test_extensions.py | 34 ++++++++++-----------------
 1 file changed, 13 insertions(+), 21 deletions(-)

diff --git a/modoboa/core/tests/test_extensions.py b/modoboa/core/tests/test_extensions.py
index 94fec8479..b99c1502c 100644
--- a/modoboa/core/tests/test_extensions.py
+++ b/modoboa/core/tests/test_extensions.py
@@ -10,6 +10,8 @@ from django.test import TestCase, override_settings
 
 from .. import extensions
 
+sys.path.append(os.path.dirname(__file__))
+
 CUSTOM_APPS = (
     "modoboa",
     "modoboa.core",
@@ -21,32 +23,22 @@ CUSTOM_APPS = (
     "stupid_extension_2",
 )
 
-sys.path.append(os.path.dirname(__file__))
 
-
-class StupidExtension2(extensions.ModoExtension):
-    """Stupid extension to use with tests."""
-
-    name = "stupid_extension_2"
-    label = "Stupid extension"
-    version = "1.0.0"
-    description = "A stupid extension"
-
-    def load(self):
-        pass
-
-    def load_initial_data(self):
-        pass
-
-
-@override_settings(MODOBOA_APPS=CUSTOM_APPS)
 class ExtensionTestCase(TestCase):
     """Extensions related tests."""
 
-    def setUp(self):
+    @classmethod
+    def setUpClass(cls):
         """Initiate test context."""
-        self.pool = extensions.exts_pool
-        self.pool.load_all()
+        cls.pool = extensions.exts_pool
+        with override_settings(MODOBOA_APPS=CUSTOM_APPS):
+            cls.pool.load_all()
+
+    @classmethod
+    def tearDownClass(cls):
+        """Reset context."""
+        del cls.pool.extensions["stupid_extension_1"]
+        del cls.pool.extensions["stupid_extension_2"]
 
     def test_register(self):
         """Test if registering works."""