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

Question

抱歉标题，我一定要更新...我有一个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 在那里不存在......我该如何解决这个问题？

Answer 1

您可以定义一个 fixture 以依赖另一个 fixture ，然后在您的测试中将它们都设为 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")