commit 3c1a23206963f7dcc964aaebd8446e42eb0f32e5
author Benjamin Shai <bshai@google.com> Fri Feb 10 17:29:46 2023 +0000
committer Chromeos LUCI <chromeos-scoped@luci-project-accounts.iam.gserviceaccount.com> Wed Feb 15 19:41:42 2023 +0000
tree 77d551bac137053a2eedebd89a4e4f15fede85c2
parent 991bdc55089ba4a34a2a81ce01c4f7214a1226fd

api: add all payloads to paygen response

Instead of just returning the version 1 payloads, return all of the
payloads from GeneratePayload and populate the output proto with them.

BUG=b:268365767
TEST=unit-test

Change-Id: If250c1db3e6622c02dbcd23a8c294dbc25475292
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/4240333
Commit-Queue: Benjamin Shai <bshai@google.com>
Reviewed-by: Jae Hoon Kim <kimjae@chromium.org>
Tested-by: Benjamin Shai <bshai@google.com>
Reviewed-by: George Engelbrecht <engeg@google.com>
Reviewed-by: Alex Klein <saklein@chromium.org>

api/controller/payload_unittest.py

diff --git a/api/controller/payload_unittest.py b/api/controller/payload_unittest.py
index 5bfb482..0028d50 100644
--- a/api/controller/payload_unittest.py
+++ b/api/controller/payload_unittest.py
@@ -63,8 +63,13 @@
         )
 
         self.result = payload_pb2.GenerationResponse(
-            local_path="/tmp/aohiwdadoi/delta.bin",
-            remote_uri="gs://something",
+            versioned_artifacts=[
+                payload_pb2.GenerationResponse.VersionedArtifact(
+                    version=1,
+                    local_path="/tmp/aohiwdadoi/delta.bin",
+                    remote_uri="gs://something",
+                )
+            ]
         )
 
         self.PatchObject(
@@ -84,7 +89,7 @@
         # Deep patch the paygen lib, this is a full run through service as well.
         patch_obj = self.PatchObject(paygen_payload_lib, "PaygenPayload")
         patch_obj.return_value.Run.return_value = {
-            1: ("/local/foo", "gs://something")
+            1: ("/tmp/aohiwdadoi/delta.bin", "gs://something")
         }
         res = payload.GeneratePayload(self.req, self.result, self.api_config)
         self.assertEqual(res, controller.RETURN_CODE_SUCCESS)
@@ -113,7 +118,7 @@
         """Test a miniOS paygen request."""
         patch = self.PatchObject(paygen_payload_lib, "PaygenPayload")
         patch.return_value.Run.return_value = {
-            1: ("/local/foo", "gs://minios/something")
+            1: ("/tmp/aohiwdadoi/delta.bin", "gs://minios/something")
         }
         res = payload.GeneratePayload(
             self.minios_req, self.result, self.api_config