commit 4f0eb438987f5f26ac42f4a494e72bd311389ea7
author Alex Klein <saklein@chromium.org> Thu May 02 13:56:04 2019 -0600
committer Alex Klein <saklein@chromium.org> Fri May 03 21:41:26 2019 +0000
tree 1b1feae61829b596902aad707785347276ec8059
parent 6db9cdfffdec0ea7c41e48e9e6e829662cede6ac

Build API: Add ImageService/CreateVm method.

BUG=chromium:957758, b:131710961
TEST=run_tests
Cq-Depend: chromium:1592723, chromium:1593932

Change-Id: Id2a481d3e22243ba2846279f61d55713b40983d3
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/1593420
Commit-Queue: Alex Klein <saklein@chromium.org>
Tested-by: Alex Klein <saklein@chromium.org>
Legacy-Commit-Queue: Commit Bot <commit-bot@chromium.org>
Reviewed-by: Evan Hernandez <evanhernandez@chromium.org>

api/controller/test_unittest.py

diff --git a/api/controller/test_unittest.py b/api/controller/test_unittest.py
index 77faaf2..07daa6c 100644
--- a/api/controller/test_unittest.py
+++ b/api/controller/test_unittest.py
@@ -9,6 +9,7 @@
 
 from chromite.api.controller import test as test_controller
 from chromite.api.gen.chromiumos import common_pb2
+from chromite.api.gen.chromite.api import image_pb2
 from chromite.api.gen.chromite.api import test_pb2
 from chromite.cbuildbot import commands
 from chromite.lib import cros_build_lib
@@ -133,7 +134,7 @@
   def _GetInput(self, **kwargs):
     values = dict(
         build_target=common_pb2.BuildTarget(name='target'),
-        vm_image=test_pb2.VmTestRequest.VmImage(path='/path/to/image.bin'),
+        vm_image=image_pb2.VmImage(path='/path/to/image.bin'),
         test_harness=test_pb2.VmTestRequest.TAST,
         vm_tests=[test_pb2.VmTestRequest.VmTest(pattern='suite')],
         ssh_options=test_pb2.VmTestRequest.SshOptions(