pytest fixture-參數(shù)化fixture

??fixture??函數(shù)可以被參數(shù)化，在這種情況下，它們將被多次調用，每次執(zhí)行一組依賴于該??fixture??的測試。測試函數(shù)通常不需要知道它們的重新運行。夾具參數(shù)化有助于為組件編寫詳盡的功能測試，這些組件本身可以以多種方式配置。

擴展前面的示例，我們可以標記??fixture??來創(chuàng)建兩個??smtp_connection fixture??實例，這將導致使用該??fixture??的所有測試運行兩次。??fixture??函數(shù)通過特殊的??request??對象訪問每個參數(shù):

# content of conftest.py
import pytest
import smtplib


@pytest.fixture(scope="module", params=["smtp.gmail.com", "mail.python.org"])
def smtp_connection(request):
    smtp_connection = smtplib.SMTP(request.param, 587, timeout=5)
    yield smtp_connection
    print("finalizing {}".format(smtp_connection))
    smtp_connection.close()

主要的變化是用??@pytest??聲明??params??。??Fixture??，一個值列表，??Fixture??函數(shù)將執(zhí)行其中的每個值，并可以通過??request.param??訪問一個值。不需要更改測試函數(shù)代碼。我們再運行一次

$ pytest -q test_module.py
FFFF                                                                 [100%]
================================= FAILURES =================================
________________________ test_ehlo[smtp.gmail.com] _________________________

smtp_connection = <smtplib.SMTP object at 0xdeadbeef0004>

    def test_ehlo(smtp_connection):
        response, msg = smtp_connection.ehlo()
        assert response == 250
        assert b"smtp.gmail.com" in msg
>       assert 0  # for demo purposes
E       assert 0

test_module.py:7: AssertionError
________________________ test_noop[smtp.gmail.com] _________________________

smtp_connection = <smtplib.SMTP object at 0xdeadbeef0004>

    def test_noop(smtp_connection):
        response, msg = smtp_connection.noop()
        assert response == 250
>       assert 0  # for demo purposes
E       assert 0

test_module.py:13: AssertionError
________________________ test_ehlo[mail.python.org] ________________________

smtp_connection = <smtplib.SMTP object at 0xdeadbeef0005>

    def test_ehlo(smtp_connection):
        response, msg = smtp_connection.ehlo()
        assert response == 250
>       assert b"smtp.gmail.com" in msg
E       AssertionError: assert b'smtp.gmail.com' in b'mail.python.org\nPIPELINING\nSIZE 51200000\nETRN\nSTARTTLS\nAUTH DIGEST-MD5 NTLM CRAM-MD5\nENHANCEDSTATUSCODES\n8BITMIME\nDSN\nSMTPUTF8\nCHUNKING'

test_module.py:6: AssertionError
-------------------------- Captured stdout setup ---------------------------
finalizing <smtplib.SMTP object at 0xdeadbeef0004>
________________________ test_noop[mail.python.org] ________________________

smtp_connection = <smtplib.SMTP object at 0xdeadbeef0005>

    def test_noop(smtp_connection):
        response, msg = smtp_connection.noop()
        assert response == 250
>       assert 0  # for demo purposes
E       assert 0

test_module.py:13: AssertionError
------------------------- Captured stdout teardown -------------------------
finalizing <smtplib.SMTP object at 0xdeadbeef0005>
========================= short test summary info ==========================
FAILED test_module.py::test_ehlo[smtp.gmail.com] - assert 0
FAILED test_module.py::test_noop[smtp.gmail.com] - assert 0
FAILED test_module.py::test_ehlo[mail.python.org] - AssertionError: asser...
FAILED test_module.py::test_noop[mail.python.org] - assert 0
4 failed in 0.12s

我們看到，我們的兩個測試函數(shù)針對不同的??smtp_connection??實例運行了兩次。還要注意，對于??mail.python.org??連接，第二個測試在??test_ehlo??中失敗，因為預期的服務器字符串與到達的服務器字符串不同。

pytest將構建一個字符串，它是參數(shù)化??fixture??中每個??fixture??值的測試ID，例如上面示例中的??test_ehlo[smtp.gmail.com]??和??test_ehlo[mail.python.org]??。這些ids可以與??-k??一起使用，以選擇要運行的特定案例，并且它們還將在某個案例失敗時標識特定案例。使用??——collect-only??運行pytest將顯示生成的id。

數(shù)字、字符串、布爾值和None將在測試ID中使用它們通常的字符串表示。對于其他對象，pytest將根據參數(shù)名生成一個字符串。通過使用?ids?關鍵字參數(shù)，可以定制測試ID中用于特定??fixture??值的字符串:

# content of test_ids.py
import pytest


@pytest.fixture(params=[0, 1], ids=["spam", "ham"])
def a(request):
    return request.param


def test_a(a):
    pass


def idfn(fixture_value):
    if fixture_value == 0:
        return "eggs"
    else:
        return None


@pytest.fixture(params=[0, 1], ids=idfn)
def b(request):
    return request.param


def test_b(b):
    pass

上面展示了如何將ids作為一個字符串列表來使用，或者作為一個函數(shù)來調用??fixture??值，然后返回一個字符串來使用。在后一種情況下，如果函數(shù)返回?None?，那么將使用pytest自動生成的ID。

運行上述測試會在以下正在使用的測試ids中產生結果:

$ pytest --collect-only
=========================== test session starts ============================
platform linux -- Python 3.x.y, pytest-7.x.y, pluggy-1.x.y
rootdir: /home/sweet/project
collected 11 items

<Module test_anothersmtp.py>
  <Function test_showhelo[smtp.gmail.com]>
  <Function test_showhelo[mail.python.org]>
<Module test_emaillib.py>
  <Function test_email_received>
<Module test_ids.py>
  <Function test_a[spam]>
  <Function test_a[ham]>
  <Function test_b[eggs]>
  <Function test_b[1]>
<Module test_module.py>
  <Function test_ehlo[smtp.gmail.com]>
  <Function test_noop[smtp.gmail.com]>
  <Function test_ehlo[mail.python.org]>
  <Function test_noop[mail.python.org]>

======================= 11 tests collected in 0.12s ========================