Centralize settings modifications for unittests in a single class

1 files changed, 50 insertions, 0 deletions

diff --git a/hyperkitty/tests/utils.py b/hyperkitty/tests/utils.py
new file mode 100644
index 0000000..5124f39
--- /dev/null
+++ b/hyperkitty/tests/utils.py
@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+# Copyright (C) 1998-2012 by the Free Software Foundation, Inc.
+#
+# This file is part of HyperKitty.
+#
+# HyperKitty is free software: you can redistribute it and/or modify it under
+# the terms of the GNU General Public License as published by the Free
+# Software Foundation, either version 3 of the License, or (at your option)
+# any later version.
+#
+# HyperKitty is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+# more details.
+#
+# You should have received a copy of the GNU General Public License along with
+# HyperKitty.  If not, see <http://www.gnu.org/licenses/>.
+#
+# Author: Aurelien Bompard <abompard@fedoraproject.org>
+#
+
+
+from django.test import TestCase as DjangoTestCase
+from django.conf import settings
+
+
+OVERRIDE_SETTINGS = {
+    "TEMPLATE_DEBUG": True,
+    "ASSETS_DEBUG": True,
+    "USE_SSL": False,
+    "KITTYSTORE_URL": 'sqlite:',
+    "KITTYSTORE_SEARCH_INDEX": None,
+    "KITTYSTORE_DEBUG": False,
+    "USE_MOCKUPS": False,
+}
+
+
+class TestCase(DjangoTestCase):
+
+    def _pre_setup(self):
+        super(TestCase, self)._pre_setup()
+        self._old_settings = {}
+        for key, value in OVERRIDE_SETTINGS.iteritems():
+            self._old_settings[key] = getattr(settings, key)
+            setattr(settings, key, value)
+
+    def _post_teardown(self):
+        super(TestCase, self)._post_teardown()
+        for key, value in self._old_settings.iteritems():
+            setattr(settings, key, value)