pytest框架的优缺点（Pytest2-setup和teardown）

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

• 执行每一条用例时，都重新启动一次浏览器
• 每一条用例执行结束时，都清除测试数据

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

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

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

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

pytest的setup/teardown方法包括：

• 模块级别(setupmodule/teardownmodule)
• 函数级别(setupfunction/teardownfunction)
• 类级别(setupclass/ teardownclass)
• 方法级别(setupmethod/teardownmethond或者setup/teardown)

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

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

运行结果如下：

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

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

运行结果如下：

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

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

运行结果如下：

setupmethod/teardownmethod和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

运行结果如下：

• 模块级（setupmodule/teardownmodule）开始于模块始末，全局的
• 函数级（setupfunction/teardownfunction）只对函数用例生效（不在类中）
• 类级（setupclass/teardownclass）只在类中前后运行一次(在类中)
• 方法级（setupmethod/teardownmethod或setup/teardown）开始于方法始末（在类中）