为sys.exit创建单元测试

Question

我有一个简单的python函数。

def test():
    print "test"
    sys.exit(1)

我正在使用python 2.6。 如何为此功能创建单元测试？ 这是因为sys.exit可以在python 2.7之后的单元测试用例中处理。

谁能让我知道如何为这个简单的代码创建单元测试？

Answer 1

根据我所看到的，在2.x期间sys.exit几乎没有变化。 我可以看到的唯一区别是，较早的sys.exit并没有检查它的参数，而是引发了SystemExit异常（现在只有在提供适当的参数的情况下它才会引发）。

因此，例如，我使用python 2.6（也称为2.6.0）尝试了以下操作：

import sys
try:
   sys.exit(1)
except SystemExit(1) as e:
   print repr(e)

它以SystemExit(1,)响应

因此，您的测试应类似于：

def test():
    passed = False
    try:
        sys.exit(1)
    except SystemExit as e:
        if e.code == 1:
            passed = True

    assert passed