在上一篇文章中，我們建立了 Pytest 夾具，它將在測試方法之前/之後建立/刪除 Postgres 資料庫。在這一部分中，我想在 Pytest 工廠固定裝置的幫助下改進固定裝置，使其更加靈活和可配置。
靜態夾具的限制
例如，如果您有多個資料庫要在測試中模擬
def test_create_user(test_db1, test_db2):
...
登入後複製
您必須創建幾乎兩個相同的燈具：
TEST_DB_URL = "postgresql://localhost"
TEST_DB1_NAME = "test_foo"
TEST_DB2_NAME = "test_bar"
@pytest.fixture
def test_db1():
with psycopg.connect(TEST_DB_URL, autocommit=True) as conn:
cur = conn.cursor()
cur.execute(f'DROP DATABASE IF EXISTS "{TEST_DB1_NAME}" WITH (FORCE)')
cur.execute(f'CREATE DATABASE "{TEST_DB1_NAME}"')
with psycopg.connect(TEST_DB_URL, dbname=TEST_DB1_NAME) as conn:
yield conn
cur.execute(f'DROP DATABASE IF EXISTS "{TEST_DB1_NAME}" WITH (FORCE)')
@pytest.fixture
def test_db2():
with psycopg.connect(TEST_DB_URL, autocommit=True) as conn:
cur = conn.cursor()
cur.execute(f'DROP DATABASE IF EXISTS "{TEST_DB2_NAME}" WITH (FORCE)')
cur.execute(f'CREATE DATABASE "{TEST_DB2_NAME}"')
with psycopg.connect(TEST_DB_URL, dbname=TEST_DB2_NAME) as conn:
yield conn
cur.execute(f'DROP DATABASE IF EXISTS "{TEST_DB2_NAME}" WITH (FORCE)')
登入後複製
Pytest 夾具工廠
「靜態」裝置在這裡有點限制。當需要幾乎相同而僅有細微差別時，您需要複製程式碼。希望 Pytest 有工廠作為固定裝置的概念。
工廠固定裝置是一個返回另一個固定裝置的固定裝置。 因為，像每個工廠一樣，它是一個函數，它可以接受參數來自訂返回的固定裝置。按照慣例，您可以在它們前面加上 make_* 前綴，例如 make_test_db。
專用夾具
我們的裝置工廠 make_test_db 的唯一參數將是要建立/刪除的測試資料庫名稱。
所以，讓我們基於 make_test_db 工廠裝置來建立兩個「專用」裝置。
用法如下：
@pytest.fixture
def test_db_foo(make_test_db):
yield from make_test_db("test_foo")
@pytest.fixture
def test_db_bar(make_test_db):
yield from make_test_db("test_bar")
登入後複製
附註：產量來自
你注意到產量了嗎？ Yield 和 Yield 之間的一個關鍵區別在於它們如何處理生成器內的資料流和控制。
在Python中，yield和yield from都在生成器函數中使用來產生一系列值，但是
-
Yield 用於暫停生成器函數的執行並向呼叫者傳回單一值。
- 而yield from用於將值的產生委託給另一個產生器。它本質上「展平」了嵌套生成器，將其生成的值直接傳遞給外部生成器的呼叫者。
也就是說，我們不想從專門的夾具“屈服”，而是從夾具工廠“屈服”。因此這裡需要yield from。
用於建立/刪除資料庫的夾具工廠
除了將程式碼包裝到內部函數之外，對我們原始夾具創建/刪除資料庫所需的更改實際上幾乎不需要任何更改。
@pytest.fixture
def make_test_db():
def _(test_db_name: str):
with psycopg.connect(TEST_DB_URL, autocommit=True) as conn:
cur = conn.cursor()
cur.execute(f'DROP DATABASE IF EXISTS "{test_db_name}" WITH (FORCE)') # type: ignore
cur.execute(f'CREATE DATABASE "{test_db_name}"') # type: ignore
with psycopg.connect(TEST_DB_URL, dbname=test_db_name) as conn:
yield conn
cur.execute(f'DROP DATABASE IF EXISTS "{test_db_name}" WITH (FORCE)') # type: ignore
yield _
登入後複製
獎勵：將遷移固定裝置重寫為工廠固定裝置
在上一部分中，我還有一個固定裝置，將 Yoyo 遷移應用於剛剛建立的空資料庫。它也不是很靈活。讓我們做同樣的事情並將實際程式碼包裝到內部函數中。
在這種情況下，因為程式碼不需要在從測試方法返回後進行清理（其中沒有yield），所以
- 工廠裝置回傳（不是yield）內部函數
- 專門的夾具呼叫（不是從工廠夾具產生）
@pytest.fixture
def make_yoyo():
"""Applies Yoyo migrations to test DB."""
def _(test_db_name: str, migrations_dir: str):
url = (
urlparse(TEST_DB_URL)
.
_replace(scheme="postgresql+psycopg")
.
_replace(path=test_db_name)
.geturl()
)
backend = get_backend(url)
migrations = read_migrations(migrations_dir)
if len(migrations) == 0:
raise ValueError(f"No Yoyo migrations found in '{migrations_dir}'")
with backend.lock():
backend.apply_migrations(backend.to_apply(migrations))
return _
@pytest.fixture
def yoyo_foo(make_yoyo):
migrations_dir = str(Path(__file__, "../../foo/migrations").resolve())
make_yoyo("test_foo", migrations_dir)
@pytest.fixture
def yoyo_bar(make_yoyo):
migrations_dir = str(Path(__file__, "../../bar/migrations").resolve())
make_yoyo("test_bar", migrations_dir)
登入後複製
需要兩個資料庫並對它們應用遷移的測試方法：
from psycopg import Connection
def test_get_new_users_since_last_run(
test_db_foo: Connection,
test_db_bar: Connection,
yoyo_foo,
yoyo_bar):
test_db_foo.execute("...")
...
登入後複製
結論
建立自己的夾具工廠，為 Pytest 方法建立和刪除資料庫實際上是練習 Python 產生器和運算子的產量/產量的一個很好的練習。
我希望這篇文章對您自己的資料庫測試套件有所幫助。請隨時在評論中留下您的問題，祝您編碼愉快！
以上是Pytest 和 PostgreSQL：每次測試的新資料庫（第二部分）的詳細內容。更多資訊請關注PHP中文網其他相關文章！
