如何使用pytest monkeypatch注册表连接

Question

I am currently writing unit tests for my python script. One of my functions has the following code:

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)]

I am using monkeypatch for most of my tests. I was wondering if there was a way to monkeypatch the ConnectRegistry (or even the OpenKey) function to return the data I want. I would feed it different data and test the behavior.

Answer 1

Is the import actually in the function being tested? If not, would something like this work?

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

Would you also need to monkeypatch OpenKey and EnumKey?