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()
