diff options
Diffstat (limited to 'scripts/update_payload/test_utils.py')
| -rw-r--r-- | scripts/update_payload/test_utils.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/scripts/update_payload/test_utils.py b/scripts/update_payload/test_utils.py index 8746fcef..61a91f53 100644 --- a/scripts/update_payload/test_utils.py +++ b/scripts/update_payload/test_utils.py @@ -219,9 +219,11 @@ class PayloadGenerator(object): """Writes a payload heaer to a file.""" # We need to access protected members in Payload for writing the header. # pylint: disable=W0212 - file_obj.write(payload.Payload._MAGIC) - _WriteInt(file_obj, payload.Payload._VERSION_SIZE, True, self.version) - _WriteInt(file_obj, payload.Payload._MANIFEST_LEN_SIZE, True, manifest_len) + file_obj.write(payload.Payload._PayloadHeader._MAGIC) + _WriteInt(file_obj, payload.Payload._PayloadHeader._VERSION_SIZE, True, + self.version) + _WriteInt(file_obj, payload.Payload._PayloadHeader._MANIFEST_LEN_SIZE, True, + manifest_len) def WriteToFile(self, file_obj, manifest_len=-1, data_blobs=None, sigs_data=None, padding=None): |