Merge pull request #2 from RobinRamael/master

Added optional location parameter to assertHttpRedirect.

django_test_mixins.py

@@ -1,6 +1,8 @@
 from django.test import TestCase
 from django.core.cache import cache
 
+import urlparse
+
 
 class HttpCodeTestCase(TestCase):
     # TODO: this should be a private method.
@@ -16,7 +18,7 @@ def assertHttpOK(self, response):
     def assertHttpCreated(self, response):
         self.assertHttpCode(response, 201, "Created")
 
-    def assertHttpRedirect(self, response):
+    def assertHttpRedirect(self, response, location=None):
         """Assert that we had any redirect status code.
 
         """
@@ -26,9 +28,18 @@ def assertHttpRedirect(self, response):
             response.status_code
         )
 
+        if location:
+            if location.startswith("http://testserver/"):
+                absolute_location = location
+            else:
+                absolute_location = urlparse.urljoin("http://testserver/",  location)
+
+            self.assertEqual(response['Location'], absolute_location)
+
+
     def assertHttpBadRequest(self, response):
         self.assertHttpCode(response, 400, "Bad Request")
-        
+
     def assertHttpUnauthorized(self, response):
         self.assertHttpCode(response, 401, "Unauthorized")