From ae63c49777f4d5b766b85a4b28f6328bd6f9516a Mon Sep 17 00:00:00 2001
From: Tom Christie <tom@tomchristie.com>
Date: Mon, 15 Jul 2013 11:38:38 +0100
Subject: [PATCH] Added test case classes

---
 docs/api-guide/testing.md | 30 ++++++++++++++++++++++++++++++
 rest_framework/test.py    | 17 +++++++++++++++++
 2 files changed, 47 insertions(+)

diff --git a/docs/api-guide/testing.md b/docs/api-guide/testing.md
index aba9283e0..40b077633 100644
--- a/docs/api-guide/testing.md
+++ b/docs/api-guide/testing.md
@@ -167,6 +167,36 @@ As usual CSRF validation will only apply to any session authenticated views.  Th
 
 ---
 
+# Test cases
+
+REST framework includes the following test case classes, that mirror the existing Django test case classes, but use `APIClient` instead of Django's default `Client`.
+
+* `APISimpleTestCase`
+* `APITransactionTestCase`
+* `APITestCase`
+* `APILiveServerTestCase`
+
+## Example
+
+You can use any of REST framework's test case classes as you would for the regular Django test case classes.  The `self.client` attribute will be an `APIClient` instance.
+
+    from django.core.urlresolvers import reverse
+    from rest_framework import status
+    from rest_framework.test import APITestCase 
+
+    class AccountTests(APITestCase):
+        def test_create_account(self):
+            """
+            Ensure we can create a new account object.
+            """
+            url = reverse('account-list')
+            data = {'name': 'DabApps'}
+            response = self.client.post(url, data, format='json')
+            self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+            self.assertEqual(response.data, data)
+
+---
+
 # Testing responses
 
 ## Checking the response data
diff --git a/rest_framework/test.py b/rest_framework/test.py
index 29d017ee4..ed436976a 100644
--- a/rest_framework/test.py
+++ b/rest_framework/test.py
@@ -6,6 +6,7 @@ from __future__ import unicode_literals
 from django.conf import settings
 from django.test.client import Client as DjangoClient
 from django.test.client import ClientHandler
+from django.test import testcases
 from rest_framework.settings import api_settings
 from rest_framework.compat import RequestFactory as DjangoRequestFactory
 from rest_framework.compat import force_bytes_or_smart_bytes, six
@@ -137,3 +138,19 @@ class APIClient(APIRequestFactory, DjangoClient):
         # Ensure that any credentials set get added to every request.
         kwargs.update(self._credentials)
         return super(APIClient, self).request(**kwargs)
+
+
+class APISimpleTestCase(testcases.SimpleTestCase):
+    client_class = APIClient
+
+
+class APITransactionTestCase(testcases.TransactionTestCase):
+    client_class = APIClient
+
+
+class APITestCase(testcases.TestCase):
+    client_class = APIClient
+
+
+class APILiveServerTestCase(testcases.LiveServerTestCase):
+    client_class = APIClient