base_utils_unittest.py revision 0afbb6369aa5aa9a75ea67dd9e95ec4b21c0c181 
1#!/usr/bin/python2.4
2
3__author__ = 'raphtee@google.com (Travis Miller)'
4
5import unittest
6import common
7from autotest_lib.server import utils
8
9
10class UtilsTest(unittest.TestCase):
11
12    def setUp(self):
13        # define out machines here
14        self.machines = ['mach1', 'mach2', 'mach3', 'mach4', 'mach5',
15                                'mach6', 'mach7']
16
17        self.ntuples = [['mach1', 'mach2'], ['mach3', 'mach4'],
18                        ['mach5', 'mach6']]
19        self.failures = []
20        self.failures.append(('mach7', "machine can not be tupled"))
21
22
23    def test_form_cell_mappings(self):
24        (ntuples, failures) = utils.form_ntuples_from_machines(self.machines)
25        self.assertEquals(self.ntuples, ntuples)
26        self.assertEquals(self.failures, failures)
27
28
29if __name__ == "__main__":
30    unittest.main()
31