Commits

Mikhail Korobov committed 31efcca Merge

Merge pull request #10 from yrik/master

WebTestMixin is extracted. WebTest and TransactionWebTest are provided.

Comments (0)

Files changed (2)

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

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
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.