Pytest学习笔记2-setup和teardown

我们在做自动化的时候，常常有这样的需求：

执行每一条用例时，都重新启动一次浏览器

每一条用例执行结束时，都清除测试数据

在unittest中，我们可以使用 setUp() 和 tearDown() 两个方法来实现以上需求，其中 setUp() 方法用于初始化测试固件；而 tearDown() 方法用于销毁测试固件。程序会在运行每个测试用例（以 test_ 开头的方法）之前自动执行 setUp() 方法来初始化测试固件，井在每个测试用例（以 test_ 开头的方法）运行完成之后自动执行 tearDown() 方法来销毁测试固件。

那么如何实现只启动一次浏览器，所有测试用例执行结束后再清除数据这样的需求呢？

unittest提供了setUpClass()和tearDownClass()两个方法，配合@classmethod装饰器使用即可

作为比unittest更强大的框架，pytest自然也有类似的方法

pytest的setup/teardown方法包括：

模块级别(setup_module/teardown_module)

函数级别(setup_function/teardown_function)

类级别(setup_class/ teardown_class)

方法级别(setup_method/teardown_methond或者setup/teardown)

模块级别

模块中的第一个用例开始前执行一次setup_module方法，模块中的最后一个测试用例结束后执行一次teardown_module方法

import pytest def setup_module(): print("执行setup_module") def teardown_module(): print("执行teardown_module") class TestDemo(object): def test_case1(self): print("执行测试用例1") assert 1 + 1 == 2 def test_case2(self): print("执行测试用例2") assert 1 + 3 == 4 def test_case3(self): print("执行测试用例3") assert 1 + 5 == 6

运行结果如下：

函数级别

在每个测试函数前运行一次setup_function方法，在每个测试函数结束后运行一次teardown_function方法，只对函数用例生效，不在类中。

import pytest def setup_function(): print("执行setup_function") def teardown_function(): print("执行teardown_function") def test_case1(): print("执行测试用例1") assert 1 + 1 == 2 def test_case2(): print("执行测试用例2") assert 1 + 3 == 4 def test_case3(): print("执行测试用例3") assert 1 + 5 == 6

运行结果如下：

类级别

setup_class/teardown_class 对类有效，位于类中，在执行测试类之前和之后各调用一次

import pytest class TestDemo(object): def setup_class(self): print("执行setup_class") def teardown_class(self): print("执行teardown_class") def test_case1(self): print("执行测试用例1") assert 1 + 1 == 2 def test_case2(self): print("执行测试用例2") assert 1 + 3 == 4 def test_case3(self): print("执行测试用例3") assert 1 + 5 == 6

运行结果如下：

方法级别

setup_method/teardown_method和setup/teardown，在测试类中每个测试方法前后调用一次。这两个方法效果是一样的

import pytest class TestDemo(object): def setup_method(self): print("执行setup_method") def teardown_method(self): print("执行teardown_method") def test_case1(self): print("执行测试用例1") assert 1 + 1 == 2 def test_case2(self): print("执行测试用例2") assert 1 + 3 == 4 def test_case3(self): print("执行测试用3") assert 1 + 5 == 6

运行结果如下：

四种级别混合使用

如果把这四种级别的方法混合使用，运行顺序如何呢？

import pytest 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("执行测试函数a") def test_b(): print("执行测试函数b") class TestDemo(object): 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_case1(self): print("执行测试用例1") assert 1 + 1 == 2 def test_case2(self): print("执行测试用例2") assert 1 + 3 == 4 def test_case3(self): print("执行测试用例3") assert 1 + 5 == 6

运行结果如下：

总结

模块级（setup_module/teardown_module）开始于模块始末，全局的

函数级（setup_function/teardown_function）只对函数用例生效（不在类中）

类级（setup_class/teardown_class）只在类中前后运行一次(在类中)

方法级（setup_method/teardown_method或setup/teardown）开始于方法始末（在类中）