commit fd68cb1b30f379ae312a5912272d526ea6f7be49
author LaMont Jones <lamontjones@chromium.org> Wed Apr 29 16:43:06 2020 -0600
committer Commit Bot <commit-bot@chromium.org> Thu Apr 30 19:35:02 2020 +0000
tree 5d9f0d8f820074abf67963a97bebd47c23d953a1
parent 4b2d2462450f990d77a4f91aa33eff638f6981c9

toolchain: use artifacts.PrepareForBuildResponse relevance.

BUG=chromium:1034529
TEST=unit tests

Cq-Depend: chromium:2172716
Change-Id: Icf422a6301c44045538933695e69867a31c6a8f9
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/2171567
Tested-by: LaMont Jones <lamontjones@chromium.org>
Reviewed-by: David Burger <dburger@chromium.org>
Reviewed-by: Alex Klein <saklein@chromium.org>
Commit-Queue: LaMont Jones <lamontjones@chromium.org>

api/controller/toolchain.py

diff --git a/api/controller/toolchain.py b/api/controller/toolchain.py
index aef942e..7d5a236 100644
--- a/api/controller/toolchain.py
+++ b/api/controller/toolchain.py
@@ -18,6 +18,7 @@
 from chromite.api.gen.chromiumos.builder_config_pb2 import BuilderConfig
 from chromite.lib import cros_logging as logging
 from chromite.lib import toolchain_util
+from chromite.api.gen.chromite.api.artifacts_pb2 import PrepareForBuildResponse
 
 
 assert sys.version_info >= (3, 6), 'This module requires Python 3.6+'
@@ -160,15 +161,14 @@
   #   elif any UNKNOWN => UNKNOWN
   #   elif any POINTLESS => POINTLESS
   #   else UNKNOWN.
-  proto_resp = toolchain_pb2.PrepareForToolchainBuildResponse
   if toolchain_util.PrepareForBuildReturn.NEEDED in results:
-    output_proto.build_relevance = proto_resp.NEEDED
+    output_proto.build_relevance = PrepareForBuildResponse.NEEDED
   elif toolchain_util.PrepareForBuildReturn.UNKNOWN in results:
-    output_proto.build_relevance = proto_resp.UNKNOWN
+    output_proto.build_relevance = PrepareForBuildResponse.UNKNOWN
   elif toolchain_util.PrepareForBuildReturn.POINTLESS in results:
-    output_proto.build_relevance = proto_resp.POINTLESS
+    output_proto.build_relevance = PrepareForBuildResponse.POINTLESS
   else:
-    output_proto.build_relevance = proto_resp.UNKNOWN
+    output_proto.build_relevance = PrepareForBuildResponse.UNKNOWN
   return controller.RETURN_CODE_SUCCESS