commit dc3f0bbee0627bdb0f166cfc19490deda44bacb7
author Evan Hernandez <evanhernandez@chromium.org> Thu Jun 06 12:46:52 2019 -0600
committer Commit Bot <commit-bot@chromium.org> Mon Jun 10 20:22:41 2019 +0000
tree 38496b89ae10fc55343f27d0679bff20da6307e9
parent 7e40d25684e2973dc7b52a7f768b96097047653b

test: Add MoblabVmTest controller.

TEST=./run_tests
BUG=chromium:954276

Change-Id: I52c1208bce4daf6d86747a9d182b1d303d2b3a48
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/1648466
Reviewed-by: Evan Hernandez <evanhernandez@chromium.org>
Tested-by: Evan Hernandez <evanhernandez@chromium.org>
Commit-Queue: Evan Hernandez <evanhernandez@chromium.org>

api/controller/test.py

diff --git a/api/controller/test.py b/api/controller/test.py
index 88508d8..b15f501 100644
--- a/api/controller/test.py
+++ b/api/controller/test.py
@@ -19,9 +19,11 @@
 from chromite.lib import constants
 from chromite.lib import cros_build_lib
 from chromite.lib import failures_lib
+from chromite.lib import image_lib
 from chromite.lib import osutils
 from chromite.lib import portage_util
 from chromite.lib import sysroot_lib
+from chromite.scripts import cros_set_lsb_release
 from chromite.service import test
 
 
@@ -186,3 +188,30 @@
   with osutils.TempDir(prefix='vm-test-results.') as results_dir:
     cmd.extend(['--results-dir', results_dir])
     cros_build_lib.RunCommand(cmd, kill_timeout=10 * 60)
+
+
+def MoblabVmTest(input_proto, _output_proto):
+  """Run Moblab VM tests."""
+  image_payload_dir = input_proto.image_payload.path.path
+  cache_payload_dirs = [cp.path.path for cp in input_proto.cache_payloads]
+
+  # Autotest and Moblab depend on the builder path, so we must read it from
+  # the image.
+  image_file = os.path.join(image_payload_dir, constants.TEST_IMAGE_BIN)
+  with image_lib.LoopbackPartitions(image_file) as image_mount:
+    lsb_release_file = os.path.join(image_mount.destination,
+                                    constants.LSB_RELEASE_PATH.strip('/'))
+    lsb_release_kvs = cros_build_lib.LoadKeyValueFile(lsb_release_file)
+    builder = lsb_release_kvs.get(cros_set_lsb_release.LSB_KEY_BUILDER_PATH)
+
+  if not builder:
+    cros_build_lib.Die('Image did not contain key %s in %s',
+                       cros_set_lsb_release.LSB_KEY_BUILDER_PATH,
+                       constants.LSB_RELEASE_PATH)
+
+  # Now we can run the tests.
+  with osutils.TempDir() as workspace_dir, osutils.TempDir() as results_dir:
+    vms = test.CreateMoblabVm(workspace_dir, image_payload_dir)
+    cache_dir = test.PrepareMoblabVmImageCache(vms, builder, cache_payload_dirs)
+    test.RunMoblabVmTest(vms, builder, cache_dir, results_dir)
+    test.ValidateMoblabVmTest(results_dir)