Fix urljoin import for Python 3

Wilfred · Jun 30, 2018 · f437fdd · f437fdd
1 parent f2f8624
commit f437fdd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/django_test_mixins.py b/django_test_mixins.py
@@ -1,7 +1,10 @@
 from django.test import TestCase
 from django.core.cache import cache
 
-import urlparse
+try:
+    from urlparse import urljoin  # python2
+except ImportError:
+    from urllib.parse import urljoin  # python3
 
 
 class HttpCodeTestCase(TestCase):
@@ -32,7 +35,7 @@ def assertHttpRedirect(self, response, location=None):
             if location.startswith("http://testserver/"):
                 absolute_location = location
             else:
-                absolute_location = urlparse.urljoin("http://testserver/",  location)
+                absolute_location = urljoin("http://testserver/", location)
 
             self.assertEqual(response['Location'], absolute_location)