suite_scheduler: Support schedule multi-DUTs test with Android boards/models
BUG=b:214074008
TEST=bin/run_tests
Change-Id: I2083312e50461e457b7eebf3f4ad21371d7f7ddd
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/infra/suite_scheduler/+/3496724
Tested-by: Garry Wang <xianuowang@chromium.org>
Reviewed-by: Jared Loucks <jaredloucks@google.com>
Commit-Queue: Garry Wang <xianuowang@chromium.org>
diff --git a/config_reader_unittest.py b/config_reader_unittest.py
index 4b4ed18..8cec942 100644
--- a/config_reader_unittest.py
+++ b/config_reader_unittest.py
@@ -47,6 +47,17 @@
lab_config = config_reader.LabConfig(self.config)
self.assertEqual(lab_config.get_cros_model_map(), {})
+ def testGetAndroidModelMap(self):
+ """Ensure android_model_map can be correctly fetched."""
+ self.config.set(config_reader.ANDROID_SETTINGS, 'model_list',
+ '\npixel3_pixel3xl,\npixel3_pixel3a,\npixel4_pixel4a')
+ lab_config = config_reader.LabConfig(self.config)
+ expected = {'pixel3': ['pixel3xl', 'pixel3a'],
+ 'pixel4': ['pixel4a']}
+ actual = lab_config.get_android_model_map()
+ self.assertEqual(len(actual), 2)
+ self.assertEqual(actual, expected)
+
if __name__ == '__main__':
unittest.main()