autotest: Set servo_type when init servo host.
This changes temporally till we upload deploy script and fill al missed servo_types in inventory.
BUG=chromium:1071524
TEST=run local
./server/autoserv -s --host-info-subdir host_info_store -m chromeos4-row9-rack8-host21 --lab True --local-only-host-info True -R -r /usr/local/google/home/otabek/test_result/
Change-Id: I1a85a5ee3df289d7986641368cc821a57b9b780b
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/third_party/autotest/+/2182035
Commit-Queue: Otabek Kasimov <[email protected]>
Tested-by: Otabek Kasimov <[email protected]>
Reviewed-by: Garry Wang <[email protected]>
diff --git a/server/hosts/cros_host_unittest.py b/server/hosts/cros_host_unittest.py
index cb7b1b5..924548d 100755
--- a/server/hosts/cros_host_unittest.py
+++ b/server/hosts/cros_host_unittest.py
@@ -125,7 +125,9 @@
host = MockHost()
host.servo = None
host._servo_host = mock.Mock()
- host._servo_host.get_servo.return_value = 'Not Empty'
+ servo = mock.Mock()
+ servo.get_servo_type.return_value = None
+ host._servo_host.get_servo.return_value = servo
host._servo_host.get_servo_state.return_value = 'SOME_STATE'
host.host_info_store = host_info.InMemoryHostInfoStore()
return host
@@ -169,7 +171,9 @@
def test_set_servo_host_update_servo_state_when_host_exist(self):
host = self.create_host()
host._servo_host = mock.Mock()
- host._servo_host.get_servo.return_value = 'Not Empty'
+ servo = mock.Mock()
+ servo.get_servo_type.return_value = None
+ host._servo_host.get_servo.return_value = servo
host._servo_host.get_servo_state.return_value = 'SOME_STATE'
self.assertEqual(host.host_info_store.get().get_label_value(SERVO_STATE_PREFIX), '')
@@ -190,7 +194,9 @@
def test_set_servo_host_use_servo_state_from_host_when_host_is_passed(self):
host = self.create_host()
servo_host = mock.Mock()
- servo_host.get_servo.return_value = 'Not Empty'
+ servo = mock.Mock()
+ servo.get_servo_type.return_value = None
+ servo_host.get_servo.return_value = servo
servo_host.get_servo_state.return_value = 'state_of_host'
host.set_servo_host(servo_host)
