def main(argv):
parser = argparse.ArgumentParser(
prog="fetch_browser",
description=sys.modules['__main__'].__doc__,
formatter_class=argparse.RawTextHelpFormatter)
fetch_browser.add_argparse_options(parser)
infra_libs.logs.add_argparse_options(parser)
args = parser.parse_args(argv)
infra_libs.logs.process_argparse_options(args)
# Do more processing here
LOGGER.info('Fetch_browser starting.')
browser = args.browser[0]
cache_dir = os.path.abspath(os.path.expanduser(args.cache_dir))
result = fetch_browser.run(browser, cache_dir, args.platform, args.version)
if args.output_json:
with open(args.output_json, 'w') as f:
json.dump(result, f)
def test_arguments(self): parser = argparse.ArgumentParser() fetch_browser.add_argparse_options(parser) args = parser.parse_args(['firefox', '--output-json', 'some_path']) self.assertEqual(args.output_json, 'some_path') self.assertEqual(args.browser, ['firefox'])
