python - 皮查姆 : Cant run unit tests with command line arguments

Question

这是我的测试类(class)

class TestProjectCreation(unittest.TestCase):

    def setUp(self):
        self.project_path = os.path.dirname(os.path.abspath(__file__))
        self.project = ClassA('adasdasd', self.project_path)

    def test_etl_project(self):
        self.assertEqual(self.project.project_name,'adasdasd')


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

这里是ClassA

class ClassA(BaseClass):

    def __init__(self, project_name, project_path):
        super(ClassA, self).__init__(project_name, project_path)

最后，这是BaseClass

class BaseClass(object):

    def __init__(self, project_name, project_path):
        self.project_path = project_path
        self.project_name = project_name
        self.parser = self.create_option_parser()
        (self.options, args) = self.parser.parse_args()
        self.check_mandatory_options(self.options)
        self.check_project_name(self.options)

    def check_mandatory_options(self, options):
        missing_arguments = []

        if not options.para1:
            missing_arguments.append('para1')
        if not options.para2:
            missing_arguments.append('para2')
        if len(missing_arguments) > 0:
            raise ValueError('Required argument(s) missing: {}'.format(missing_arguments))

注意

self.parser = self.create_option_parser()
(self.options, args) = self.parser.parse_args()

因此 BaseClass 需要一些命令行参数

在我的 pycharm 配置中

其他参数:--para1 val1 --para2 val2

但是当我运行测试时，我得到

Testing started at 11:05 AM ...
/path/to/python "/Applications/PyCharm CE.app/Contents/helpers/pydev/pydevd.py" --multiproc --qt-support=auto --client 127.0.0.1 --port 51688 --file "/Applications/PyCharm CE.app/Contents/helpers/pycharm/_jb_unittest_runner.py" --path /path/to/project/tests/project_unit_tests.py -- --para1 val1 --para2 val2
pydev debugger: process 98200 is connecting

Connected to pydev debugger (build 181.5087.37)
Usage: python -m unittest discover [options]

python -m unittest discover: error: no such option: --para1
Launching unittests with arguments python -m unittest discover -s /path/to/project/test -p project_unit_tests.py -t /path/to/project/test --para1 val1 --para2 val2 in /path/to/project/test


Process finished with exit code 2
Empty test suite.

为什么我会得到这个python -m unittest 发现:错误:没有这样的选项:--para1？

如何通过单元测试将命令行参数传递给我的对象？请注意，当我运行 main_project.py

时

def main():

    project_path = os.path.dirname(os.path.abspath(__file__))
    project = ClassA('adasdasd', self.project_path)

if __name__ == "__main__":
    sys.exit(main())

作为

python main_project.py --para1 val1 --para2 val2

这运行得很好。我也尝试过

if __name__ == '__main__':
    if len(sys.argv) != 2:
        sys.exit("ERROR command-line parameter must be supplied for these tests")
    command_line_param = sys.argv[1]
    del sys.argv[1:]
    unittest.main()

作为我的TestProjectCreation的一部分，但仍然遇到相同的错误。

我错过了什么？

Answer 1

我提供这个答案有点晚了，但也许它会对遇到这篇文章的其他人有所帮助。

您可以将单元测试中的每个测试用例视为运行测试目标的准备脚本。因为大多数单元测试框架应该在每个单独的测试用例开始时将系统重置回相同的状态，所以您可以在设置函数中将 sys.argv 设置为您希望测试的状态(大多数单元测试框架应该运行每个测试用例之间的设置函数)。

话虽如此，我一时想不起这个声明是否有异常(exception)，所以如果这不起作用，一个好的起点就是确认你的单元测试框架如何使用重置测试用例之间的测试环境。

考虑到这一点，创建看起来像这样的设置函数似乎是明智的，至少对于您的情况来说:

def setUp(self):
    import sys
    # first argument in argv is always the path and name of the script being
    # executed, in this case, it's your unit test script
    # followed by your various command args, in the order written into 
    # the terminal.
    sys.argv.extend(['--para1','val1','--para2','val2'])
    self.project_path = os.path.dirname(os.path.abspath(__file__))
    self.project = ClassA('adasdasd', self.project_path)

对于我自己的测试，这工作得很好。但我无法说明这种方法对于更复杂的测试情况是否可以推广得很好。