时间戳

首页 > 解决方案 > 如何将模拟对象传递给 pytest 夹具

python-3.x - 如何将模拟对象传递给 pytest 夹具

问题描述

对不起标题，我一定要更新它......我有一个 pytest 测试功能：

def test_update_db():
    mock_connection = Mock(spec=Connection)
    db_updater = DbUpdater(mock_connection)

    db_updater.run("some parameter")

    mock_connection.gna.assert_called_with("some different parameter")

这行得通，但很难看：db_updater 应该真的是一个固定装置。但我必须将连接对象传递给它，所以我宁愿：

@pytest.fixture
def db_updater():
    mock_connection = Mock(spec=Connection)
    return DbUpdater(mock_connection)


def test_update_db(db_updater):
    db_updater.run("some parameter")
    mock_connection.gna.assert_called_with("some different parameter")

更好的测试功能有一个问题：mock_connection 不存在那里......我该如何解决这个问题？

标签： python-3.xmockingpytestfixtures

解决方案

您可以定义一个夹具以依赖于另一个夹具，然后让它们在您的测试中都是 args。这应该有效：

import pytest

@pytest.fixture
def conn():
    return Mock(spec=Connection)

@pytest.fixture
def db_updater(conn):
    return DbUpdater(conn)

def test_update_db(conn, db_updater):
    db_updater.run("some parameter")
    conn.gna.assert_called_with("some different parameter")

推荐阅读