def test_arguments(self):
parser = argparse.ArgumentParser()
cache_updater.add_argparse_options(parser)
args = cache_updater.parse_args(
parser, ['--shard-total', '10', '--shard-index', '5'])
self.assertEqual(args.shard_total, 10)
self.assertEqual(args.shard_index, 5)
self.assertRaises(SystemExit, cache_updater.parse_args, parser,
['--shard-total', '5', '--shard-index', '10'])
self.assertRaises(SystemExit, cache_updater.parse_args, parser,
['--shard-total', '-5', '--shard-index', '10'])
self.assertRaises(SystemExit, cache_updater.parse_args, parser,
['--shard-total', '5', '--shard-index', '-10'])
def main(argv):
parser = argparse.ArgumentParser(
prog="cache_updater",
description=sys.modules['__main__'].__doc__,
formatter_class=argparse.RawTextHelpFormatter)
cache_updater.add_argparse_options(parser)
infra_libs.logs.add_argparse_options(parser)
args = cache_updater.parse_args(parser, argv)
infra_libs.logs.process_argparse_options(args)
# Do more processing here
LOGGER.info('Cache_updater starting.')
cache_updater.run(args.cache_dir, args.shard_index, args.shard_total)
def test_arguments(self):
parser = argparse.ArgumentParser()
cache_updater.add_argparse_options(parser)
args = cache_updater.parse_args(
parser, ['--shard-total', '10', '--shard-index', '5'])
self.assertEqual(args.shard_total, 10)
self.assertEqual(args.shard_index, 5)
self.assertRaises(
SystemExit, cache_updater.parse_args, parser,
['--shard-total', '5', '--shard-index', '10'])
self.assertRaises(
SystemExit, cache_updater.parse_args, parser,
['--shard-total', '-5', '--shard-index', '10'])
self.assertRaises(
SystemExit, cache_updater.parse_args, parser,
['--shard-total', '5', '--shard-index', '-10'])