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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#!/usr/bin/python
# Copyright 2012 Google Inc. All Rights Reserved.
"""Unittest for machine_manager."""
import unittest
import label
import machine_manager
class MyMachineManager(machine_manager.MachineManager):
def __init__(self, chromeos_root):
super(MyMachineManager, self).__init__(chromeos_root, 0, "average")
def _TryToLockMachine(self, cros_machine):
self._machines.append(cros_machine)
cros_machine.checksum = ""
def AddMachine(self, machine_name):
with self._lock:
for m in self._all_machines:
assert m.name != machine_name, "Tried to double-add %s" % machine_name
cm = machine_manager.MockCrosMachine(machine_name, self.chromeos_root,
"average")
assert cm.machine_checksum, ("Could not find checksum for machine %s" %
machine_name)
self._all_machines.append(cm)
CHROMEOS_ROOT = "/tmp/chromeos-root"
MACHINE_NAMES = ["lumpy1", "lumpy2", "lumpy3", "daisy1", "daisy2"]
LABEL_LUMPY = label.MockLabel("lumpy", "image", CHROMEOS_ROOT, "lumpy",
["lumpy1", "lumpy2", "lumpy3", "lumpy4"],
"", "", False, "")
LABEL_MIX = label.MockLabel("mix", "image", CHROMEOS_ROOT, "mix",
["daisy1", "daisy2", "lumpy3", "lumpy4"],
"", "", False, "")
class MachineManagerTest(unittest.TestCase):
def testAreAllMachineSame(self):
manager = MyMachineManager(CHROMEOS_ROOT)
for m in MACHINE_NAMES:
manager.AddMachine(m)
self.assertEqual(manager.AreAllMachineSame(LABEL_LUMPY), True)
self.assertEqual(manager.AreAllMachineSame(LABEL_MIX), False)
def testGetMachines(self):
manager = MyMachineManager(CHROMEOS_ROOT)
for m in MACHINE_NAMES:
manager.AddMachine(m)
names = [m.name for m in manager.GetMachines(LABEL_LUMPY)]
self.assertEqual(names, ["lumpy1", "lumpy2", "lumpy3"])
def testGetAvailableMachines(self):
manager = MyMachineManager(CHROMEOS_ROOT)
for m in MACHINE_NAMES:
manager.AddMachine(m)
for m in manager._all_machines:
if int(m.name[-1]) % 2:
manager._TryToLockMachine(m)
names = [m.name for m in manager.GetAvailableMachines(LABEL_LUMPY)]
self.assertEqual(names, ["lumpy1", "lumpy3"])
if __name__ == "__main__":
unittest.main()
|
