使用 MagicMock 在 Python 中模拟嵌套导入
Mock nested import in Python with MagicMock
我的文件(ensure_path.py
):
import os
def ensure_path(path):
if not os.path.exists(path):
os.makedirs(path)
return path
我的测试:
import unittest
from unittest.mock import patch, MagicMock
from src.util.fs.ensure_path import ensure_path
FAKE_PATH = '/foo/bar'
class EnsurePathSpec(unittest.TestCase):
@patch('os.path.exists', side_effect=MagicMock(return_value=False))
@patch('os.makedirs', side_effect=MagicMock(return_value=True))
def test_path_exists_false(self, _mock_os_path_exists_false, _mock_os_makedirs):
ensure_path(FAKE_PATH)
_mock_os_path_exists_false.assert_called_with(FAKE_PATH)
_mock_os_makedirs.assert_called_with(FAKE_PATH)
@patch('os.path.exists', side_effect=MagicMock(return_value=True))
@patch('os.makedirs', side_effect=MagicMock(return_value=True))
def test_path_exists_true(self, _mock_os_path_exists_true, _mock_os_makedirs):
ensure_path(FAKE_PATH)
_mock_os_path_exists_true.assert_called_with(FAKE_PATH)
_mock_os_makedirs.assert_not_called()
这是给出失败的断言 Expected call: makedirs('/foo/bar')
,我认为这是有道理的,因为我认为我在错误的级别上嘲笑 os.makedirs
。
我试过用 @patch('src.util.fs.ensure_path.os.makedirs',
替换 @patch('os.makedirs',
和一些变体,但我得到
ImportError: No module named 'src.util.fs.ensure_path.os'; 'src.util.fs.ensure_path' is not a package
这是我的 __init__.py
流程:
我是否遗漏了明显的修复方法?
您的补丁参数需要与@patch 装饰器的顺序相反。
我的文件(ensure_path.py
):
import os
def ensure_path(path):
if not os.path.exists(path):
os.makedirs(path)
return path
我的测试:
import unittest
from unittest.mock import patch, MagicMock
from src.util.fs.ensure_path import ensure_path
FAKE_PATH = '/foo/bar'
class EnsurePathSpec(unittest.TestCase):
@patch('os.path.exists', side_effect=MagicMock(return_value=False))
@patch('os.makedirs', side_effect=MagicMock(return_value=True))
def test_path_exists_false(self, _mock_os_path_exists_false, _mock_os_makedirs):
ensure_path(FAKE_PATH)
_mock_os_path_exists_false.assert_called_with(FAKE_PATH)
_mock_os_makedirs.assert_called_with(FAKE_PATH)
@patch('os.path.exists', side_effect=MagicMock(return_value=True))
@patch('os.makedirs', side_effect=MagicMock(return_value=True))
def test_path_exists_true(self, _mock_os_path_exists_true, _mock_os_makedirs):
ensure_path(FAKE_PATH)
_mock_os_path_exists_true.assert_called_with(FAKE_PATH)
_mock_os_makedirs.assert_not_called()
这是给出失败的断言 Expected call: makedirs('/foo/bar')
,我认为这是有道理的,因为我认为我在错误的级别上嘲笑 os.makedirs
。
我试过用 @patch('src.util.fs.ensure_path.os.makedirs',
替换 @patch('os.makedirs',
和一些变体,但我得到
ImportError: No module named 'src.util.fs.ensure_path.os'; 'src.util.fs.ensure_path' is not a package
这是我的 __init__.py
流程:
我是否遗漏了明显的修复方法?
您的补丁参数需要与@patch 装饰器的顺序相反。