commit f2674464ef8666bfaec7aa2ab011c0e9fb4bd3db
author Alex Klein <saklein@chromium.org> Thu May 16 16:47:24 2019 -0600
committer Commit Bot <commit-bot@chromium.org> Fri May 17 21:18:01 2019 +0000
tree 560bdb94a169b1781785e5daff39051f850869ad
parent 76f76ab6546da6ad64052742d0b501aacb6b8904

BuildTargetUnitTest: Add package blacklist support.

BUG=chromium:964128
TEST=run_tests

Change-Id: Icae780e987389057f8a4d477e0432c4ba0b70d97
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/1615655
Tested-by: Alex Klein <saklein@chromium.org>
Reviewed-by: Evan Hernandez <evanhernandez@chromium.org>
Commit-Queue: Alex Klein <saklein@chromium.org>

api/controller/test_unittest.py

diff --git a/api/controller/test_unittest.py b/api/controller/test_unittest.py
index 4ffdf34..48a151b 100644
--- a/api/controller/test_unittest.py
+++ b/api/controller/test_unittest.py
@@ -26,12 +26,18 @@
   """Tests for the UnitTest function."""
 
   def _GetInput(self, board=None, result_path=None, chroot_path=None,
-                cache_dir=None, empty_sysroot=None):
+                cache_dir=None, empty_sysroot=None, blacklist=None):
     """Helper to build an input message instance."""
+    formatted_blacklist = []
+    for pkg in blacklist or []:
+      formatted_blacklist.append({'category': pkg.category,
+                                  'package_name': pkg.package})
+
     return test_pb2.BuildTargetUnitTestRequest(
         build_target={'name': board}, result_path=result_path,
         chroot={'path': chroot_path, 'cache_dir': cache_dir},
-        flags={'empty_sysroot': empty_sysroot}
+        flags={'empty_sysroot': empty_sysroot},
+        package_blacklist=formatted_blacklist,
     )
 
   def _GetOutput(self):
@@ -123,8 +129,10 @@
     self.PatchObject(portage_util, 'ParseParallelEmergeStatusFile',
                      return_value=[])
 
+    pkgs = ['foo/bar', 'cat/pkg']
+    blacklist = [portage_util.SplitCPV(p, strict=False) for p in pkgs]
     input_msg = self._GetInput(board='board', result_path=self.tempdir,
-                               empty_sysroot=True)
+                               empty_sysroot=True, blacklist=blacklist)
     output_msg = self._GetOutput()
 
     rc = test_controller.BuildTargetUnitTest(input_msg, output_msg)
@@ -132,7 +140,8 @@
     self.assertNotEqual(0, rc)
     self.assertFalse(output_msg.failed_packages)
     patch.assert_called_with(constants.SOURCE_ROOT, 'board', extra_env=mock.ANY,
-                             chroot_args=mock.ANY, build_stage=False)
+                             chroot_args=mock.ANY, build_stage=False,
+                             blacklist=pkgs)
 
 
 class VmTestTest(cros_test_lib.MockTestCase):