python - 系统("pause")在单元测试中

Question

我有一个已编译为 .exe 的 Python 应用程序。
在某些情况下，程序以 sys.exit () 终止，但为了不关闭控制台窗口，我使用下面的代码执行此操作，效果很好:

try:
    sys.exit()
except SystemExit:
    raise
finally:
    os.system("pause")

问题是，我需要测试我的代码是否引发 SystemExit，但由于最终阻塞，测试无法通过 - 它们只是无限的。
有没有可能让unittest通过os.system("pause")？
我可以添加一个默认参数，这将防止暂停，但在我的情况下会很复杂 - 如果有任何可能性我想在单元测试中进行。

Answer 1

我会简单地模拟os.system。甚至 sys.exit 还可以断言它是否被调用。

# your_module.py
import sys 
import os


def something():
    try:
        sys.exit()
    except SystemExit:
        raise
    finally:
        os.system("pause")

还有测试

import unittest
import your_module
try:
    from unittest.mock import patch
except ImportError:
    from mock import patch


class TestSysexit(unittest.TestCase):

    @patch('your_module.os.system')
    def test_system_exit(self, mock_os_system):

        with self.assertRaises(SystemExit):
            your_module.something()
        mock_os_system.assert_called_once_with('pause')

if __name__ == '__main__':
    unittest.main()

模拟和补丁位于 python3 上的 unittest.mock 和 python2 上的模拟库中。