UI自动化测试-pytest框架

🔥 交流讨论：欢迎加入我们一起学习！

🔥 资源分享：耗时200+小时精选的「软件测试」资料包

🔥 教程推荐：火遍全网的《软件测试》教程

📢欢迎点赞 👍 收藏 ⭐留言 📝 如有错误敬请指正！

在进行UI自动化测试的时候，我们需要工具来对测试用例进行收集，执行，标记，参数化。pytest就是这样一个工具。

pytest实际是python的一个单元测试框架，其他还有如unittest等，它可以实现按照规则搜索测试用例，对测试用例进行标记，如只执行L1

级别的测试用例，测试执行用例失败后重新执行，测试用例的参数化等。

pytest不仅适用于UI自动化测试，也适用于接口自动化，对python语言进行单元测试等。

安装pytest

可以使用pip或者pycharm方式进行安装

打开pycharm，左上角File-settings-project:项目名-python interpreter，点击+，搜索pytest，然后install package

第一个pytest脚本

import pytestclass TestDemo:def test_01(self):assert 1==1  # 断言成功def test_02(self):assert 1==2  # 断言失败
if __name__ == '__main__':pytest.main(['-s','test.py'])

在文件输入如上代码。

在这个脚本中，我们首先导入了pytest模块，然后定义了一个class，并定义了两个方法，在方法里用assert进行断言。

assert是python自带的关键字。assert后面接一个表达式，只要表达式的最终结果为True，那么断言通过，用例执行成功，否则用例执行失败。可以看出，第一个方法里结果为true，第二个方法里结果为false。

在main主入口里，调用pytest的main方法，并传入模块名（即文件名）。

点击运行按钮，结果如下：

由结果可以看出，一共收集了两个用例，其中一个运行成功（用.表示），一个运行失败（F表示），失败的原因是1！=2

用例收集规则

pytest按照一定的规则进行用例的收集。

1. 模块名必须以test_开头或_test结尾

这里适用于未指定模块时的收集，如果运行单个文件，在main里指定要执行的文件，则无论这个文件名是什么，都会被执行。

如果在main里不指定模块，如只写pytest.main()，则pytest会默认从当前路径及其所有子目录中搜索py源文件，所有名字以test_开头或者以_test结尾的python源文件（.py文件）被认为是测试模块源文件，不符合这个命名规则的文件会被忽略。

2. 测试类必须以Test开头且不能有init方法

一个文件也可以只写测试方法，不写测试类。

3. 测试方法必须以test开头

在上面的示例脚本中，我们定义了两个test_开头的方法，所以收集了两个用例。非test开头的方法，会被pytest忽略。

运行方式

main函数运行

我们可以在main入口中调用pytest的main方法来运行文件，如上例。

不指定模块名

pytest.main(['-s'])

-s参数表示在控制台输出信息，当不指定模块名时，可以看到，收集了7条用例

指定模块名

 pytest.main(['-s','test.py'])

当指定了模块后，运行文件，只收集当前文件的两条用例。

指定class

 pytest.main(["-s", "test.py::TestDemo"])

我们也可以在模块名上加上指定的class名

命令行窗口运行

不指定模块名

打开terminal，进入目标目录，直接执行“pytest”即可自动寻找当前目录下的测试用例

指定模块名

打开terminal，输入pytest+模块名，指定模块执行。

pytest test.py

指定函数名

打开terminal，输入pytest+模块名+函数名，指定模块执行。

pytest test.py::TestDemo

test runner运行

在File-Settings如图路径，设置default test runner为pytest，点击OK。

再回到测试文件，可以看到类，方法前面都显示运行按钮，点击对应按钮，可以运行类，单独运行某个test。

也可以在不写main函数的情况下，右键以pytest运行文件。注意在配置了runner后，执行不会进入main主入口。

标记机制

pytest提供了标记机制，借助“mark”关键字，我们可以对测试函数（类、方法）进行标记。

我们可以利用标记，对测试用例进行分级，例如某些主流程的用例可以标记为L1，次要流程的用例标记为L2等。这样有一个好处，我们可以在不同的情况执行不同的测试用例，例如，在做冒烟测试的时候，只需要执行L1级别的用例就行了。

●一个测试函数（类、方法）可以有多个标记。

●一个标记也可以应用于多个函数（类、方法）。

●执行参数使用：pytest -m mark名。

●执行多个标记：pytest -m “L1 or L2”。

import pytestclass TestDemo:@pytest.mark.L1@pytest.mark.L2def test_01(self):assert 1==1  # 断言成功@pytest.mark.L2def test_02(self):assert 1==2  # 断言失败
if __name__ == '__main__':pytest.main(['-m L2','test.py'])

如上，给测试用例添加标记，并在main中以-m参数指定选择的标记，也可以使用or和and。

  pytest.main(['-m L2 or L1','test.py']) # 执行标记为L1或者L2的

  pytest.main(['-m L2 and L1','test.py']) #执行标记同时为L1和L2的

测试固件

在进行UI测试时，可能每个用例的操作前提都是需要登录系统，那么我们也不能在每次写用例的时候，都把登录写一遍，或者调用一遍，这样会造成代码冗余。有没有可能，在整个测试类或者所有测试用例执行之前，调用一次登录呢？

这样的工具，就是测试固件，测试固件有不同级别。

函数中的测试固件

●setup_module、teardown_module，在整个文件的开始和最后执行一次。

●setup_function和teardown_function，在每个函数开始前后执行。


import pytest'''
在函数中使用
1.setup_module、teardown_module，在整个文件的开始和最后执行一次
2.setup_function和teardown_function，在每个函数开始前后执行
'''
def setup_module():print('setup_module')def teardown_module():print('teardown_module')def setup_function():print('setup_function')def teardown_function():print('teardown_function')def test_a():print('aaaa')assert 1 == 1def test_b():print('bbbb')assert 1 == 2if __name__ == '__main__':pytest.main(["-s", "test.py"])

class中的测试固件

●setup_class、teardown_class，在整个class的开始和最后执行一次。

●setup_method和teardown_method，在每个方法开始前后执行。

class TestDemo():def setup_class(self):print('setup_class')def teardown_class(self):print('teardown_class')def setup_method(self):print('setup_method')def teardown_method(self):print('teardown_method')def test_a(self):print('aaaa')assert 1 == 1def test_b(self):print('bbbb')assert 1 == 2if __name__ == '__main__':pytest.main(["-s", "test.py"])

setup和teardown

setup和teardown既可以应用在函数中，也可以应用在class中，作用对象是函数或方法，在每个测试用例执行前后执行一次。

我们通常使用setup来进行一些数据准备工作，使用teardown来进行数据清理工作。

import pytestclass TestDemo:def setup(self):print('setup')def teardown(self):print('teardown')@pytest.mark.L1@pytest.mark.L2def test_01(self):assert 1==1  # 断言成功@pytest.mark.L2def test_02(self):assert 1==2  # 断言失败
if __name__ == '__main__':pytest.main(['-s','test.py'])

断言

我们在进行手工测试时，测试用例有预期结果，那么在进行UI测试时，我们也需要来判定结果是否符合我们的预期。

在python中用断言来实现。断言assert用于检查指定的表达式的结果是否为True。

一个测试用例可以写多个断言，当有一个断言失败时，pytest就认为测试的结果为失败，该测试用例的执行会被终止。

如果一条测试用例不写断言，那么它的结果为通过。所以一定要写断言，否则测试执行就没有了意义。

import pytestclass TestDemo:@pytest.mark.L1@pytest.mark.L2def test_01(self):assert 1==1  # 断言成功assert 1==3@pytest.mark.L2def test_02(self):assert 1==2  # 断言失败
if __name__ == '__main__':pytest.main(['-s','test.py'])

参数化

在进行UI测试时，我们需要给测试用例传测试数据，那么如何做呢？

pytest使用@pytest.mark.parametrize装饰器来实现数据驱动测试，也就是常说的参数化。

单个参数

import pytestclass TestDemo:data = ["小红", "小明"]@pytest.mark.parametrize("username",data)def test_01(self,username):print("我是{}".format(username))if __name__ == '__main__':pytest.main(['-s','test.py'])

运行结果如下，这里给一条测试用例传入两条数据，显示收集到了两条用例。

多个参数

列表嵌套字典

import pytestclass TestDemo:data_1 = [{"username": "admin1", "password": "123456"},{"username": "admin2", "password": "12345678"},]@pytest.mark.parametrize("data", data_1)def test_login(self,data):print("账号:{},密码:{}".format(data["username"], data["password"]))if __name__ == '__main__':pytest.main(['-s','test.py'])

可以看到这里传入两条数据，显示收集了两条用例。

列表嵌套列表

import pytestclass TestDemo:data_1 = [["admin1", "123456"],["admin2", "12345678"],]@pytest.mark.parametrize("username,password", data_1)def test_login(self,username, password):print("账号:{},密码:{}".format(username, password))if __name__ == '__main__':pytest.main(['-s','test.py'])

列表嵌套元组

import pytestclass TestDemo:data_1 = [("admin1", "123456"),("admin2", "12345678"),]@pytest.mark.parametrize("username,password", data_1)def test_login(self,username, password):print("账号:{},密码:{}".format(username, password))if __name__ == '__main__':pytest.main(['-s','test.py'])