from selenium import webdriver
from time import sleep
from random import randint
?
?
def setup_module():
    print("====  模塊級的 setup 操作  ====")
?
?
def teardown_module():
    print("====  模塊級的 teardown 操作  ====")
?
?
def test_random():
    assert randint(1, 5) == 3
?
?
class TestWeb:
?
    @classmethod
    def setup_class(cls):
        print("====  測試類級的 setup 操作  ====")
?
    @classmethod
    def teardown_class(cls):
        print("====  測試類級的 teardown 操作  ====")
?
    def setup_method(self):
        print("====  測試用例級的 setup 操作  ====")
        self.driver = webdriver.Chrome()
?
    def teardown_method(self):
        print("====  測試用例級的 teardown 操作  ====")
        self.driver.quit()
?
    def test_search(self):
        self.driver.get("https://www.baidu.com/")
        self.driver.find_element_by_id("kw").send_keys("檸檬班")
        self.driver.find_element_by_id("su").click()
        sleep(1)

pytest也支持運行unittest的測試用例。支持unittest以下特性：

@unittest.skip
setUp/tearDown;
setUpClass/tearDownClass;
setUpModule/tearDownModule;

pytest的fixture實現方式二：fixture機制

通過@pytest.fixture裝飾器來定義fixture。一個函數被@pytest.fixture裝飾，那么這個函數就是fixture。

使用fixture時，分為二個部分：fixture定義、fixture調用。

除此之外，還有fixture的共享機制，嵌套調用機制。

1、定義fixture。

1）fixture通過函數實現。

2）使用@pytest.fixture進行裝飾

import pytest
?
@pytest.fixture
def init():
    pass

3）前置準備工作代碼和后置清理工作代碼，都寫在一個函數里面。

4）通過yeild關鍵字，區(qū)分前置代碼和后置代碼。yeild之前的代碼為前置代碼，yeild之后的代碼為后置代碼

在實際應用場景當中，可以只有前置準備工作代碼，也可以只有后置清理工作代碼。

import pytest
?
?
@pytest.fixture
def init():
    print("用例執(zhí)行之前，執(zhí)行的代碼")  # 前置代碼
    yield
    print("用例執(zhí)行之后，執(zhí)行的代碼")  # 后置代碼
  
?
@pytest.fixture
def init2():
    print("用例執(zhí)行之前，執(zhí)行的代碼")  # 只有用例執(zhí)行之前的前置準備代碼
     
?
@pytest.fixture
def init3():
    yield
    print("用例執(zhí)行之后，執(zhí)行的代碼")  # 只有用例執(zhí)行之后的后置清理代碼

5）fixture有4個作用域：測試會話(session)、測試模塊(module)、測試類(class)、測試用例(function)

測試會話：pytest執(zhí)行測試用例的整個過程，稱為會話。

比如pytest收集到了100條用例并執(zhí)行完成，這個過程稱為測試會話。

設置fixture的作用域：通過@pytest.fixture(scope=作用域)來設置。默認情況下，scope=function

import pytest
?
?
# 沒有設置scope，默認為測試函數級別。即調用此fixture的測試類/模塊/函數下，每個測試函數都會執(zhí)行一次這個fixture
@pytest.fixture
def init():
    print("用例執(zhí)行之前，執(zhí)行的代碼")  # 前置代碼
    yield
    print("用例執(zhí)行之后，執(zhí)行的代碼")  # 后置代碼
?
?
# 設置scope為class。調用此fixture的測試類下，只執(zhí)行一次這個fixture.
@pytest.fixture(scope="class")
def init2():
    print("用例執(zhí)行之前，執(zhí)行的代碼")  # 只有用例執(zhí)行之前的前置準備代碼
?
?
# 設置scope為session。autouse表示自動使用。
# 那么在pytest收集用例后，開始執(zhí)行用例之前會自動化執(zhí)行這個fixture當中的前置代碼，
# 當所有用例執(zhí)行完成之后，自動化執(zhí)行這個fixture的后置代碼。
@pytest.fixture(scope="session",autouse=True)
def init3():
    yield
    print("用例執(zhí)行之后，執(zhí)行的代碼")  # 只有用例執(zhí)行之后的后置清理代

6）fixture的返回值設置：yeild 返回值

當測試用例當中，要使用fixture里生成的數據時，則需要fixture返回數據。

若有數據返回則：yeild 返回值

import pytest
from selenium import webdriver
from time import sleep
?
?
# 設置scope為class。調用此fixture的測試類下，只執(zhí)行一次這個fixture.
@pytest.fixture(scope="class")
def init2():
    print("==== 測試類下，執(zhí)行所有用例之前，執(zhí)行的代碼 ====")
    driver = webdriver.Chrome()
    yield driver   # 返回driver對象
    print("==== 測試類下，執(zhí)行所有用例之后，執(zhí)行的代碼 ====")
    driver.quit()