如何测试或模拟“if\u name\uuu=='\uu main\uu'”内容

Question

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

def main():
    pass

if __name__ == "__main__":
    main()

我想为下半部分编写一个单元测试（我想实现100%的覆盖率）。我发现了执行import/__name__设置机制的runpy内置模块，但是我不知道如何模拟或检查是否调用了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__'，可以编写覆盖率配置文件并在节报告中添加：

[report]

exclude_lines =
    if __name__ == .__main__.:

有关覆盖率配置文件的详细信息可以找到here。

希望这能有所帮助。