如何用pytes修补注册表连接

Question

我目前正在为我的python脚本编写单元测试。我的一个函数具有以下代码：

from _winreg import *

aReg = ConnectRegistry(None, HKEY_LOCAL_MACHINE)
aKey = OpenKey(aReg, LIBRARIES_REG_KEY)

count, value, type = QueryInfoKey(aKey)

ver_list = [EnumKey(aKey, x) for x in range(count) if EnumKey(aKey, x).startswith(version)]

我大部分测试都使用monkeypatch。我想知道是否有一种方法可以通过monkeypatch ConnectRegistry（甚至OpenKey）函数来返回我想要的数据。我会给它提供不同的数据并测试它的行为。你知道吗

Answer 1

导入是否真的在被测试的函数中？如果不是的话，会有这样的工作吗？你知道吗

def my_CR(arg1, arg2):
    return something_useful

def my_OK(arg1, arg2):
    return something_else_useful

def test_myfunc():
    ConnectRegistry = my_CR
    OpenKey = my_OK
    assert function_calling_CR_and_OK(...) == expected_value

您还需要monkeypatch OpenKey和EnumKey吗？你知道吗