76 lines
3.0 KiB
Python
Executable File
76 lines
3.0 KiB
Python
Executable File
#!/usr/bin/python2
|
|
|
|
__author__ = 'raphtee@google.com (Travis Miller)'
|
|
|
|
import unittest
|
|
import common
|
|
from autotest_lib.server import utils
|
|
|
|
|
|
class UtilsTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
# define out machines here
|
|
self.machines = ['mach1', 'mach2', 'mach3', 'mach4', 'mach5',
|
|
'mach6', 'mach7']
|
|
|
|
self.ntuples = [['mach1', 'mach2'], ['mach3', 'mach4'],
|
|
['mach5', 'mach6']]
|
|
self.failures = []
|
|
self.failures.append(('mach7', "machine can not be tupled"))
|
|
|
|
|
|
def test_form_cell_mappings(self):
|
|
(ntuples, failures) = utils.form_ntuples_from_machines(self.machines)
|
|
self.assertEquals(self.ntuples, ntuples)
|
|
self.assertEquals(self.failures, failures)
|
|
|
|
|
|
# parse_machine() test cases
|
|
def test_parse_machine_good(self):
|
|
'''test that parse_machine() is outputting the correct data'''
|
|
gooddata = (('host', ('host', 'root', '', 22)),
|
|
('host:21', ('host', 'root', '', 21)),
|
|
('user@host', ('host', 'user', '', 22)),
|
|
('user:pass@host', ('host', 'user', 'pass', 22)),
|
|
('user:pass@host:1234', ('host', 'user', 'pass', 1234)),
|
|
|
|
('user:pass@10.3.2.1',
|
|
('10.3.2.1', 'user', 'pass', 22)),
|
|
('user:pass@10.3.2.1:1234',
|
|
('10.3.2.1', 'user', 'pass', 1234)),
|
|
|
|
('::1', ('::1', 'root', '', 22)),
|
|
('user:pass@abdc::ef', ('abdc::ef', 'user', 'pass', 22)),
|
|
('abdc::ef:99', ('abdc::ef:99', 'root', '', 22)),
|
|
('user:pass@[abdc::ef:99]',
|
|
('abdc::ef:99', 'user', 'pass', 22)),
|
|
('user:pass@[abdc::ef]:1234',
|
|
('abdc::ef', 'user', 'pass', 1234)),
|
|
)
|
|
for machine, result in gooddata:
|
|
self.assertEquals(utils.parse_machine(machine), result)
|
|
|
|
|
|
def test_parse_machine_override(self):
|
|
'''Test that parse_machine() defaults can be overridden'''
|
|
self.assertEquals(utils.parse_machine('host', 'bob', 'foo', 1234),
|
|
('host', 'bob', 'foo', 1234))
|
|
|
|
|
|
def test_parse_machine_bad(self):
|
|
'''test that bad data passed to parse_machine() will raise an exception'''
|
|
baddata = ((':22', IndexError), # neglect to pass a hostname #1
|
|
('user@', IndexError), # neglect to pass a hostname #2
|
|
('user@:22', IndexError), # neglect to pass a hostname #3
|
|
(':pass@host', ValueError), # neglect to pass a username
|
|
('host:', ValueError), # empty port after hostname
|
|
('[1::2]:', ValueError), # empty port after IPv6
|
|
)
|
|
for machine, exception in baddata:
|
|
self.assertRaises(exception, utils.parse_machine, machine)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|