Python unittest para argparse
Tengo una función dentro de un módulo que crea unargparse
:
def get_options(prog_version='1.0', prog_usage='', misc_opts=None):
options = [] if misc_opts is None else misc_opts
parser = ArgumentParser(usage=prog_usage) if prog_usage else ArgumentParser()
parser.add_argument('-v', '--version', action='version', version='%(prog)s {}'.format(prog_version))
parser.add_argument('-c', '--config', dest='config', required=True, help='the path to the configuration file')
for option in options:
if 'option' in option and 'destination' in option:
parser.add_argument(option['option'],
dest=option.get('destination', ''),
default=option.get('default', ''),
help=option.get('description', ''),
action=option.get('action', 'store'))
return parser.parse_args()
Una muestramyapp.py
sería:
my_options = [
{
"option": "-s",
"destination": "remote_host",
"default": "127.0.0.1",
"description": "The remote server name or IP address",
"action": "store"
},
]
# Get Command Line Options
options = get_options(misc_opts=my_options)
print options.config
print options.remote_host
y esto se llamará como:
$> python myapp.py -c config.yaml
$> config.yaml
127.0.0.1
gt; python myapp.py -c config.yaml
$> python myapp.py -c config.yaml
$> config.yaml
127.0.0.1
gt; config.yaml
127.0.0.1
Ahora, estoy tratando de crear una prueba unitaria para esta función, pero mi problema es que no puedo pasar los parámetros de la línea de comandos a través del código de prueba.
# mytest.py
import unittest
from mymodule import get_options
class argParseTestCase(unittest.TestCase):
def test_parser(self):
options = get_options()
# ...pass the command line arguments...
self.assertEquals('config.yaml', options.config) # ofcourse this fails because I don't know how I will pass the command line arguments
Mi problema es que necesito pasar los argumentos de la línea de comando aget_options()
Pero no sé cómo hacerlo correctamente.
Llamada adecuada esperada:python mytest.py
(-c config.yaml
debe pasarse de alguna manera dentro del código de prueba).
Lo que está "funcionando" / no funciona en este momento:
python mytest.py -c config.yaml
Tampoco está funcionando. DevolucionesAttributeError: 'module' object has no attribute 'config'
ya que espera que llameargParseTestCase
en lugar. En otras palabras,python mytest.py -c argParseTestCase
"funciona" pero por supuesto sería un retornoAssertionError: 'config.yaml' != 'argParseTestCase'
python mytest.py -v
ejecutar la prueba unitaria en modo detallado también falla. Vuelve:
test_parser (principal.argParseTestCase) ... mytest.py 1.0 ERROR ERROR: test_parser (principal.argParseTestCase)
Rastreo (última llamada reciente): archivo "tests / unit_tests / mytest.py", línea 376, en test_parser options = get_options () File "/root/test/lib/python2.7/site-packages/mymodule.py" , línea 61, en get_options return parser.parse_args ()
Archivo "/usr/local/lib/python2.7/argparse.py", línea 1701, en parse_args args, argv = self.parse_known_args (args, namespace)
Archivo "/usr/local/lib/python2.7/argparse.py", línea 1733, en el espacio de nombres parse_known_args, args = self._parse_known_args (args, namespace)
Archivo "/usr/local/lib/python2.7/argparse.py", línea 1939, en _parse_known_args start_index = consume_optional (start_index)
Archivo "/usr/local/lib/python2.7/argparse.py", línea 1879, en consume_optional take_action (action, args, option_string)
Archivo "/usr/local/lib/python2.7/argparse.py", línea 1807, en la acción take_action (self, namespace, argumento_valores, cadena_opción)
Archivo "/usr/local/lib/python2.7/argparse.py", línea 1022, enllamada parser.exit (mensaje = formatter.format_help ())
Archivo "/usr/local/lib/python2.7/argparse.py", línea 2362, en la salida _sys.exit (estado) Salida del sistema: 0