Added unit tests for changes in Parsers and Request

codders · auvipy · commit 6dcb9dd7f5f2 · 2023-11-11T00:56:50.000+06:00
diff --git a/flask_api/tests/test_parsers.py b/flask_api/tests/test_parsers.py
@@ -31,6 +31,12 @@ def test_valid_json(self):
         data = parser.parse(stream, "application/json")
         self.assertEqual(data, {"key": 1, "other": "two"})
 
+    def test_parse_urlencoded(self):
+        parser = parsers.URLEncodedParser()
+        stream = io.BytesIO(b'next=http://www.example.com&test1=val1&test%5c=val%2f')
+        data = parser.parse(stream, "application/x-www-form-urlencoded")
+        self.assertEqual(data, {"next": "http://www.example.com", "test1": "val1", "test\\": "val/"})
+
     def test_invalid_json(self):
         parser = parsers.JSONParser()
         stream = io.BytesIO(b'{key: 1, "other": "two"}')
diff --git a/flask_api/tests/test_request.py b/flask_api/tests/test_request.py
@@ -19,6 +19,15 @@ def test_json_request(self):
         with app.test_request_context(**kwargs):
             self.assertEqual(request.data, {"key": 1, "other": "two"})
 
+    def test_urlencoded_post_request(self):
+        kwargs = {
+            "method": "POST",
+            "input_stream": io.BytesIO(b'next=http://www.example.com&test1=val1&test%5c=val%2f'),
+            "content_type": "application/x-www-form-urlencoded",
+        }
+        with app.test_request_context(**kwargs):
+            self.assertEqual(request.data, {"next": "http://www.example.com", "test1": "val1", "test\\": "val/"})
+
     def test_invalid_content_type_request(self):
         kwargs = {
             "method": "PUT",
