diff options
Diffstat (limited to 'scripts/update_payload/checker_unittest.py')
| -rwxr-xr-x | scripts/update_payload/checker_unittest.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/scripts/update_payload/checker_unittest.py b/scripts/update_payload/checker_unittest.py index f7182342..68f18078 100755 --- a/scripts/update_payload/checker_unittest.py +++ b/scripts/update_payload/checker_unittest.py @@ -1127,8 +1127,8 @@ class PayloadCheckerTest(mox.MoxTestBase): def DoRunTest(self, rootfs_part_size_provided, kernel_part_size_provided, fail_wrong_payload_type, fail_invalid_block_size, - fail_mismatched_block_size, fail_excess_data, - fail_rootfs_part_size_exceeded, + fail_mismatched_metadata_size, fail_mismatched_block_size, + fail_excess_data, fail_rootfs_part_size_exceeded, fail_kernel_part_size_exceeded): """Tests Run().""" # Generate a test payload. For this test, we generate a full update that @@ -1178,6 +1178,11 @@ class PayloadCheckerTest(mox.MoxTestBase): else: use_block_size = block_size + # For the unittests 246 is the value that generated for the payload. + metadata_size = 246 + if fail_mismatched_metadata_size: + metadata_size += 1 + kwargs = { 'payload_gen_dargs': { 'privkey_file_name': test_utils._PRIVKEY_FILE_NAME, @@ -1196,9 +1201,10 @@ class PayloadCheckerTest(mox.MoxTestBase): kwargs = {'pubkey_file_name': test_utils._PUBKEY_FILE_NAME, 'rootfs_part_size': rootfs_part_size, + 'metadata_size': metadata_size, 'kernel_part_size': kernel_part_size} should_fail = (fail_wrong_payload_type or fail_mismatched_block_size or - fail_excess_data or + fail_mismatched_metadata_size or fail_excess_data or fail_rootfs_part_size_exceeded or fail_kernel_part_size_exceeded) if should_fail: @@ -1353,6 +1359,7 @@ def AddAllParametricTests(): 'kernel_part_size_provided': (True, False), 'fail_wrong_payload_type': (True, False), 'fail_invalid_block_size': (True, False), + 'fail_mismatched_metadata_size': (True, False), 'fail_mismatched_block_size': (True, False), 'fail_excess_data': (True, False), 'fail_rootfs_part_size_exceeded': (True, False), |