Merge pull request #72 from davideme/master

Added JWT TestCase Class and Client

Author: jpadilla

rest_framework_jwt/test.py

@@ -0,0 +1,24 @@
+from rest_framework.test import APITestCase, APIClient
+from rest_framework import status
+from rest_framework_jwt.settings import api_settings
+
+
+class APIJWTClient(APIClient):
+    def login(self, username, password):
+        """
+        Returns True if login is possible; False if the provided credentials
+        are incorrect, or the user is inactive.
+        """
+        response = self.post('/api-token-auth/', {"username": username, "password": password},
+                             format='json')
+        if response.status_code == status.HTTP_200_OK:
+            self.credentials(
+                HTTP_AUTHORIZATION="{0} {1}".format(api_settings.JWT_AUTH_HEADER_PREFIX, response.data['token']))
+
+            return True
+        else:
+            return False
+
+
+class APIJWTTestCase(APITestCase):
+    client_class = APIJWTClient