39pytest使用yaml文件编写测试用例

前言

• 本篇来学习在pytest中使用yaml编写测试用例

项目结构

conftest.py

• 只需在 conftest.py 即可实现使用yaml编写测试用例

# -*- coding: utf-8 -*-
# &＃64;Time : 2021/10/13
# &＃64;Author : 大海import jsonpath
import pytest
import requestsdef pytest_collect_file(parent, file_path):if file_path.suffix &＃61;&＃61; ".yaml" and file_path.name.startswith("test"):return YamlFile.from_parent(parent, path&＃61;file_path)class YamlFile(pytest.File):def collect(self):import yamlyml_raw &＃61; self.fspath.open(encoding&＃61;&＃39;utf-8&＃39;).read()print(&＃39;yml_raw&＃39;, yml_raw)yml_var &＃61; Template(yml_raw).safe_substitute(os.environ)raw &＃61; yaml.safe_load(yml_var)print(&＃39;raw&＃39;, raw)for yaml_case in raw:name &＃61; yaml_case["test"]["name"]values &＃61; yaml_case["test"]yield YamlItem.from_parent(self, name&＃61;name, spec&＃61;values)class YamlItem(pytest.Item):def __init__(self, name, parent, spec):super().__init__(name, parent)self.name &＃61; nameself.values &＃61; specself.request &＃61; self.values.get("request")self.validate &＃61; self.values.get("validate")self.s &＃61; requests.session()def values_render_variable(self, values):# 替换测试用例中的关联值yaml_test &＃61; Template(json.dumps(values)).safe_substitute(os.environ)values &＃61; yaml.safe_load(yaml_test)print(&＃39;values&＃39;, values)return valuesdef runtest(self):values &＃61; self.values_render_variable(self.values)print(&＃39;values:&＃39;, values)request_data &＃61; values["request"]response &＃61; self.s.request(**request_data)print("响应数据&＃xff1a;", response.text)# 判断是否有extract提取参数,实现参数关联if values.get("extract"):for key, value in values.get("extract").items():os.environ[key] &＃61; jsonpath.jsonpath(response.json(), value)[0]print(&＃39;key&＃39;, key)print(&＃39;value&＃39;, jsonpath.jsonpath(response.json(), value)[0])# 断言print(&＃39;validate&＃xff1a;&＃39;, self.validate)self.assert_response(response, self.validate)def assert_response(self, response, validate):"""自定义断言"""for i in validate:if "eq" in i.keys():yaml_result &＃61; i.get("eq")[0]actual_result &＃61; jsonpath.jsonpath(response.json(), yaml_result)expect_result &＃61; i.get("eq")[1]print("实际结果&＃xff1a;%s" % actual_result[0])print("期望结果&＃xff1a;%s" % expect_result)assert actual_result[0] &＃61;&＃61; expect_result


yaml文件

• test_method.yaml

说明&＃xff1a;
此yaml支持参数化

• extract &＃xff1a;提取关键字
  • name: 后面引用变量的key值
  • $.args.name&＃xff1a;jsonpath 提取变量表达式

引用变量

• $name : $key

- test:name: get caserequest:url: https://postman-echo.com/getmethod: GETheaders:Content-Type: application/jsonUser-Agent: python-requests/2.18.4params:name: DHcity: Beijingextract:name: $.args.namevalidate:- eq: [$.args.name, DH]- eq: [$.args.city, Beijing]
- test:name: post caserequest:url: https://postman-echo.com/postmethod: POSTheaders:Content-Type: application/jsonUser-Agent: python-requests/2.18.4json:name: $namecity: Beijingvalidate:- eq: [$.json.name, DH]- eq: [$.json.city, Beijing]


执行并查看结果

pytest -s -v