Add test for dropping fields in from_response request body

ffef828a · Harrison Gregg · 45a32302 · ffef828a
隐藏空白更改
内联并排

Showing with 11 addition and 0 deletion

tests/test_http_request.py tests/test_http_request.py +11 -0

未找到文件。
--- a/tests/test_http_request.py
+++ b/tests/test_http_request.py
@@ -425,6 +425,17 @@ class FormRequestTest(RequestTest):
        self.assertEqual(fs[b'one'], [b'1'])
        self.assertEqual(fs[b'two'], [b'2'])

+    def test_from_response_drop_params(self):
+        response = _buildresponse(
+            """<form action="get.php" method="POST">
+            <input type="hidden" name="one" value="1">
+            <input type="hidden" name="two" value="3">
+            </form>""")
+        req = self.request_class.from_response(response, formdata={'two': None})
+        fs = _qs(req)
+        self.assertEqual(fs[b'one'], [b'1'])
+        self.assertNotIn(b'two', fs)
+
    def test_from_response_override_method(self):
        response = _buildresponse(
                '''<html><body>