Question

以下粘贴包含来自三个单独Python文件的相关代码段。第一个是从命令行调用的脚本，它在给定某些参数的情况下实例化CIPuller。会发生什么是脚本被调用类似于： script.py ci（其他args被argparse吞噬）。

第二个是名为Puller的子类的一部分。第三部分是Puller的子类的一部分，名为CIPuller。

这非常有效，因为调用了正确的子类，并且使用错误的其他args的任何用户都可以看到给定子类的正确args，以及来自超类的泛型参数。（虽然我已经离线了解，也许我应该使用argparse sub-commands。）

我试图为这些类编写测试。目前，我需要ArgumentParser来实例化类，但在测试中我没有从命令行实例化事物，因此我的ArgumentParser没用。

我尝试在测试工具中创建ArgumentParser以传递给测试代码中的CIPuller's构造函数，但如果我在那里使用add_argument，则argparse可以理解为抱怨双重（重复）参数当它在add_argument构造函数中调用CIPuller时。

使用参数测试这些类的合适设计是什么？

#!/usr/bin/env python                                                             

from ci_puller import CIPuller                                                    
import argparse                                                                   
import sys                                                                        

# Using sys.argv[1] for the argument here, as we don't want to pass that onto     
# the subclasses, which should receive a vanilla ArgumentParser                   
puller_type = sys.argv.pop(1)                                                     

parser = argparse.ArgumentParser(                                                 
    description='Throw data into Elasticsearch.'                                  
)                                                                                 

if puller_type == 'ci':                                                           
    puller = CIPuller(parser, 'single')                                         
else:                                                                             
    raise ValueError("First parameter must be a supported puller. Exiting.")      

puller.run()                                                                      


class Puller(object):                                                             

    def __init__(self, parser, insert_type):                                      
        self.add_arguments(parser)                                                
        self.args = parser.parse_args()                                           

        self.insert_type = insert_type                                            

    def add_arguments(self,parser):
        parser.add_argument(                                                      
            "-d", "--debug",                                                      
            help="print debug info to stdout",                                    
            action="store_true"                                                   
        )                                                                         

        parser.add_argument(                                                      
            "--dontsend",                                                         
            help="don't actually send anything to Elasticsearch",                 
            action="store_true"                                                   
        )                                                                         

        parser.add_argument(                                                      
            "--host",                                                             
            help="override the default host that the data is sent to",            
            action='store',                                                       
            default='kibana.munged.tld'                                     
        )                             

class CIPuller(Puller):                                                           

    def __init__(self, parser, insert_type):                                      
        self.add_arguments(parser)

        self.index_prefix = "code"                                                
        self.doc_type = "cirun"                                                   

        self.build_url = ""                                                       
        self.json_url = ""                                                        
        self.result = []                                                          

        super(CIPuller, self).__init__(parser, insert_type)                       

    def add_arguments(self, parser):                                              
        parser.add_argument(                                                      
            '--buildnumber',                                                      
            help='CI build number',                                               
            action='store',                                                       
            required=True                                                         
        )                                                                         

        parser.add_argument(                                                      
            '--testtype',                                                         
            help='Job type per CI e.g. minitest / feature',                       
            choices=['minitest', 'feature'],                                      
            required=True                                                         
        )                                                                         

        parser.add_argument(                                                      
            '--app',                                                              
            help='App e.g. sapi / stats',                                         
            choices=['sapi', 'stats'],                                            
            required=True                                                         
        )

Answer 1

argparse的单元测试很棘手。有一个test/test_argparse.py文件作为整个Python单元测试的一部分运行。但它有一个复杂的自定义测试工具来处理大多数情况。

有三个基本问题，1）使用测试值调用parse_args，2）测试结果args，3）测试错误。

测试结果args相对简单。 argparse.Namespace类具有简单的__eq__方法，因此您可以测试一个名称空间与另一个名称空间。

有两种测试输入的方法。一种是修改sys.argv。最初sys.argv包含适用于测试人员的字符串。

self.args = parser.parse_args()

将sys.argv[1:]作为默认值进行测试。因此，如果您更改sys.argv，则可以测试自定义值。

但您也可以为parse_args提供自定义列表。 argparse文档在大多数示例中都使用了这个。

self.args = parser.parse_args(argv=myargv)

如果myarg为None则使用sys.argv[1:]。否则它会使用该自定义列表。

测试错误需要自定义parse.error方法（请参阅文档）或将parse_args包装在可以捕获try/except异常的sys.exit块中。

How do you write tests for the argparse portion of a python module?

python unittest for argparse

Argparse unit tests: Suppress the help message

Unittest with command-line arguments

Using unittest to test argparse - exit errors

如何测试依赖于argparse的Python类？

1 个答案: