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'])