commit c9707c367cc06d63f7e494de057e328e78b7ade7
author Jack Neus <jackneus@google.com> Fri Jul 23 21:48:54 2021 +0000
committer Commit Bot <commit-bot@chromium.org> Wed Aug 25 20:15:39 2021 +0000
tree fcd5a9f5b9f696406f4db5ce2d40a8e0a8f6faf1
parent 21a8435ae33f830059372742dc8fd8fc86073756

Artifacts service: add support for hwqual bundle

BUG=b:196966502
TEST=run_tests

Change-Id: I14ec89cca8c42779c1679164510a147c1d5ea120
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/3049996
Reviewed-by: Alex Klein <saklein@chromium.org>
Tested-by: Jack Neus <jackneus@google.com>
Commit-Queue: Jack Neus <jackneus@google.com>

api/controller/test.py

diff --git a/api/controller/test.py b/api/controller/test.py
index 3f8cdf3..81014d6 100644
--- a/api/controller/test.py
+++ b/api/controller/test.py
@@ -7,6 +7,7 @@
 Handles all testing related functionality, it is not itself a test.
 """
 
+import functools
 import os
 
 from chromite.api import controller
@@ -29,6 +30,7 @@
 from chromite.lib import sysroot_lib
 from chromite.lib.parser import package_info
 from chromite.scripts import cros_set_lsb_release
+from chromite.service import packages as packages_service
 from chromite.service import test
 from chromite.third_party.google.protobuf import json_format
 from chromite.third_party.google.protobuf import text_format
@@ -333,6 +335,7 @@
 
 def GetArtifacts(in_proto: common_pb2.ArtifactsByService.Test,
     chroot: chroot_lib.Chroot, sysroot_class: sysroot_lib.Sysroot,
+    build_target: build_target_lib.BuildTarget,
     output_dir: str) -> list:
   """Builds and copies test artifacts to specified output_dir.
 
@@ -343,6 +346,7 @@
     in_proto: Proto request defining reqs.
     chroot: The chroot class used for these artifacts.
     sysroot_class: The sysroot class used for these artifacts.
+    build_target: The build target used for these artifacts.
     output_dir: The path to write artifacts to.
 
   Returns:
@@ -354,6 +358,8 @@
     in_proto.ArtifactType.UNIT_TESTS: test.BuildTargetUnitTestTarball,
     in_proto.ArtifactType.CODE_COVERAGE_LLVM_JSON:
         test.BundleCodeCoverageLlvmJson,
+    in_proto.ArtifactType.HWQUAL: functools.partial(test.BundleHwqualTarball,
+      build_target.name, packages_service.determine_full_version()),
   }
 
   for output_artifact in in_proto.output_artifacts: