Pytest 和 PostgreSQL：每次測試的新資料庫（第二部分）

在上一篇文章中，我們建立了 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中文網其他相關文章！