在 pytest 中，有没有办法模拟一个已经导入的变量而不必为每个测试打补丁？

Question

因此，具体来说，我们使用 SqlAlchemy 和 session，并且我们只在utils/sessions.py中定义了一次。 像下面这样：

#utils/sessions.py

from sqlalchemy.orm import scoped_session, sessionmaker

session = scoped_session (sessionmaker(
 ....
))

我们在实际的存储库层上使用它，例如：

#app/user/users.py
from utils.sessions import session

class UserRepo :
  def get_user(self,id) :
    return session.query(User).filter(User.id=id)

现在我正在尝试执行某种单元/集成测试，我需要在测试中的几乎所有地方模拟来自utils.sessions的session变量（因此我想使用自动夹具 mocking util.sessions.session ）。

到目前为止，似乎唯一可行的是在测试代码本身中修补app.user.users.session ，而我想要的是在不同的测试中基本上模拟session的每一次出现。 比如，如果我想测试app/articles ，我不想重复输入with patch('apps.articles.article.session'不必要的。

所以理想情况下，我想要这样的东西，并期望我在个人回购中导入的所有session都能正常工作：

# tests/conftest.py

def mocked_session() :
 return scoped_session("my_new_parameters_for_test_purpose_only")

@pytest.fixture(autouse=true)
def mock_all_sessions():
 with patch("utils.sessions.session", mocked_session) as s :
  yield s 

但这对我不起作用。 （这对我来说很奇怪，因为修补socket.socket成功阻止了网络尝试。）

如果有一种方法可以模拟utils.sessions.session而不是手动修补apps/{all apps i have}/{all.py files}.session/ 。

有没有办法做到这一点？ 如果没有，是否可以替代为每次测试手动写下模块路径？

Answer 1

问题是您仅在一个命名空间中模拟session ，即utils.sessions.__dict__ ，但名称“session”存在于许多其他命名空间中，例如apps.articles.article.__dict__ 。

通过简单地更改 import 语句，可以有一个低技术含量的解决方案——而不是

from utils.sessions import session

你可以使用

from utils import sessions

然后你只需要像你已经在做的那样在一个命名空间（查找名称的地方）中进行模拟。 需要注意的是，所有这些子模块现在都需要使用sessions.session而不仅仅是session ，这样它们都在您应用补丁的同一个命名空间中查找。

Answer 2

请参阅单元测试文档https://docs.python.org/3/library/unittest.mock.html#where-to-patch

主要问题是，当你打补丁时，你只是在模块上下文中打补丁名称变量。 当users.py from utils.sessions import session时，它会创建自己对 session 的引用。即有两个变量名称指向同一个 object。

utils.sessions.session = <your session object>
app.user.users.session = <your session object>

当您只修补utils.sessions.session时，您只是替换了该模块中名称的内容。 例如

utils.sessions.session = <the patched mock object>
app.user.users.session = <your session object>

如果您改为from utils import sessions执行导入，那么您会将 object 引用为sessions.session ，这意味着当您修补sessions模块上的值时，它将在运行时找到对修补值的正确覆盖。

IMO 这变得很尴尬，因为它强加了一种执行导入的方式，而这两种方式在技术上都应该是有效的。 一个更简单的模式可能是在sessions.py中创建一个代理方法，它将在模块的导入器和访问 session object 之间设置一个障碍。这看起来像这样：

_session = scoped_session(sessionmaker(...))

def session():
    return _session

像这样使用：

from utils.sessions import session

class UserRepo:
  def get_user(self,id):
    return session().query(User).filter(User.id=id)

现在，当您 go 打补丁时，您只需要打补丁utils.sessions._session ，无论您如何导入都无关紧要。 对 function session()的引用保持不变，但它返回的 object 对每个人都发生了变化。