python，unittest：有没有办法将命令行选项传递给应用程序

Question

我有一个导入 unittest 并有一些 TestCases 的模块。我想 接受一些命令行选项（例如下面的数据文件的名称）， 但是当我尝试传递选项时，我收到消息option -i not recognized。是否可以让unittest + 为应用程序提供选项（注意：我使用optparse 来处理选项）？谢谢。

$ python test_app_data.py -i data_1.txt

option -i not recognized

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

跟进：这是建议解决方案的实施：

import cfg_master  #has the optparse option-handling code

...

if __name__ == '__main__':    
    #add you app's options here...
    options_tpl = ('-i', '--in_dir', '-o', '--out_dir')
    del_lst = []
    for i,option in enumerate(sys.argv):
        if option in options_tpl:
            del_lst.append(i)
            del_lst.append(i+1)

    del_lst.reverse()
    for i in del_lst:
        del sys.argv[i]
        
    unittest.main()

Answer 1

以 Alex 的回答为基础，使用 argparse 实际上很容易做到：

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--input', default='My Input')
    parser.add_argument('filename', default='some_file.txt')
    parser.add_argument('unittest_args', nargs='*')

    args = parser.parse_args()
    # TODO: Go do something with args.input and args.filename

    # Now set the sys.argv to the unittest_args (leaving sys.argv[0] alone)
    sys.argv[1:] = args.unittest_args
    unittest.main()

我还没有测试你可以传递给 unittest 的所有标志以查看它们是否有效，但是传递测试名称确实有效，例如：

python test.py --input=foo data.txt MyTest

使用 foo 和 data.txt 运行 MyTest。

Answer 2

在您没有向我们展示的if __name__ == '__main__': 部分中，您需要先先optparse，然后再将del sys.argv[1:] 传递给unittest 代码，这样后面的代码就不会尝试当你已经处理过它们时，再次解释你的命令行选项。 （有一些您自己的选项有点困难并且也将一些选项传递给unittest，尽管如果您确实有如此复杂的需求也可以这样做）。

Answer 3

我想我会分享我的解决方案，在测试中添加一个 --debug 开关来控制记录器：

if __name__=='__main__':
     parser = argparse.ArgumentParser(description="Build a compilation script")
     parser.add_argument('--debug', help='Turn on debug messages', action='store_true', default=False)

     args = parser.parse_args()

     if args.debug:
         log_level = logging.DEBUG
     else:
         log_level = logging.INFO

     logging.basicConfig(level=log_level)
     sys.argv.pop()
     unittest.main()

然后我扩展 unittest.TestCase 以添加日志记录：

 class mcs_TestCase(unittest.TestCase, object):
     def __init__(self, *args, **kwargs):
         super(mcs_TestCase,self).__init__(*args,**kwargs)
         logging.basicConfig()
         self.logger = logging.getLogger(__name__)
         ...

现在我可以使用 --debug 在我的测试中打开和关闭消息，但它在常规回归中会被忽略。

Answer 4

对于小型独立应用程序，我使用初始标记选项 (-t) 并在调用 argparse.ArgumentParser() 之前调用 unittest.main()

if __name__ == "__main__":
    if len(sys.argv) > 1 and sys.argv[1] in ["-t", "--test"]:
        del(sys.argv[1])
        sys.exit(unittest.main()) # pass sys.argv[

    p = argparse.ArgumentParser()
    . . .

Answer 5

除了其他答案之外，我想就这个问题提供不同的看法。我知道这个问题要求提供一种为单元测试提供命令行选项的方法，但是所陈述的实际问题只是“如何将参数传递给单元测试”。

这也可以通过环境变量来实现。

这是一个示例，您可以如何在本地和构建服务器上为测试提供不同的数据：

my-script.py:

from os import environ

database = MyDatabase(
    user = 'root',
    password = 'some-pw',
    host = environ['MYSQL_HOST'],
    port = environ['MYSQL_TCP_PORT']
)

test-local.sh:

MYSQL_TCP_PORT="3311"
MYSQL_HOST="localhost"
python3.7 -m unittest

test-ci.sh:

MYSQL_TCP_PORT="3306"
MYSQL_HOST="my-database"
python3.7 -m unittest

Answer 6

您不应该使用参数和选项来运行单元测试，因为这样会使它们在不同的、不可预测的条件下运行。您应该弄清楚为什么需要使用不同的数据运行测试，并使您的测试套件足够完整以涵盖所有数据集的基础，而不会每次都以不同的方式运行。