如何在单元测试期间测试必需的argparse参数？

Question

我正在为argparser做TDD测试。 如何使用必需的选项测试参数？我需要测试所有选项，例如：

• 争论太多，
• 没有给出任何论据，
• 给出了错误的论点。

我可以提高SystemExit，但这并不是我真正需要的：

    def test_no_arguments(self):
        with patch.object(sys, 'exit') as mock_method:
            self.parser.parse_arguments()
            self.assertTrue(mock_method.called)

但是，在不提高系统退出强度的情况下，我总是会遇到如下错误：

zbx-check-mount.py

class CommandLine:
    def __init__(self):
        self.args_parser = argparse.ArgumentParser(description="Monitoring mounted filesystems",
                                                   formatter_class=argparse.RawTextHelpFormatter)
        self.parsed_args = None

        self.add_arguments()

    def add_arguments(self):
        """
        Add arguments to parser.
        """
        try:
            self.args_parser._action_groups.pop()  # pylint: disable=protected-access
            required = self.args_parser.add_argument_group('required arguments')
            required.add_argument('--fs_name', required=True, help='Given filesystem')
        except argparse.ArgumentError as err:
            log.error('argparse.ArgumentError: %s', err)
            sys.exit(1)

    def parse_arguments(self, args=None):
        """
        Parse added arguments. Then run private method to return values
        """
        self.parsed_args = self.args_parser.parse_args()

        return self.parsed_args.fs_name,

测试

from pyfakefs.fake_filesystem_unittest import TestCase
import os
import sys

try:
    from StringIO import StringIO
except ImportError:
    from io import StringIO

if sys.version_info[0] == 3:
    from unittest.mock import MagicMock, patch
else:
    from mock import MagicMock, patch


sys.path.extend([os.path.join(os.path.dirname(os.path.abspath(__file__)),'..','..', "bin")])
module_name = __import__('zbx-check-mount')


class TestCommandLine(TestCase):

    def setUp(self):
        """
        Method called to prepare the test fixture. This is called immediately before calling the test method
        """
        self.parser = module_name.CommandLine()

    def test_no_arguments(self):
        opts = self.parser.parse_arguments([])
        assert opts.fs_name

    def tearDown(self):
        """
        Method called immediately after the test method has been called and the result recorded.
        """
        pass

如何避免这种情况并测试其他选项？

Answer 1

在def parse_arguments(self, args=None): ，您应该将args传递给解析器，如下所示：

self.args_parser.parse_args(args)

parse_args()解析sys.argv[1:] ，或者如果给定参数为None 。 否则，它将解析提供的列表。

在完整的python发行版中，有一个argparse的unittest文件（ test_argparse.py ）。 它有点复杂，定义ArgumentParser的子类来捕获错误并重定向错误消息。

测试argparse非常棘手，因为它查看sys.argv ，同时也使用unittest脚本。 它通常会尝试在错误时退出。 已经在许多SO问题中对此进行了讨论。

Answer 2

如果我正确地解释了您的症状，那么您在测试工具中就遇到了问题，因为您的猴子补丁的sys.exit实现实际上返回了，这是argparse库所期望的。

引入引发异常的side_effect ，然后可以在单元测试中进行捕获和验证，可以解决该问题。