Unquote response location in assertRedirect

flask_testing/utils.py

@@ -22,6 +22,7 @@
     from urllib.request import urlopen
 import multiprocessing
 
+from urllib import unquote
 from werkzeug import cached_property
 
 # Use Flask's preferred JSON module so that our runtime behavior matches.
@@ -215,7 +216,7 @@ def assertRedirects(self, response, location):
         :param location: relative URL (i.e. without **http://localhost**)
         """
         self.assertTrue(response.status_code in (301, 302))
-        self.assertEqual(response.location, "http://localhost" + location)
+        self.assertEqual(unquote(response.location), "http://localhost" + location)
 
     assert_redirects = assertRedirects
 

tests/flask_app/__init__.py

@@ -22,6 +22,10 @@ def bad_url():
     def redirect_to_index():
         return redirect(url_for("index"))
 
+    @app.route("/redirect-next/")
+    def redirect_from_somewhere():
+        return redirect(url_for("index", next='/somewhere/'))
+
     @app.route("/ajax/")
     def ajax():
         return jsonify(name="test")

tests/test_utils.py

@@ -89,6 +89,10 @@ def test_assert_redirects(self):
         response = self.client.get("/redirect/")
         self.assertRedirects(response, "/")
 
+    def test_assert_redirects_with_query(self):
+        response = self.client.get("/redirect-next/")
+        self.assertRedirects(response, "/?next=/somewhere/")
+
     def test_assert_template_used(self):
         try:
             self.client.get("/template/")