refactor: relocate python test files

2026-05-20 23:59:49 +08:00 · 2022-03-23 17:28:54 +08:00
parent 8ac27790b2
commit 64ff34a341
38 changed files with 127 additions and 43 deletions
--- a/httprunner/response_test.py
+++ b/httprunner/response_test.py
@@ -0,0 +1,70 @@
+import unittest
+
+import requests
+
+from httprunner.response import ResponseObject
+
+
+class TestResponse(unittest.TestCase):
+    def setUp(self) -> None:
+        resp = requests.post(
+            "https://httpbin.org/anything",
+            json={
+                "locations": [
+                    {"name": "Seattle", "state": "WA"},
+                    {"name": "New York", "state": "NY"},
+                    {"name": "Bellevue", "state": "WA"},
+                    {"name": "Olympia", "state": "WA"},
+                ]
+            },
+        )
+        self.resp_obj = ResponseObject(resp)
+
+    def test_extract(self):
+        variables_mapping = {
+            'body': 'body'
+        }
+        functions_mapping = {
+            'get_name': lambda: 'name',
+        }
+        extract_mapping = self.resp_obj.extract(
+            {
+                "var_1": "body.json.locations[0]",
+                "var_2": "body.json.locations[3].name",
+                "var_3": "$body.json.locations[3].name",
+                "var_4": "$body.json.locations[3].${get_name()}",
+            },
+            variables_mapping=variables_mapping,
+            functions_mapping=functions_mapping,
+        )
+        self.assertEqual(extract_mapping["var_1"], {"name": "Seattle", "state": "WA"})
+        self.assertEqual(extract_mapping["var_2"], "Olympia")
+        self.assertEqual(extract_mapping["var_3"], "Olympia")
+        self.assertEqual(extract_mapping["var_4"], "Olympia")
+
+    def test_validate(self):
+        self.resp_obj.validate(
+            [
+                {"eq": ["body.json.locations[0].name", "Seattle"]},
+                {"eq": ["body.json.locations[0]", {"name": "Seattle", "state": "WA"}]},
+            ],
+        )
+
+    def test_validate_variables(self):
+        variables_mapping = {"index": 1, "var_empty": ""}
+        self.resp_obj.validate(
+            [
+                {"eq": ["body.json.locations[$index].name", "New York"]},
+                {"eq": ["$var_empty", ""]},
+            ],
+            variables_mapping=variables_mapping,
+        )
+
+    def test_validate_functions(self):
+        variables_mapping = {"index": 1}
+        functions_mapping = {"get_num": lambda x: x}
+        self.resp_obj.validate(
+            [{"eq": ["${get_num(0)}", 0]}, {"eq": ["${get_num($index)}", 1]},],
+            variables_mapping=variables_mapping,
+            functions_mapping=functions_mapping,
+        )