在自动化测试中,参数化非常常见。当你在测试某一个接口时,通常会给测试函数传递很多参数,达到遍历的目的。比如测试登录接口,我们需要模拟各种场景的账号密码。又如我们上一篇文章中介绍的获取天气接口,你需要验证很多城市的天气信息是否返回正确。我们通常的做法是把需要验证的城市数据参数写在测试方法内部进行遍历。但仍然是在一个测试方法中执行,如果某个请求导致断言失败,就此测试也会被终止。当然传统做法我们可以捕获异常继续进行后续测试,但是这样做太麻烦了。那pytest有没有提供更好的方式来解决这个问题呢?
pytest为我们提供了pytest.mark.parametrize(argnames, argvalues)来进行参数化,解决这个问题。每一组参数都是一次独立的测试执行
基本语法
- 使用方法 在需要参数化的函数上方标注 @pytest.mark.parametrize(argnames, argvalues)
- 参数详解
参数 | 备注 |
---|---|
argnames | 参数名称 |
argvalues | 参数值,类型必须为list |
验证获取天气信息接口
下面是pytest.mark.parametrize(argnames, argvalues)的使用示例,可供参考
传递单个参数
传递单个参数只需要将值定义为一个list即可。
import requests, pytest
url = "http://t.weather.sojson.com/api/weather/city/"
class TestDemo:
@pytest.mark.parametrize("city_code", ["101010100", "101020100", "101280101",
"101280601", "101210101"])
def test_get_weather(self, city_code):
r = requests.get(url + city_code)
result_data = r.json()
assert result_data["status"] == 200
assert result_data["cityInfo"]["citykey"] == city_code
上面的测试方法依次验证了北京、上海、广州、深州、杭州五个城市的天气信息返回是否正确。参数名为:city_code,其值为一个列表list类型结构,包含了5个城市的code。通过运行可以看到执行了5次。
使用命令:pytest -v demo.py
执行结果:
传递多个参数
多个参数传递只需要把list中的每个元素定义为元组类型即可。假设我们每个城市的接口地址也不一样,我们可以这么修改来实现传递多个参数。(是假设奥,在此只是为了演示传递多个参数,相信没人会针对一个操作开发不同的接口)
import requests, pytest
class TestDemo:
@pytest.mark.parametrize("url,city_code",
[("http://t.weather.sojson.com/api/weather/city/", "101010100"),
("http://t.weather.sojson.com/api/weather/city/", "101020100"),
("http://t.weather.sojson.com/api/weather/city/", "101280101"),
("http://t.weather.sojson.com/api/weather/city/", "101280601"),
("http://t.weather.sojson.com/api/weather/city/", "101210101")])
def test_get_weather(self, url, city_code):
r = requests.get(url + city_code)
result_data = r.json()
assert result_data["status"] == 200
assert result_data["cityInfo"]["citykey"] == city_code
使用命令:pytest -v demo.py
执行结果:
总结:通过上面的演示我们可以到在pytest中是通过标记函数来实现参数化的,很简洁。这么实现就不会出现因为一个断言失败,而不再进行后续的验证测试了。而且在执行结果中我们也可以清晰的看到每次执行传递的参数值。