我使用的是pytest和5+测试,它们的前五行完全相同。是否可以为重复代码创建设置函数?
@mock.patch('abcde.xyz')
def test_1(mocked_function):
x_mock = mock.Mock(X)
x_mock.producer = mock.Mock()
x_mock.producer.func2 = lambda : None
mocked_function.return_value = x_mock # xyz() returns x_mock
......
@mock.patch('abcde.xyz')
def test_2(mocked_function):
x_mock = mock.Mock(X)
x_mock.producer = mock.Mock()
x_mock.producer.func2 = lambda : None
mocked_function.return_value = x_mock
......
@mock..... # more
发布于 2020-10-08 07:57:01
您应该考虑使用fixture,因为它建议使用传统的设置/拆卸方法。
来自pytest documentation
虽然这些设置/拆卸方法对于那些来自单元测试或鼻部背景的人来说是简单和熟悉的,但您也可以考虑使用pytest的更强大的fixture机制,该机制利用了依赖注入的概念,允许使用更模块化和更具可扩展性的方法来管理测试状态,特别是对于大型项目和功能测试。
对于您的示例-考虑到mocked_function
本身是一个fixture -它将是:
@pytest.fixture()
def mock_abcde_xyz(mocker):
mocker.patch("abcde.xyz")
@pytest.fixture()
@pytest.mark.usefixtures("mock_abcde_xyz")
def patched_mocked_function(mocker, mocked_function):
x_mock = mocker.Mock(X)
x_mock.producer = mocker.Mock()
x_mock.producer.func2 = lambda : None
mocked_function.return_value = x_mock
return mocked_function
def test_1(patched_mocked_function):
......
def test_2(patched_mocked_function):
......
请注意,我使用了pytest-mock而不是mock,这样您就可以使用'mocker‘fixture了。
如果您不需要pytest-mock,只需执行以下操作:
@pytest.fixture()
@mock.patch('abcde.xyz')
def patched_mocked_function(mocked_function):
x_mock = mock.Mock(X)
x_mock.producer = mock.Mock()
x_mock.producer.func2 = lambda : None
mocked_function.return_value = x_mock
return mocked_function
def test_1(patched_mocked_function):
......
def test_2(patched_mocked_function):
......
发布于 2020-10-08 07:13:07
可以,您可以实现setUp()
(或setUpClass()
)方法。
另一种方法是实现一个helper函数,就像在任何其他Python代码中一样。
https://stackoverflow.com/questions/64253531
复制相似问题