commit b20b3d98acab910f3cabea21b33f9f2488647008
author LaMont Jones <lamontjones@chromium.org> Sat Nov 23 11:47:48 2019 -0700
committer Commit Bot <commit-bot@chromium.org> Fri Dec 06 00:42:08 2019 +0000
tree 0dce056d8a8541d375edcd6718717793b13f69e7
parent 4e14195550903b22e14a974f09dd5c1bf8425380

toolchain: start adding PrepareForBuild/Bundle endpoints.

Begin adding the endpoints for the toolchain builders.
Drive-by typo fix in validate.py

BUG=chromium:1019868
TEST=pending

Change-Id: I33334cd28b6d1dcf2d0acffdc3cf1797798c3f85
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/1949018
Tested-by: LaMont Jones <lamontjones@chromium.org>
Reviewed-by: Alex Klein <saklein@chromium.org>
Reviewed-by: Tiancong Wang <tcwang@google.com>
Commit-Queue: Alex Klein <saklein@chromium.org>
Commit-Queue: Tiancong Wang <tcwang@google.com>
Auto-Submit: LaMont Jones <lamontjones@chromium.org>

api/controller/toolchain_unittest.py

diff --git a/api/controller/toolchain_unittest.py b/api/controller/toolchain_unittest.py
index 0c5973d..0d024d1 100644
--- a/api/controller/toolchain_unittest.py
+++ b/api/controller/toolchain_unittest.py
@@ -10,6 +10,7 @@
 from chromite.api import api_config
 from chromite.api.controller import toolchain
 from chromite.api.gen.chromite.api import toolchain_pb2
+from chromite.api.gen.chromiumos.builder_config_pb2 import BuilderConfig
 
 from chromite.lib import cros_build_lib
 from chromite.lib import cros_test_lib
@@ -166,3 +167,43 @@
         build_target=self.board, artifact_type=toolchain_pb2.CHROME_AFDO)
     toolchain.UploadVettedAFDOArtifacts(request, self.response, self.api_config)
     self.command.assert_called_once_with('chrome_afdo', self.board)
+
+
+class PrepareForBuildTest(cros_test_lib.MockTestCase,
+                          api_config.ApiConfigMixin):
+  """Unittests for PrepareForBuild."""
+
+  def setUp(self):
+    self.response = toolchain_pb2.PrepareForToolchainBuildResponse()
+
+  def _GetRequest(self, artifact_types=None):
+    if artifact_types is None:
+      artifact_types = []
+    return toolchain_pb2.PrepareForToolchainBuildRequest(
+        artifact_types=artifact_types,
+    )
+
+  def testReturnsUnknownForUnknown(self):
+    request = self._GetRequest([BuilderConfig.Artifacts.IMAGE_ARCHIVES])
+    toolchain.PrepareForBuild(request, self.response, self.api_config)
+    self.assertEqual(toolchain_pb2.PrepareForToolchainBuildResponse.UNKNOWN,
+                     self.response.build_relevance)
+
+
+class BundleToolchainTest(cros_test_lib.MockTempDirTestCase,
+                          api_config.ApiConfigMixin):
+  """Unittests for BundleToolchain."""
+
+  def setUp(self):
+    self.response = toolchain_pb2.BundleToolchainResponse()
+
+  def _GetRequest(self, artifact_types=None):
+    return toolchain_pb2.BundleToolchainRequest(
+        artifact_types=artifact_types,
+        output_dir=self.tempdir,
+    )
+
+  def testReturnsUnknownForUnknown(self):
+    request = self._GetRequest([BuilderConfig.Artifacts.IMAGE_ARCHIVES])
+    toolchain.BundleArtifacts(request, self.response, self.api_config)
+    self.assertEqual([], list(self.response.artifacts_info))