在上一篇文章中，我们创建了 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 有工厂作为固定装置的概念。
工厂固定装置是一个返回另一个固定装置的固定装置。因为，像每个工厂一样，它是一个函数，它可以接受参数来自定义返回的固定装置。按照惯例，您可以在它们前面加上 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 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），所以
@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中文网其他相关文章！
