paycheck: allow to disable specific checks
This became necessary as the delta generator appears to generate
payloads that fail certain checks (e.g. during update_engine unit
testing).
BUG=None
TEST=Disabled checks not being triggered
Change-Id: I4491e0cb32ef44f85e11ffb0402b40d1371525ae
Reviewed-on: https://gerrit.chromium.org/gerrit/49676
Tested-by: Gilad Arnold <garnold@chromium.org>
Reviewed-by: Chris Sosa <sosa@chromium.org>
Commit-Queue: Gilad Arnold <garnold@chromium.org>
diff --git a/scripts/update_payload/payload.py b/scripts/update_payload/payload.py
index dbb385a..b4760b2 100644
--- a/scripts/update_payload/payload.py
+++ b/scripts/update_payload/payload.py
@@ -175,7 +175,7 @@
def Check(self, pubkey_file_name=None, metadata_sig_file=None,
report_out_file=None, assert_type=None, block_size=0,
- allow_unhashed=False):
+ allow_unhashed=False, disabled_tests=()):
"""Checks the payload integrity.
Args:
@@ -185,6 +185,7 @@
assert_type: assert that payload is either 'full' or 'delta'
block_size: expected filesystem / payload block size
allow_unhashed: allow unhashed operation blobs
+ disabled_tests: list of tests to disable
Raises:
PayloadError if payload verification failed.
@@ -192,11 +193,12 @@
self._AssertInit()
# Create a short-lived payload checker object and run it.
- helper = checker.PayloadChecker(self)
+ helper = checker.PayloadChecker(
+ self, assert_type=assert_type, block_size=block_size,
+ allow_unhashed=allow_unhashed, disabled_tests=disabled_tests)
helper.Run(pubkey_file_name=pubkey_file_name,
metadata_sig_file=metadata_sig_file,
- report_out_file=report_out_file, assert_type=assert_type,
- block_size=block_size, allow_unhashed=allow_unhashed)
+ report_out_file=report_out_file)
def Apply(self, dst_kernel_part, dst_rootfs_part, src_kernel_part=None,
src_rootfs_part=None):