api: refresh proto & fix payload unittests
Refresh generated proto. Fix failures in payload unittests
resulting from changed message names.
BUG=None
TEST=./run_pytest
Change-Id: I2443dec939f93d34b68826b24eb9605a9213001d
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/2427505
Commit-Queue: Alex Klein <saklein@chromium.org>
Tested-by: Alex Klein <saklein@chromium.org>
Auto-Submit: Alex Klein <saklein@chromium.org>
Reviewed-by: Michael Mortensen <mmortensen@google.com>
diff --git a/api/controller/payload_unittest.py b/api/controller/payload_unittest.py
index e58fcc2..3d06849 100644
--- a/api/controller/payload_unittest.py
+++ b/api/controller/payload_unittest.py
@@ -20,7 +20,7 @@
"""Unittests for SetBinhost."""
def setUp(self):
- self.response = payload_pb2.PayloadGenerationResult()
+ self.response = payload_pb2.GenerationResponse()
src_build = payload_pb2.Build(version='1.0.0', bucket='test',
channel='test-channel', build_target=
@@ -36,7 +36,7 @@
tgt_image = payload_pb2.UnsignedImage(
build=tgt_build, image_type=6, milestone='R70')
- self.req = payload_pb2.PayloadGenerationRequest(
+ self.req = payload_pb2.GenerationRequest(
tgt_unsigned_image=tgt_image,
src_unsigned_image=src_image,
bucket='test-destination-bucket',
@@ -44,7 +44,7 @@
keyset='update_signer',
dryrun=False)
- self.result = payload_pb2.PayloadGenerationResult()
+ self.result = payload_pb2.GenerationResponse()
def testValidateOnly(self):
"""Sanity check that a validate only call does not execute any logic."""
@@ -54,7 +54,7 @@
self.assertEqual(res, controller.RETURN_CODE_VALID_INPUT)
def testCallSucceeds(self):
- """Check that a call is made succesfully."""
+ """Check that a call is made successfully."""
# Deep patch the paygen lib, this is a full run through service as well.
self.PatchObject(paygen_payload_lib, 'PaygenPayload')
res = payload.GeneratePayload(self.req, self.result, self.api_config)