# pylint: disable-msg=C0111
#!/usr/bin/python
#
# Copyright 2008 Google Inc. All Rights Reserved.
"""Test for atest."""
import unittest, os, sys, urllib2
import common
from autotest_lib.cli import cli_mock, topic_common, rpc
class topic_common_misc_tests(unittest.TestCase):
def test_get_item_key(self):
get_item_key = topic_common._get_item_key
self.assertRaises(ValueError, get_item_key, {}, '')
self.assertRaises(ValueError, get_item_key, {}, '.')
self.assertRaises(KeyError, get_item_key, {}, 'a')
self.assertRaises(KeyError, get_item_key, {}, 'a.')
self.assertRaises(ValueError, get_item_key, {'a': {}}, 'a.')
self.assertRaises(KeyError, get_item_key, {'a': {}}, 'a.b')
self.assertEquals(2, get_item_key({'a.b': 2, 'a': {}}, 'a.b'))
self.assertEquals(9, get_item_key({'a': {'b': 9}}, 'a.b'))
self.assertEquals(3, get_item_key({'a': {'b': {'c': 3}}}, 'a.b.c'))
self.assertEquals(5, get_item_key({'a': 5}, 'a'))
self.assertEquals({'b': 9}, get_item_key({'a': {'b': 9}}, 'a'))
class item_parse_info_unittest(cli_mock.cli_unittest):
def __test_parsing_flist_bad(self, options):
parse_info = topic_common.item_parse_info
test_parse_info = parse_info(attribute_name='testing',
filename_option='flist')
self.assertRaises(topic_common.CliError,
test_parse_info.get_values, options, [])
def __test_parsing_flist_good(self, options, expected):
parse_info = topic_common.item_parse_info
test_parse_info = parse_info(attribute_name='testing',
filename_option='flist')
result, leftover = test_parse_info.get_values(options, [])
self.assertEqualNoOrder(expected, result)
os.unlink(options.flist)
def __test_parsing_inline_good(self, options, expected):
parse_info = topic_common.item_parse_info
test_parse_info = parse_info(attribute_name='testing',
inline_option='inline')
result, leftover = test_parse_info.get_values(options, [])
self.assertEqualNoOrder(expected, result)
def __test_parsing_leftover_good(self, leftover, expected):
class opt(object):
pass
parse_info = topic_common.item_parse_info
test_parse_info = parse_info(attribute_name='testing',
inline_option='inline',
use_leftover=True)
result, leftover = test_parse_info.get_values(opt(), leftover)
self.assertEqualNoOrder(expected, result)
def __test_parsing_all_good(self, options, leftover, expected):
parse_info = topic_common.item_parse_info
test_parse_info = parse_info(attribute_name='testing',
inline_option='inline',
filename_option='flist',
use_leftover=True)
result, leftover = test_parse_info.get_values(options, leftover)
self.assertEqualNoOrder(expected, result)
os.unlink(options.flist)
def __test_parsing_all_bad(self, options, leftover):
parse_info = topic_common.item_parse_info
test_parse_info = parse_info(attribute_name='testing',
inline_option='inline',
filename_option='flist',
use_leftover=True)
self.assertRaises(topic_common.CliError,
test_parse_info.get_values, options, leftover)
def test_file_list_wrong_file(self):
class opt(object):
flist = './does_not_exist'
self.__test_parsing_flist_bad(opt())
def test_file_list_empty_file(self):
class opt(object):
flist_obj = cli_mock.create_file('')
flist = flist_obj.name
self.__test_parsing_flist_bad(opt())
def test_file_list_ok(self):
class opt(object):
flist_obj = cli_mock.create_file('a\nb\nc\n')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b', 'c'])
def test_file_list_one_line_space(self):
class opt(object):
flist_obj = cli_mock.create_file('a b c\nd e\nf\n')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b', 'c', 'd', 'e', 'f'])
def test_file_list_one_line_comma(self):
class opt(object):
flist_obj = cli_mock.create_file('a,b,c\nd,e\nf\n')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b', 'c', 'd', 'e', 'f'])
def test_file_list_one_line_mix(self):
class opt(object):
flist_obj = cli_mock.create_file('a,b c\nd,e\nf\ng h,i')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b', 'c', 'd', 'e',
'f', 'g', 'h', 'i'])
def test_file_list_one_line_comma_space(self):
class opt(object):
flist_obj = cli_mock.create_file('a, b c\nd,e\nf\ng h,i')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b', 'c', 'd', 'e',
'f', 'g', 'h', 'i'])
def test_file_list_line_end_comma_space(self):
class opt(object):
flist_obj = cli_mock.create_file('a, b c\nd,e, \nf,\ng h,i ,')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b', 'c', 'd', 'e',
'f', 'g', 'h', 'i'])
def test_file_list_no_eof(self):
class opt(object):
flist_obj = cli_mock.create_file('a\nb\nc')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b', 'c'])
def test_file_list_blank_line(self):
class opt(object):
flist_obj = cli_mock.create_file('\na\nb\n\nc\n')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b', 'c'])
def test_file_list_escaped_commas(self):
class opt(object):
flist_obj = cli_mock.create_file('a\nb\\,c\\,d\nef\\,g')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b,c,d', 'ef,g'])
def test_file_list_escaped_commas_slashes(self):
class opt(object):
flist_obj = cli_mock.create_file('a\nb\\\\\\,c\\,d\nef\\\\,g')
flist = flist_obj.name
self.__test_parsing_flist_good(opt(), ['a', 'b\\,c,d', 'ef\\', 'g'])
def test_file_list_opt_list_one(self):
class opt(object):
inline = 'a'
self.__test_parsing_inline_good(opt(), ['a'])
def test_file_list_opt_list_space(self):
class opt(object):
inline = 'a b c'
self.__test_parsing_inline_good(opt(), ['a', 'b', 'c'])
def test_file_list_opt_list_mix_space_comma(self):
class opt(object):
inline = 'a b,c,d e'
self.__test_parsing_inline_good(opt(), ['a', 'b', 'c', 'd', 'e'])
def test_file_list_opt_list_mix_comma_space(self):
class opt(object):
inline = 'a b,c, d e'
self.__test_parsing_inline_good(opt(), ['a', 'b', 'c', 'd', 'e'])
def test_file_list_opt_list_end_comma_space(self):
class opt(object):
inline = 'a b, ,c,, d e, '
self.__test_parsing_inline_good(opt(), ['a', 'b', 'c', 'd', 'e'])
def test_file_list_opt_list_escaped_commas(self):
class opt(object):
inline = 'a\\,b,c, d'
self.__test_parsing_inline_good(opt(), ['a,b', 'c', 'd'])
def test_file_list_opt_list_escaped_commas_slashes(self):
class opt(object):
inline = 'a\\,b\\\\\\,c,d,e'
self.__test_parsing_inline_good(opt(), ['a,b\\,c', 'd', 'e'])
def test_file_list_add_on_space(self):
self.__test_parsing_leftover_good(['a','c','b'],
['a', 'b', 'c'])
def test_file_list_add_on_mix_space_comma(self):
self.__test_parsing_leftover_good(['a', 'c','b,d'],
['a', 'b', 'c', 'd'])
def test_file_list_add_on_mix_comma_space(self):
self.__test_parsing_leftover_good(['a', 'c', 'b,', 'd'],
['a', 'b', 'c', 'd'])
def test_file_list_add_on_end_comma_space(self):
self.__test_parsing_leftover_good(['a', 'c', 'b,', 'd,', ','],
['a', 'b', 'c', 'd'])
def test_file_list_add_on_escaped_commas(self):
self.__test_parsing_leftover_good(['a', 'c', 'b,', 'd\\,e\\,f'],
['a', 'b', 'c', 'd,e,f'])
def test_file_list_add_on_escaped_commas_slashes(self):
self.__test_parsing_leftover_good(['a', 'c', 'b,', 'd\\\\\\,e,f'],
['a', 'b', 'c', 'd\\,e', 'f'])
def test_file_list_all_opt(self):
class opt(object):
flist_obj = cli_mock.create_file('f\ng\nh\n')
flist = flist_obj.name
inline = 'a b,c,d e'
self.__test_parsing_all_good(opt(), ['i', 'j'],
['a', 'b', 'c', 'd', 'e',
'f', 'g', 'h', 'i', 'j'])
def test_file_list_all_opt_empty_file(self):
class opt(object):
flist_obj = cli_mock.create_file('')
flist = flist_obj.name
inline = 'a b,c,d e'
self.__test_parsing_all_bad(opt(), ['i', 'j'])
def test_file_list_all_opt_in_common(self):
class opt(object):
flist_obj = cli_mock.create_file('f\nc\na\n')
flist = flist_obj.name
inline = 'a b,c,d e'
self.__test_parsing_all_good(opt(), ['i','j,d'],
['a', 'b', 'c', 'd', 'e', 'f', 'i', 'j'])
def test_file_list_all_opt_in_common_space(self):
class opt(object):
flist_obj = cli_mock.create_file('a b c\nd,e\nf\ng')
flist = flist_obj.name
inline = 'a b,c,d h'
self.__test_parsing_all_good(opt(), ['i','j,d'],
['a', 'b', 'c', 'd', 'e',
'f', 'g', 'h', 'i', 'j'])
def test_file_list_all_opt_in_common_weird(self):
class opt(object):
flist_obj = cli_mock.create_file('a b c\nd,e\nf\ng, \n, ,,')
flist = flist_obj.name
inline = 'a b,c,d h, , ,, '
self.__test_parsing_all_good(opt(), ['i','j,d'],
['a', 'b', 'c', 'd', 'e',
'f', 'g', 'h', 'i', 'j'])
def test_file_list_all_opt_in_common_escaped_commas(self):
class opt(object):
flist_obj = cli_mock.create_file('a\\,b\\,c\nd,e\nf\ng')
flist = flist_obj.name
inline = 'a\\,b\\,c,d h'
self.__test_parsing_all_good(opt(), ['i','j,d'],
['a,b,c', 'd', 'e', 'f', 'g', 'h',
'i', 'j'])
def test_file_list_all_opt_in_common_escaped_commas_slashes(self):
class opt(object):
flist_obj = cli_mock.create_file('a\\,b\\\\\\,c\nd,e\nf,ghi, ,, j,')
flist = flist_obj.name
inline = 'a\\,b\\\\\\,c,d h,ijk'
self.__test_parsing_all_good(opt(), ['i','j,d'],
['a,b\\,c', 'd', 'e', 'f', 'ghi', 'h',
'i', 'j', 'ijk'])
class atest_unittest(cli_mock.cli_unittest):
def setUp(self):
super(atest_unittest, self).setUp()
self.atest = topic_common.atest()
self.atest.afe = rpc.afe_comm()
if 'AUTOTEST_WEB' in os.environ:
del os.environ['AUTOTEST_WEB']
def tearDown(self):
self.atest = None
super(atest_unittest, self).tearDown()
def test_invalid_arg_kill(self):
self.atest.kill_on_failure = True
self.god.mock_io()
sys.exit.expect_call(1).and_raises(cli_mock.ExitException)
self.assertRaises(cli_mock.ExitException,
self.atest.invalid_arg, 'This is bad')
(output, err) = self.god.unmock_io()
self.god.check_playback()
self.assert_(err.find('This is bad') >= 0)
def test_invalid_arg_continue(self):
self.god.mock_io()
self.atest.invalid_arg('This is sort of ok')
(output, err) = self.god.unmock_io()
self.assert_(err.find('This is sort of ok') >= 0)
def test_failure_continue(self):
self.atest.failure('This is partly bad', item='item0',
what_failed='something important')
err = self.atest.failed['something important']
self.assert_('This is partly bad' in err.keys())
def test_failure_continue_multiple_different_errors(self):
self.atest.failure('This is partly bad', item='item0',
what_failed='something important')
self.atest.failure('This is really bad', item='item0',
what_failed='something really important')
err = self.atest.failed['something important']
self.assert_('This is partly bad' in err)
self.assert_('This is really bad' not in err)
err = self.atest.failed['something really important']
self.assert_('This is partly bad' not in err)
self.assert_('This is really bad' in err)
def test_failure_continue_multiple_same_errors(self):
self.atest.failure('This is partly bad', item='item0',
what_failed='something important')
self.atest.failure('This is really bad', item='item1',
what_failed='something important')
errs = self.atest.failed['something important']
self.assert_('This is partly bad' in errs)
self.assert_('This is really bad' in errs)
self.assert_(set(['item0']) in errs.values())
self.assert_(set(['item1']) in errs.values())
def test_failure_continue_multiple_errors_mixed(self):
self.atest.failure('This is partly bad', item='item0',
what_failed='something important')
self.atest.failure('This is really bad', item='item0',
what_failed='something really important')
self.atest.failure('This is really bad', item='item1',
what_failed='something important')
errs = self.atest.failed['something important']
self.assert_('This is partly bad' in errs)
self.assert_('This is really bad' in errs)
self.assert_(set(['item0']) in errs.values())
self.assert_(set(['item1']) in errs.values())
errs = self.atest.failed['something really important']
self.assert_('This is really bad' in errs)
self.assert_('This is partly bad' not in errs)
self.assert_(set(['item0']) in errs.values())
self.assert_(set(['item1']) not in errs.values())
def test_failure_continue_multiple_errors_mixed_same_error(self):
self.atest.failure('This is partly bad', item='item0',
what_failed='something important')
self.atest.failure('This is really bad', item='item0',
what_failed='something really important')
self.atest.failure('This is partly bad', item='item1',
what_failed='something important')
errs = self.atest.failed['something important']
self.assert_('This is partly bad' in errs)
self.assert_('This is really bad' not in errs)
self.assert_(set(['item0', 'item1']) in errs.values())
errs = self.atest.failed['something really important']
self.assert_('This is really bad' in errs)
self.assert_('This is partly bad' not in errs)
self.assert_(set(['item0']) in errs.values())
self.assert_(set(['item1']) not in errs.values())
def test_failure_exit(self):
self.atest.kill_on_failure = True
self.god.mock_io()
sys.exit.expect_call(1).and_raises(cli_mock.ExitException)
self.assertRaises(cli_mock.ExitException,
self.atest.failure, 'This is partly bad')
(output, err) = self.god.unmock_io()
self.god.check_playback()
self.assert_(err.find('This is partly bad') >= 0)
def test_failure_exit_item(self):
self.atest.kill_on_failure = True
self.god.mock_io()
sys.exit.expect_call(1).and_raises(cli_mock.ExitException)
self.assertRaises(cli_mock.ExitException,
self.atest.failure, 'This is partly bad',
item='item0')
(output, err) = self.god.unmock_io()
self.god.check_playback()
self.assertWords(err, ['This is partly bad'], ['item0'])
def test_show_all_failures_common(self):
self.atest.failure('This is partly bad', item='item0',
what_failed='something important')
self.atest.failure('This is partly bad', item='item1',
what_failed='something important')
self.god.mock_io()
self.atest.show_all_failures()
(output, err) = self.god.unmock_io()
self.assertWords(err, ['something important',
'This is partly bad', 'item0', 'item1'])
def test_parse_add_on(self):
flist = cli_mock.create_file('host1\nhost2\nleft2')
sys.argv = ['atest', '--web', 'fooweb', '--parse',
'--kill-on-failure', 'left1', 'left2', '-M', flist.name]
self.atest.parser.add_option('-M', '--mlist', type='string')
item_info = topic_common.item_parse_info(attribute_name='hosts',
filename_option='mlist',
use_leftover=True)
(options, leftover) = self.atest.parse([item_info])
self.assertEqualNoOrder(self.atest.hosts,
['left1', 'left2', 'host1', 'host2'])
self.assertEqual({'mlist': flist.name,
'web_server': 'fooweb',
'parse': True,
'parse_delim': '|',
'kill_on_failure': True,
'verbose': False,
'no_confirmation': False,
'debug': False}, options)
self.assertEqual(leftover, [])
flist.clean()
def test_parse_no_add_on(self):
flist = cli_mock.create_file('host1\nhost2\nleft2')
sys.argv = ['atest', '--web', 'fooweb', '--parse', '-g',
'--kill-on-failure', 'left1', 'left2', '-M', flist.name]
self.atest.parser.add_option('-M', '--mlist', type='string')
item_info = topic_common.item_parse_info(attribute_name='hosts',
filename_option='mlist')
(options, leftover) = self.atest.parse([item_info])
self.assertEqualNoOrder(self.atest.hosts,
['left2', 'host1', 'host2'])
self.assertEqual({'mlist': flist.name,
'web_server': 'fooweb',
'parse': True,
'parse_delim': '|',
'kill_on_failure': True,
'verbose': False,
'no_confirmation': False,
'debug': True}, options)
self.assertEqual(leftover, ['left1', 'left2'])
flist.clean()
def test_parse_add_on_first(self):
flist = cli_mock.create_file('host1\nhost2\nleft2')
ulist = cli_mock.create_file('user1\nuser2\nuser3\n')
sys.argv = ['atest', '-g', '--parse', '--ulist', ulist.name,
'-u', 'myuser,youruser',
'--kill-on-failure', 'left1', 'left2', '-M', flist.name]
self.atest.parser.add_option('-M', '--mlist', type='string')
self.atest.parser.add_option('-U', '--ulist', type='string')
self.atest.parser.add_option('-u', '--user', type='string')
host_info = topic_common.item_parse_info(attribute_name='hosts',
filename_option='mlist',
use_leftover=True)
user_info = topic_common.item_parse_info(attribute_name='users',
inline_option='user',
filename_option='ulist')
(options, leftover) = self.atest.parse([host_info, user_info])
self.assertEqualNoOrder(self.atest.hosts,
['left1', 'left2', 'host1', 'host2'])
self.assertEqualNoOrder(self.atest.users,
['user1', 'user2', 'user3',
'myuser', 'youruser'])
self.assertEqual({'mlist': flist.name,
'ulist': ulist.name,
'user': 'myuser,youruser',
'web_server': None,
'parse': True,
'parse_delim': '|',
'kill_on_failure': True,
'verbose': False,
'no_confirmation': False,
'debug': True}, options)
self.assertEqual(leftover, [])
flist.clean()
ulist.clean()
def test_parse_add_on_second(self):
flist = cli_mock.create_file('host1\nhost2\nleft2')
ulist = cli_mock.create_file('user1\nuser2\nuser3\n')
sys.argv = ['atest', '-g', '--parse', '-U', ulist.name,
'-u', 'myuser,youruser',
'--kill-on-failure', 'left1', 'left2', '-M', flist.name]
self.atest.parser.add_option('-M', '--mlist', type='string')
self.atest.parser.add_option('-U', '--ulist', type='string')
self.atest.parser.add_option('-u', '--user', type='string')
host_info = topic_common.item_parse_info(attribute_name='hosts',
filename_option='mlist',
use_leftover=True)
user_info = topic_common.item_parse_info(attribute_name='users',
inline_option='user',
filename_option='ulist')
(options, leftover) = self.atest.parse([host_info, user_info])
self.assertEqualNoOrder(self.atest.hosts,
['left1', 'left2', 'host1', 'host2'])
self.assertEqualNoOrder(self.atest.users,
['user1', 'user2', 'user3',
'myuser', 'youruser'])
self.assertEqual({'mlist': flist.name,
'ulist': ulist.name,
'user': 'myuser,youruser',
'web_server': None,
'parse': True,
'parse_delim': '|',
'kill_on_failure': True,
'verbose': False,
'no_confirmation': False,
'debug': True}, options)
self.assertEqual(leftover, [])
flist.clean()
ulist.clean()
def test_parse_all_opts(self):
flist = cli_mock.create_file('host1\nhost2\nleft2')
ulist = cli_mock.create_file('user1\nuser2\nuser3\n')
sys.argv = ['atest', '-g', '--parse', '--ulist', ulist.name,
'-u', 'myuser,youruser',
'--kill-on-failure', '-M', flist.name, 'left1', 'left2']
self.atest.parser.add_option('-M', '--mlist', type='string')
self.atest.parser.add_option('-U', '--ulist', type='string')
self.atest.parser.add_option('-u', '--user', type='string')
host_info = topic_common.item_parse_info(attribute_name='hosts',
filename_option='mlist',
use_leftover=True)
user_info = topic_common.item_parse_info(attribute_name='users',
inline_option='user',
filename_option='ulist')
(options, leftover) = self.atest.parse([host_info, user_info])
self.assertEqualNoOrder(self.atest.hosts,
['left1', 'left2', 'host1', 'host2'])
self.assertEqualNoOrder(self.atest.users,
['user1', 'user2', 'user3',
'myuser', 'youruser'])
self.assertEqual({'mlist': flist.name,
'ulist': ulist.name,
'user': 'myuser,youruser',
'web_server': None,
'parse': True,
'parse_delim': '|',
'kill_on_failure': True,
'verbose': False,
'no_confirmation': False,
'debug': True}, options)
self.assertEqual(leftover, [])
flist.clean()
ulist.clean()
def test_parse_no_add_on_2(self):
flist = cli_mock.create_file('host1\nhost2\nleft2')
ulist = cli_mock.create_file('user1\nuser2\nuser3\n')
sys.argv = ['atest', '-U', ulist.name,
'--kill-on-failure', '-M', flist.name]
self.atest.parser.add_option('-M', '--mlist', type='string')
self.atest.parser.add_option('-U', '--ulist', type='string')
self.atest.parser.add_option('-u', '--user', type='string')
host_info = topic_common.item_parse_info(attribute_name='hosts',
filename_option='mlist',
use_leftover=True)
user_info = topic_common.item_parse_info(attribute_name='users',
inline_option='user',
filename_option='ulist')
(options, leftover) = self.atest.parse([host_info, user_info])
self.assertEqualNoOrder(self.atest.hosts,
['left2', 'host1', 'host2'])
self.assertEqualNoOrder(self.atest.users,
['user1', 'user2', 'user3'])
self.assertEqual({'mlist': flist.name,
'ulist': ulist.name,
'user': None,
'web_server': None,
'parse': False,
'parse_delim': '|',
'kill_on_failure': True,
'verbose': False,
'no_confirmation': False,
'debug': False}, options)
self.assertEqual(leftover, [])
flist.clean()
ulist.clean()
def test_parse_no_flist_add_on(self):
sys.argv = ['atest', '-g', '--parse', '-u', 'myuser,youruser',
'--kill-on-failure', 'left1', 'left2']
self.atest.parser.add_option('-M', '--mlist', type='string')
self.atest.parser.add_option('-U', '--ulist', type='string')
self.atest.parser.add_option('-u', '--user', type='string')
host_info = topic_common.item_parse_info(attribute_name='hosts',
use_leftover=True)
user_info = topic_common.item_parse_info(attribute_name='users',
inline_option='user')
(options, leftover) = self.atest.parse([host_info, user_info])
self.assertEqualNoOrder(self.atest.hosts,
['left1', 'left2'])
self.assertEqualNoOrder(self.atest.users,
['myuser', 'youruser'])
self.assertEqual({'mlist': None,
'ulist': None,
'user': 'myuser,youruser',
'web_server': None,
'parse': True,
'parse_delim': '|',
'kill_on_failure': True,
'verbose': False,
'no_confirmation': False,
'debug': True}, options)
self.assertEqual(leftover, [])
def test_parse_no_flist_no_add_on(self):
sys.argv = ['atest', '-u', 'myuser,youruser', '--kill-on-failure',
'-a', 'acl1,acl2']
self.atest.parser.add_option('-u', '--user', type='string')
self.atest.parser.add_option('-a', '--acl', type='string')
acl_info = topic_common.item_parse_info(attribute_name='acls',
inline_option='acl')
user_info = topic_common.item_parse_info(attribute_name='users',
inline_option='user')
(options, leftover) = self.atest.parse([user_info, acl_info])
self.assertEqualNoOrder(self.atest.acls,
['acl1', 'acl2'])
self.assertEqualNoOrder(self.atest.users,
['myuser', 'youruser'])
self.assertEqual({'user': 'myuser,youruser',
'acl': 'acl1,acl2',
'web_server': None,
'parse': False,
'parse_delim': '|',
'kill_on_failure': True,
'verbose': False,
'no_confirmation': False,
'debug': False}, options)
self.assertEqual(leftover, [])
def test_parse_req_items_ok(self):
sys.argv = ['atest', '-u', 'myuser,youruser']
self.atest.parser.add_option('-u', '--user', type='string')
user_info = topic_common.item_parse_info(attribute_name='users',
inline_option='user')
(options, leftover) = self.atest.parse([user_info],
req_items='users')
self.assertEqualNoOrder(self.atest.users,
['myuser', 'youruser'])
self.assertEqual({'user': 'myuser,youruser',
'web_server': None,
'parse': False,
'parse_delim': '|',
'kill_on_failure': False,
'verbose': False,
'no_confirmation': False,
'debug': False}, options)
self.assertEqual(leftover, [])
def test_parse_req_items_missing(self):
sys.argv = ['atest', '-u', 'myuser,youruser', '--kill-on-failure']
self.atest.parser.add_option('-u', '--user', type='string')
acl_info = topic_common.item_parse_info(attribute_name='acls',
inline_option='acl')
user_info = topic_common.item_parse_info(attribute_name='users',
inline_option='user')
self.god.mock_io()
sys.exit.expect_call(1).and_raises(cli_mock.ExitException)
self.assertRaises(cli_mock.ExitException,
self.atest.parse,
[user_info, acl_info],
'acls')
self.assertEqualNoOrder(self.atest.users,
['myuser', 'youruser'])
self.assertEqualNoOrder(self.atest.acls, [])
self.god.check_playback()
self.god.unmock_io()
def test_parse_bad_option(self):
sys.argv = ['atest', '--unknown']
self.god.stub_function(self.atest.parser, 'error')
self.atest.parser.error.expect_call('no such option: --unknown').and_return(None)
self.atest.parse()
self.god.check_playback()
def test_parse_all_set(self):
sys.argv = ['atest', '--web', 'fooweb', '--parse', '--debug',
'--kill-on-failure', '--verbose', 'left1', 'left2',
'--parse-delim', '?']
(options, leftover) = self.atest.parse()
self.assertEqual({'web_server': 'fooweb',
'parse': True,
'parse_delim': '?',
'kill_on_failure': True,
'verbose': True,
'no_confirmation': False,
'debug': True}, options)
self.assertEqual(leftover, ['left1', 'left2'])
def test_execute_rpc_bad_server(self):
self.atest.afe = rpc.afe_comm('http://does_not_exist')
self.god.mock_io()
rpc.afe_comm.run.expect_call('myop').and_raises(urllib2.URLError("<urlopen error (-2, 'Name or service not known')>"))
sys.exit.expect_call(1).and_raises(cli_mock.ExitException)
self.assertRaises(cli_mock.ExitException,
self.atest.execute_rpc, 'myop')
(output, err) = self.god.unmock_io()
self.god.check_playback()
self.assert_(err.find('http://does_not_exist') >= 0)
#
# Print Unit tests
#
def __test_print_fields(self, func, expected, **dargs):
if not dargs.has_key('items'):
dargs['items']=[{'hostname': 'h0',
'platform': 'p0',
'labels': [u'l0', u'l1'],
'locked': 1,
'id': 'id0',
'name': 'name0'},
{'hostname': 'h1',
'platform': 'p1',
'labels': [u'l2', u'l3'],
'locked': 0,
'id': 'id1',
'name': 'name1'}]
self.god.mock_io()
func(**dargs)
(output, err) = self.god.unmock_io()
self.assertEqual(expected, output)
#
# Print fields Standard
#
def __test_print_fields_std(self, keys, expected):
self.__test_print_fields(self.atest.print_fields_std,
expected, keys=keys)
def test_print_fields_std_one_str(self):
self.__test_print_fields_std(['hostname'],
'Host: h0\n'
'Host: h1\n')
def test_print_fields_std_conv_bool(self):
"""Make sure the conversion functions are called"""
self.__test_print_fields_std(['locked'],
'Locked: True\n'
'Locked: False\n')
def test_print_fields_std_conv_label(self):
"""Make sure the conversion functions are called"""
self.__test_print_fields_std(['labels'],
'Labels: l0, l1\n'
'Labels: l2, l3\n')
def test_print_fields_std_all_fields(self):
"""Make sure the conversion functions are called"""
self.__test_print_fields_std(['hostname', 'platform','locked'],
'Host: h0\n'
'Platform: p0\n'
'Locked: True\n'
'Host: h1\n'
'Platform: p1\n'
'Locked: False\n')
#
# Print fields parse
#
def __test_print_fields_parse(self, keys, expected):
self.__test_print_fields(self.atest.print_fields_parse,
expected, keys=keys)
def test_print_fields_parse_one_str(self):
self.__test_print_fields_parse(['hostname'],
'Host=h0\n'
'Host=h1\n')
def test_print_fields_parse_conv_bool(self):
self.__test_print_fields_parse(['locked'],
'Locked=True\n'
'Locked=False\n')
def test_print_fields_parse_conv_label(self):
self.__test_print_fields_parse(['labels'],
'Labels=l0, l1\n'
'Labels=l2, l3\n')
def test_print_fields_parse_all_fields(self):
self.__test_print_fields_parse(['hostname', 'platform', 'locked'],
'Host=h0|Platform=p0|'
'Locked=True\n'
'Host=h1|Platform=p1|'
'Locked=False\n')
#
# Print table standard
#
def __test_print_table_std(self, keys, expected):
self.__test_print_fields(self.atest.print_table_std,
expected, keys_header=keys)
def test_print_table_std_all_fields(self):
self.__test_print_table_std(['hostname', 'platform','locked'],
'Host Platform Locked\n'
'h0 p0 True\n'
'h1 p1 False\n')
# TODO JME - add long fields tests
#
# Print table parse
#
def __test_print_table_parse(self, keys, expected):
self.__test_print_fields(self.atest.print_table_parse,
expected, keys_header=keys)
def test_print_table_parse_all_fields(self):
self.__test_print_table_parse(['hostname', 'platform',
'locked'],
'Host=h0|Platform=p0|Locked=True\n'
'Host=h1|Platform=p1|Locked=False\n')
def test_print_table_parse_all_fields_2(self):
self.atest.parse_delim = '?'
self.__test_print_table_parse(['hostname', 'platform',
'locked'],
'Host=h0?Platform=p0?Locked=True\n'
'Host=h1?Platform=p1?Locked=False\n')
def test_print_table_parse_empty_fields(self):
self.__test_print_fields(self.atest.print_table_parse,
'Host=h0|Platform=p0\n'
'Host=h1|Platform=p1|Labels=l2, l3\n',
items=[{'hostname': 'h0',
'platform': 'p0',
'labels': [],
'locked': 1,
'id': 'id0',
'name': 'name0'},
{'hostname': 'h1',
'platform': 'p1',
'labels': [u'l2', u'l3'],
'locked': 0,
'id': 'id1',
'name': 'name1'}],
keys_header=['hostname', 'platform',
'labels'])
#
# Print mix table standard
#
def __test_print_mix_table_std(self, keys_header, sublist_keys,
expected):
self.__test_print_fields(self.atest.print_table_std,
expected,
keys_header=keys_header,
sublist_keys=sublist_keys)
def test_print_mix_table(self):
self.__test_print_mix_table_std(['name', 'hostname'], [],
'Name Host\n'
'name0 h0\n'
'name1 h1\n')
def test_print_mix_table_sublist(self):
self.__test_print_mix_table_std(['name', 'hostname'], ['labels'],
'Name Host\n'
'name0 h0\n'
'Labels: \n'
'\tl0, l1\n\n\n'
'name1 h1\n'
'Labels: \n'
'\tl2, l3\n\n\n')
#
# Print by ID standard
#
def __test_print_by_ids_std(self, expected):
self.__test_print_fields(self.atest.print_by_ids_std,
expected)
def test_print_by_ids_std_all_fields(self):
self.__test_print_by_ids_std('Id Name\n'
'id0 name0\n'
'id1 name1\n')
#
# Print by ID parse
#
def __test_print_by_ids_parse(self, expected):
self.__test_print_fields(self.atest.print_by_ids_parse,
expected)
def test_print_by_ids_parse_all_fields(self):
self.__test_print_by_ids_parse('Id=id0|Name=name0|'
'Id=id1|Name=name1\n')
if __name__ == '__main__':
unittest.main()