python-3.x - Pytest - 如何模拟或修补全局变量

Question

我有以下文件，我正在这个文件上运行 pytests。用例之一是测试全局记录器功能。我们如何在我的 pytest 中模拟这个？

import spark_fn

def run_job(params):
    do_something()
    logger.info("some info")

def main(params):
    global logger
    app_name = "test app"
    spark, logger = spark_fn(app_name)
    run_job(params)

if __name__ == "__main__":
    params = "some params"
    main(params)


===============================================================

Answer 1

这样的事情怎么样？你不应该把它想象成一个全局变量，而更像是修补什么 spark_fn正在返回一个“假”记录器类。我也很好奇你如何使用 spark_fn模块整体以及您如何导入它。

import yoursampletest

def test_monkeytest(monkeypatch):
    """
        Monkey patches a fake logger into spark_fn
    """
    class myFakeLogger:

        def info(self, value):
            return value

    def mock_info(mylogstatement):
        return 1, myFakeLogger()

    monkeypatch.setattr(yoursampletest, 'spark_fn', mock_info)

    assert yoursampletest.main(1) == 'some info'