Beispiel #1
0
    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'])
Beispiel #2
0
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'])