aboutsummaryrefslogtreecommitdiff
path: root/automation/server/machine_manager_test.py
blob: 2fa5bb4b8e782b8e537b98042278ab01b9f548da (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
#!/usr/bin/python2
#
# 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()