commit 6242cddb06ce0b71fdf2e698d6977ac34838c876
author Arif Kasim <arifkasim@google.com> Wed Oct 19 17:51:00 2022 +0000
committer Chromeos LUCI <chromeos-scoped@luci-project-accounts.iam.gserviceaccount.com> Tue Nov 01 18:25:56 2022 +0000
tree 372b5447fd84ecd1db1e4556e742816d15ed4c2c
parent b6927cd412d7f0424a284e891aac9e3026a06b06

binhost: add support for saving multiple binhost files.

BUG=b:241593955
TEST=run_tests.py

Change-Id: I53f85cd64e48f32ab2aae3c7c14e54059c245d68
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/3966771
Tested-by: Arif Kasim <arifkasim@google.com>
Reviewed-by: Alex Klein <saklein@chromium.org>
Commit-Queue: Arif Kasim <arifkasim@google.com>

api/controller/binhost_unittest.py

diff --git a/api/controller/binhost_unittest.py b/api/controller/binhost_unittest.py
index 1ec80d7..b7b6aec 100644
--- a/api/controller/binhost_unittest.py
+++ b/api/controller/binhost_unittest.py
@@ -221,6 +221,7 @@
         request.build_target.name = "target"
         request.key = binhost_pb2.POSTSUBMIT_BINHOST
         request.uri = "gs://chromeos-prebuilt/target"
+        request.max_uris = 4
         binhost.SetBinhost(request, self.response, self.mock_call_config)
         patch.assert_not_called()
         self.assertEqual(self.response.output_file, "/path/to/BINHOST.conf")
@@ -236,7 +237,7 @@
         input_proto.private = True
         input_proto.key = binhost_pb2.POSTSUBMIT_BINHOST
         input_proto.uri = "gs://chromeos-prebuilt/target"
-
+        input_proto.max_uris = 4
         binhost.SetBinhost(input_proto, self.response, self.api_config)
 
         self.assertEqual(self.response.output_file, "/path/to/BINHOST.conf")
@@ -245,6 +246,7 @@
             "POSTSUBMIT_BINHOST",
             "gs://chromeos-prebuilt/target",
             private=True,
+            max_uris=4,
         )