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)開始於方法始末(在類中)