Commits

Mikhail Korobov  committed 31efcca Merge

Merge pull request #10 from yrik/master

WebTestMixin is extracted. WebTest and TransactionWebTest are provided.

  • Participants
  • Parent commits d6df1a6, 6265c2a

Comments (0)

Files changed (2)

 * Bruno Renié
 * Gautier Hayoun
 * Carl Meyer
+* Iurii Kriachko

File django_webtest/__init__.py

 from django.conf import settings
 from django.test.signals import template_rendered
 from django.core.handlers.wsgi import WSGIHandler
-from django.test import TestCase
+from django.test import TestCase, TransactionTestCase
 from django.test.client import store_rendered_templates
 from django.utils.functional import curry
 from django.utils.importlib import import_module
         return {}
 
 
-class WebTest(TestCase):
+class WebTestMixin(object):
 
     extra_environ = {}
     csrf_checks = True
     def __call__(self, result=None):
         self._patch_settings()
         self.renew_app()
-        res = super(WebTest, self).__call__(result)
+        res = super(WebTestMixin, self).__call__(result)
         self._unpatch_settings()
         return res
 
 
+class WebTest(WebTestMixin, TestCase):
+    pass
+
+
+class TransactionWebTest(WebTestMixin, TestCase):
+    pass
+
+
 def _get_username(user):
     """
     Return user's username. ``user`` can be standard Django User