python接口自動化33-json解析神器jsonpath

前言

做接口測試的時候，大部分情況下返回的是json數據，我們需要對返回的json斷言。
當返回的數據量比較大，並且嵌套的層級很深的時候，很多小伙伴不會取值，往往在返回結果取值上浪費很多時間，於是就有了 jsonpath 解析庫，專門解決 json 路徑深，取值難的問題。

字典取值

對接口返回的接口，轉成 dict 類型，通過字典鍵值對取值

# 作者-上海悠悠 QQ交流群:717225969
# 返回結果，這里是dict
result = {
    "code": 0,
    "data": [
        {
            "age": 20,
            "create_time": "2019-09-15",
            "id": 1,
            "mail": "[e-mail]",
            "name": "yoyo",
            "sex": "M"
        },
        {
            "age": 21,
            "create_time": "2019-09-16",
            "id": 2,
            "mail": "[e-mail]",
            "name": "yoyo111",
            "sex": "M"
        }
    ],
    "msg": "success!"
}

# 字典取值斷言
assert result["code"] == 0
assert result["msg"] == "success!"
assert result["data"][0]["name"] == "yoyo"

當層級越來越深的時候，會發現取值變得困難，並且list里面的內容往往不知道是第幾個
比如我想判斷返回的結果里面有沒有 "name": "yoyo" 並且你不知道它是 data 列表中的第幾個，這種情況斷言就寫的很復雜了

正則取值

比如我想判斷返回的結果里面有沒有 "name": "yoyo" 並且你不知道它是 data 列表中的第幾個，這種情況斷言就寫的很復雜了。
可以用萬能的正則取值，正則只能是針對字符串取值，dict 對象先轉 json 字符串。

import re
import json
# 作者-上海悠悠 QQ交流群:717225969

# 返回結果，這里是dict
result = {
    "code": 0,
    "data": [
        {
            "age": 20,
            "create_time": "2019-09-15",
            "id": 1,
            "mail": "[e-mail]",
            "name": "yoyo",
            "sex": "M"
        },
        {
            "age": 21,
            "create_time": "2019-09-16",
            "id": 2,
            "mail": "[e-mail]",
            "name": "yoyo111",
            "sex": "M"
        }
    ],
    "msg": "success!"
}

# 斷言data中包含"name": "yoyo"
result_to_json = json.dumps(result)
print(result_to_json)       # 先轉json
# 方法1 正則取值
res = re.findall('"name": "(.+?)"', result_to_json)  # 正則從json中取值
print(res)
assert "yoyo" in res

# 方法2 直接判斷包含
assert '"name": "yoyo"' in result_to_json

jsonpath 解析

接下來講一個非常強大並且方便的 jsonpath 專門用於 json 解析，解決路徑深的老大難問題！
先安裝依賴包

pip install jsonpath

學習jsonpath 不得不提到xpath，這兩者之間的語法是差不多的

Xpath	JSONPath	描述
/	$	跟節點
.	@	現行節點
/	. or []	取子節點
..	n/a	取父節點 JsonPath不支持
//	..	相對節點 就是不管位置，選擇所有符合條件的條件
*	*	匹配所有元素節點
[]	[]	迭代器標示(可以在里面做簡單的迭代操作，如數組下標，根據內容選值等)
&#124	[,]	支持迭代器中做多選
[]	?()	支持過濾操作
n/a	()	支持表達式計算
()	n/a	分組，JsonPath不支持

使用示例

$ 是查找的根節點，傳參數是python的dict 類型，當查找到的時候返回一個list結果，查找失敗的時候返回 False.

import jsonpath
# 作者-上海悠悠 QQ交流群:717225969

result = {
    "code": 0,
    "data": [
        {
            "age": 20,
            "create_time": "2019-09-15",
            "id": 1,
            "mail": "[e-mail]",
            "name": "yoyo",
            "sex": "M"
        },
        {
            "age": 21,
            "create_time": "2019-09-16",
            "id": 2,
            "mail": "[e-mail]",
            "name": "yoyo111",
            "sex": "M"
        }
    ],
    "msg": "success!"
}


msg = jsonpath.jsonpath(result, '$.msg')
print(msg)   # 輸出結果 ['success!']
names = jsonpath.jsonpath(result, '$..name')
print(names)   # 輸出結果 ['yoyo', 'yoyo111']
no = jsonpath.jsonpath(result, '$..yoyo')
print(no)   # 找不到是結果是 False

這樣就可以不用管層級結構也能取值了。