test: MoblabVmTest assumes it is outside chroot

TEST=./run_tests
BUG=chromium:954276

Change-Id: I23f6a475343570ecbc2bcb2d5d90571ad082fc29
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/1660745
Tested-by: Evan Hernandez <evanhernandez@chromium.org>
Auto-Submit: Evan Hernandez <evanhernandez@chromium.org>
Reviewed-by: Jason Clinton <jclinton@chromium.org>
diff --git a/api/controller/test_unittest.py b/api/controller/test_unittest.py
index 616b487..260d9da 100644
--- a/api/controller/test_unittest.py
+++ b/api/controller/test_unittest.py
@@ -11,6 +11,7 @@
 import mock
 
 from chromite.api import controller
+from chromite.api.controller import controller_util
 from chromite.api.controller import test as test_controller
 from chromite.api.gen.chromiumos import common_pb2
 from chromite.api.gen.chromite.api import test_pb2
@@ -199,6 +200,7 @@
 
   def _Input(self):
     return test_pb2.MoblabVmTestRequest(
+        chroot=common_pb2.Chroot(path='path/to/chroot'),
         image_payload=self._Payload(self.image_payload_dir),
         cache_payloads=[self._Payload(self.autotest_payload_dir)])
 
@@ -209,6 +211,8 @@
     self.image_cache_dir = '/mnt/moblab/cache'
     self.image_mount_dir = '/mnt/image'
 
+    self.PatchObject(controller_util, 'ParseChroot')
+
     self.mock_create_moblab_vms = self.PatchObject(
         test_service, 'CreateMoblabVm')
     self.mock_prepare_moblab_vm_image_cache = self.PatchObject(
@@ -245,7 +249,8 @@
         [mock.call(mock.ANY, self.builder, [self.autotest_payload_dir])])
     self.assertEqual(
         self.mock_run_moblab_vm_tests.call_args_list,
-        [mock.call(mock.ANY, self.builder, self.image_cache_dir, mock.ANY)])
+        [mock.call(mock.ANY, mock.ANY, self.builder, self.image_cache_dir,
+                   mock.ANY)])
     self.assertEqual(
         self.mock_validate_moblab_vm_tests.call_args_list,
         [mock.call(mock.ANY)])