如何测试或模拟“如果__name__ =='__main__'”内容

Question

说我有一个包含以下内容的模块：

def main():
    pass

if __name__ == "__main__":
    main()

我想为下半部分编写一个单元测试（我想实现100％的覆盖率）。我发现执行导入/设置机制的内置的 runpy
模块__name__，但无法弄清楚如何模拟或检查 main（） 函数是否被调用。

到目前为止，这是我尝试过的：

import runpy
import mock

@mock.patch('foobar.main')
def test_main(self, main):
    runpy.run_module('foobar', run_name='__main__')
    main.assert_called_once_with()

Answer 1

我将选择另一种替代方法，将其if __name__ == '__main__'从覆盖率报告中排除，当然，只有在测试中已经具有main（）函数的测试用例的情况下，您才能这样做。

至于为什么我选择排除而不是为整个脚本编写新的测试用例，是因为如果按照我说的那样，您已经为您的main()函数准备了一个测试用例，那么您为脚本添加了另一个测试用例（仅适用于100％的覆盖率）将只是重复项。

有关如何排除的信息，if __name__ == '__main__'您可以编写coverage配置文件并添加到部分报告中：

[report]

exclude_lines =
    if __name__ == .__main__.:

可以在此处找到有关coverage配置文件的更多信息。

希望这会有所帮助。