1 files changed, 26 insertions, 0 deletions
diff --git a/tests/base.py b/tests/base.py
new file mode 100644
index 0000000..e6f6dac
--- /dev/null
+++ b/tests/base.py
@@ -0,0 +1,26 @@
+from __future__ import unicode_literals
+
+import threading
+
+from django.http import HttpResponse
+from django.test import TestCase, RequestFactory
+
+from debug_toolbar.middleware import DebugToolbarMiddleware
+from debug_toolbar.toolbar.loader import DebugToolbar
+
+rf = RequestFactory()
+
+
+class BaseTestCase(TestCase):
+
+    def setUp(self):
+        request = rf.get('/')
+        response = HttpResponse()
+        toolbar = DebugToolbar(request)
+
+        DebugToolbarMiddleware.debug_toolbars[threading.current_thread().ident] = toolbar
+
+        self.request = request
+        self.response = response
+        self.toolbar = toolbar
+        self.toolbar.stats = {}