| #!/usr/bin/python |
| # |
| # Copyright 2010 Google Inc. All Rights Reserved. |
| |
| __author__ = 'asharif@google.com (Ahmad Sharif)' |
| |
| import unittest |
| from automation.common import machine |
| from automation.server import machine_manager |
| |
| |
| class MachineManagerTest(unittest.TestCase): |
| |
| def setUp(self): |
| self.machine_manager = machine_manager.MachineManager() |
| |
| def testPrint(self): |
| print self.machine_manager |
| |
| def testGetLinuxBox(self): |
| mach_spec_list = [machine.MachineSpecification(os='linux')] |
| machines = self.machine_manager.GetMachines(mach_spec_list) |
| self.assertTrue(machines) |
| |
| def testGetChromeOSBox(self): |
| mach_spec_list = [machine.MachineSpecification(os='chromeos')] |
| machines = self.machine_manager.GetMachines(mach_spec_list) |
| self.assertTrue(machines) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |