commit 2bfacb2bc1f7e4ce54b91ae5ce00280e55ddb12f
author Alex Klein <saklein@chromium.org> Mon Feb 04 11:42:17 2019 -0700
committer chrome-bot <chrome-bot@chromium.org> Mon Feb 04 22:49:27 2019 -0800
tree 35a853cb09b5780cad98cca36587236ba905f7cc
parent afc37154aa31042475e80602f8a17ba5aaae9983

Build API: Add chroot assertion options

BUG=chromium:912361
TEST=new unittests, run_tests

Change-Id: Ia83bca59c65ca8875af5ec463480177831b93165
Reviewed-on: https://chromium-review.googlesource.com/1453236
Commit-Ready: Alex Klein <saklein@chromium.org>
Tested-by: Alex Klein <saklein@chromium.org>
Reviewed-by: Lann Martin <lannm@chromium.org>

api/build_api.py

diff --git a/api/build_api.py b/api/build_api.py
index 684a093..40da132 100644
--- a/api/build_api.py
+++ b/api/build_api.py
@@ -18,6 +18,7 @@
 from chromite.api.gen import build_api_pb2
 from chromite.api.gen import autotest_pb2
 from chromite.lib import commandline
+from chromite.lib import cros_build_lib
 from chromite.lib import osutils
 
 
@@ -171,6 +172,10 @@
     if method_options.HasField('implementation_name'):
       method_name = method_options.implementation_name
 
+    # Check the chroot assertion settings before running.
+    service_options = svc.GetOptions().Extensions[self._service_options]
+    self._HandleChrootAssert(service_options, method_options)
+
     # Import the module and get the method.
     method_impl = self._GetMethod(module_name, method_name)
 
@@ -178,6 +183,27 @@
     method_impl(*args)
     return output_msg
 
+  def _HandleChrootAssert(self, service_options, method_options):
+    """Check the chroot assert options and execute assertion as needed.
+
+    Args:
+      service_options (google.protobuf.Message): The service options.
+      method_options (google.protobuf.Message): The method options.
+    """
+    chroot_assert = build_api_pb2.NO_ASSERTION
+    if method_options.HasField('method_chroot_assert'):
+      # Prefer the method option when set.
+      chroot_assert = method_options.method_chroot_assert
+    elif service_options.HasField('service_chroot_assert'):
+      # Fall back to the service option.
+      chroot_assert = service_options.service_chroot_assert
+
+    # Execute appropriate assertion if set.
+    if chroot_assert == build_api_pb2.INSIDE:
+      cros_build_lib.AssertInsideChroot()
+    elif chroot_assert == build_api_pb2.OUTSIDE:
+      cros_build_lib.AssertOutsideChroot()
+
   def _GetMethod(self, module_name, method_name):
     """Get the implementation of the method for the service module.