python自动化unittest中yaml用法实例说明

Posted 2021-03-13 snoweveryday

YAML 入门教程:https://www.runoob.com/w3cnote/yaml-intro.html

 

在编写unittest自动化用例时，一个请求需要编写多条用例，而涉及的参数基本相同，这时候就会用到配置文件，可以把参数配置项统一管理，避免重复代码，也方便后期维护

此处用到的是yaml，首先需要安装yaml库，pip install yaml

安装成功后，脚本导入语句，import yaml，具体语法可参照如上入门教程

举例说明：

用例-查询教师列表信息（请求地址相同，每次请求参数不同），将请求地址配置到yaml文件

1.创建yaml文件，录入请求请求url

 2.读取url信息

class teaTimeData(unittest.TestCase):
    def setUp(self):
        print("-------教师用例开始执行-------")
        self.file = open(get_path() + "/data/count_query.yaml","r",encoding="utf-8")
        self.data = yaml.load(self.file,Loader=yaml.FullLoader)
        self.file.close()
        self.base_url = self.data["query"].get("url")

在此处如果直接使用yaml.load()，会有警告出现，可以参考PyYAML yaml.load(input) Deprecation

3.编辑用例

    def test_01(self):
        headers ={
            "frontcode": "frontcode",
            "token": user_token
        }
        params={
            "pageSize": "5"
        }
        r = requests.get(url = self.query_url,headers = headers,params = params)
        self.assertEqual(r.status_code,200)

继续编辑其它用例，只需要配置不同参数即可

最后执行用例