Ticket #14378: custom-test-client.diff

django/test/testcases.py

@@ -210 +210 @@
             transaction.rollback_unless_managed(using=conn)
 
 class TransactionTestCase(unittest.TestCase):
+    # The class we'll use for the test client self.client.
+    # Can be overridden in derived classes.
+    client_class = Client
+
     def _pre_setup(self):
         """Performs any pre-test setup. This includes:
 
@@ -251 +255 @@
         set up. This means that user-defined Test Cases aren't required to
         include a call to super().setUp().
         """
-        self.client = Client()
+        self.client = self.client_class()
         try:
             self._pre_setup()
         except (KeyboardInterrupt, SystemExit):

tests/modeltests/test_client/models.py

@@ -457 +457 @@
         # The CSRF-enabled client rejects it
         response = csrf_client.post('/test_client/post_view/', {})
         self.assertEqual(response.status_code, 403)
+
+
+class CustomTestClient(Client):
+    i_am_customized = "Yes"
+
+class CustomTestClientTest(TestCase):
+    client_class = CustomTestClient
+
+    def test_custom_test_client(self):
+        """A test case can specify a custom class for self.client."""
+        self.assertEqual(hasattr(self.client, "i_am_customized"), True)
+

docs/topics/testing.txt

@@ -1075 +1075 @@
             response = self.client.get('/customer/index/')
             self.failUnlessEqual(response.status_code, 200)
 
+In your sub-classes of ``TestCase`` you can specify a custom sub-class of 
+``Client`` to instantiate::
+
+    from django.test import TestCase
+    from django.test.client import Client
+
+    class MyTestClient(Client):
+        # .. specialized methods for my environment ..
+
+    class MyTest(TestCase):
+        client_class = MyTestClient
+
+        def test_my_stuff(self):
+            # .. Here self.client is an instance of MyTestClient ..
+
 .. _topics-testing-fixtures:
 
 Fixture loading