blob: 23338637ed00cc6722964c5936f65047a23de7c0 [file] [log] [blame] [edit]
#!/usr/bin/python
#
# Copyright 2008 Google Inc. All Rights Reserved.
"""Test for atest."""
import unittest
import common
from autotest_lib.cli import cli_mock
ATEST_USAGE_STRING = ('atest [acl|host|job|label|shard|test|user|'
'server] [action] [options]')
class main_unittest(cli_mock.cli_unittest):
"""Unittest for atest command.
"""
def _test_help(self, argv, out_words_ok, err_words_ok):
"""Test help output.
@param argv: A list of argument.
@param out_words_ok: Expected output.
@param err_words_ok: Expected output when input arguments are invalid.
"""
saved_outputs = None
for help in ['-h', '--help', 'help']:
outputs = self.run_cmd(argv + [help], exit_code=0,
out_words_ok=out_words_ok,
err_words_ok=err_words_ok)
if not saved_outputs:
saved_outputs = outputs
else:
self.assertEqual(outputs, saved_outputs)
def test_main_help(self):
"""Main help level"""
self._test_help(argv=['atest'],
out_words_ok=[ATEST_USAGE_STRING],
err_words_ok=[])
def test_main_help_topic(self):
"""Topic level help"""
self._test_help(argv=['atest', 'host'],
out_words_ok=['atest host ',
'[create|delete|list|stat|mod|jobs|'
'rename|migrate] [options]'],
err_words_ok=[])
def test_main_help_action(self):
"""Action level help"""
self._test_help(argv=['atest:', 'host', 'mod'],
out_words_ok=['atest host mod [options]'],
err_words_ok=[])
def test_main_no_topic(self):
"""Test output when no topic is specified."""
self.run_cmd(['atest'], exit_code=1,
out_words_ok=[ATEST_USAGE_STRING],
err_words_ok=['No topic argument'])
def test_main_bad_topic(self):
"""Test output when an invalid topic is specified."""
self.run_cmd(['atest', 'bad_topic'], exit_code=1,
out_words_ok=[ATEST_USAGE_STRING],
err_words_ok=['Invalid topic bad_topic\n'])
def test_main_bad_action(self):
"""Test output when an invalid action is specified."""
self.run_cmd(['atest', 'host', 'bad_action'], exit_code=1,
out_words_ok=['atest host [create|delete|list|stat|'
'mod|jobs|rename|migrate] [options]'],
err_words_ok=['Invalid action bad_action'])
if __name__ == '__main__':
unittest.main()