pytest框架使用

一、介绍

在python自动化中主流的两种单元测试框架，一种是unittest，另一种就是pytest，pytest主要是的简单灵活，容易上手，支持参数化，不仅可以做单元测试，还支持复杂的功能测试，拥有丰富的第三方插件，自定义扩展，兼容unittest框架，两者可以同时使用等优点。

二、前置与后置

在unittest中有前置与后置，而在pytest中也同样拥有前置和后置

import pytest

class Test_A:

    def setup(self):
        print('每条用例之前')

    def teardown(self):
        print("每条用例之后")

    def setup_class(self):
        print('每个类之前')
    def teardown_class(self):
        print('每个类之后')

    def test1(self):
        print('这是第一条用例')
        assert 1>0

    def test2(self):
        print('这是第二条用例')
        assert 2>1

结果如下

三、ini文件配置

[pytest]
addopts = -s  --html=report/ini.html
testpaths = ./test_pytest.py    执行路径
python_files = test*.py         识别可执行的文件名
python_classes = Test*    识别可执行的类名
python_functions = test   识别可执行的方法名
markers =          打标记
    g1:group1.
    g2:group2.
    smock:group3.

代码如下

import pytest

class Test_A:

    @pytest.mark.g1
    def test1(self):
        print('这是第一条用例')
        assert 1>0

    @pytest.mark.g2
    def test2(self):
        print('这是第二条用例')
        assert 2>1

    @pytest.mark.smock
    def test3(self):
        print('这是第三条用例')
        assert 2>1

if __name__ == '__main__':
    pytest.main(['-s','aa.py','-m g1','-vv'])

显示结果如下

四、fixture

1、范围
fixture里面scope参数可以控制fixture的范围：session>module>class>function
function：每一个函数或方法都会调用
class：每一个类调用一次，一个类中可以有多个方法
module：每一个.py文件调用一次，该文件内又有多个function和class
session：是多个文件调用一次，可以跨.py文件调用，每个.py文件就是module

2、源码
fixture（scope=‘function’，params=None，autouse=False，ids=None，name=None）：
scope：有四个级别参数"function"（默认），“class”，“module”，“session”
params：一个可选的参数列表，它将导致多个参数调用fixture功能和所有测试使用它。
autouse：为True，所有测试fixture func可以看到它。为False则显示需要参考来**fixture
ids：每个字符串id的列表，每个字符串对应于params这样他们就是测试ID的一部分。如果没有提供ID它们将从params自动生成
name：fixture的名称。默认为装饰函数的名称
前置使用方法一

import pytest

@pytest.fixture()
def a():
    print('初始化数据')

class Test_A:

    def test1(self,a):  # 当作参数传入，执行用例前会先调用函数a
        print('这是第一条用例')
        assert 1 > 0

    def test2(self):
        print('这是第二条用例')
        assert 2 > 1

if __name__ == '__main__':
    pytest.main(['-s','aa.py','-vv'])

效果如下

方法二
每个方法运行一次

import pytest

@pytest.fixture()
def a():
    print('初始化数据')

@pytest.mark.usefixtures("a")  # 每个用例都执行一次
class Test_A:

    def test1(self):
        print('这是第一条用例')
        assert 1 > 0

    def test2(self):
        print('这是第二条用例')
        assert 2 > 1

if __name__ == '__main__':
    pytest.main(['-s','aa.py','-vv'])

效果如下

方法三
每个类执行一次

import pytest

@pytest.fixture(scope='class')
def a():
    print('初始化数据')

@pytest.mark.usefixtures("a")  # a,一个类只运行一次
class Test_A:

    def test1(self):
        print('这是第一条用例')
        assert 1 > 0

    def test2(self):
        print('这是第二条用例')
        assert 2 > 1


if __name__ == '__main__':
    pytest.main(['-s','aa.py','-vv'])

效果如下

方法四
参数化

import pytest

@pytest.fixture(params=[22,33])
def a(request):
    return request.param

class Test_A:

    def test1(self,a):
        print('这是第一条用例')
        assert a>0
    def test2(self):
        print('这是第二条用例')
        assert 2 > 1


if __name__ == '__main__':
    pytest.main(['-s','aa.py','-vv'])

效果如下