def Run(): """This must be called from __main__ modules main, instead of app.run(). app.run will base its actions on its stacktrace. Returns: app.run() """ app.parse_flags_with_usage = ParseFlagsWithUsage app.really_start = _CommandsStart app.usage = _ReplacementAppUsage return app.run()
def Run(): """This must be called from __main__ modules main, instead of app.run(). app.run will base its actions on its stacktrace. Returns: app.run() """ app.parse_flags_with_usage = ParseFlagsWithUsage original_really_start = app.really_start def InterceptReallyStart(): original_really_start(main=_CommandsStart) app.really_start = InterceptReallyStart app.usage = _ReplacementAppUsage return app.run()
else: raise def RunCommandLine(titan_commands, parser): """Runs a command based on args from the command line. Args: titan_commands: A TitanCommands object. parser: A ArgumentParser object. Returns: The results of the corresponding command. """ args, flags = parser.ParseArgs() if args: command = args[0] args = args[1:] return titan_commands.RunCommand(command, args=args, flags=flags) else: return titan_commands.RunCommand('help') class CommandValueError(ValueError): pass def main(unused_argv): parser = GetDefaultParser() titan_commands = TitanCommands() RunCommandLine(titan_commands, parser) if __name__ == '__main__': app.run()
self.stubs.SmartUnsetAll() def testGenerateShardNames(self): self.assertEqual([], utils.generate_shard_names(0)) self.assertEqual(['ab'], utils.generate_shard_names(1)) self.assertEqual(['abc', 'acb'], utils.generate_shard_names(2)) self.assertEqual(['abc', 'acb', 'bac'], utils.generate_shard_names(3)) # Verify all the shards are unique. tags = self.assertEqual(60, len(set(utils.generate_shard_names(60)))) # Weakly verify that larger sharding works. tags = utils.generate_shard_names(10000) self.assertEqual(10000, len(tags)) self.assertEqual(8, len(tags[0])) def testHumanizeDuration(self): self.assertEqual('10s', utils.humanize_duration(10)) def testHumanizeTimeDelta(self): self.assertEqual('1d', utils.humanize_time_delta(datetime.timedelta(days=1))) def main(unused_argv): basetest.main() if __name__ == '__main__': app.run()