diff options
Diffstat (limited to 'cli/cros/cros_payload_unittest.py')
| -rw-r--r-- | cli/cros/cros_payload_unittest.py | 165 |
1 files changed, 137 insertions, 28 deletions
diff --git a/cli/cros/cros_payload_unittest.py b/cli/cros/cros_payload_unittest.py index d99f0e9a1..36f4749bd 100644 --- a/cli/cros/cros_payload_unittest.py +++ b/cli/cros/cros_payload_unittest.py @@ -50,10 +50,17 @@ class FakeOp(object): def HasField(self, field): return hasattr(self, field) +class FakePartition(object): + """Fake PartitionUpdate field for testing.""" + + def __init__(self, partition_name, operations): + self.partition_name = partition_name + self.operations = operations + class FakeManifest(object): """Fake manifest for testing.""" - def __init__(self): + def __init__(self, major_version): FakeExtent = collections.namedtuple('FakeExtent', ['start_block', 'num_blocks']) self.install_operations = [FakeOp([], @@ -67,6 +74,11 @@ class FakeManifest(object): [FakeExtent(x, x) for x in xrange(20)], update_payload.common.OpType.SOURCE_COPY, src_length=4096)] + if major_version == cros_payload.MAJOR_PAYLOAD_VERSION_BRILLO: + self.partitions = [FakePartition('rootfs', self.install_operations), + FakePartition('kernel', + self.kernel_install_operations)] + self.install_operations = self.kernel_install_operations = [] self.block_size = 4096 self.minor_version = 4 FakePartInfo = collections.namedtuple('FakePartInfo', ['size']) @@ -81,19 +93,32 @@ class FakeManifest(object): """Fake HasField method based on the python members.""" return hasattr(self, field_name) and getattr(self, field_name) is not None +class FakeHeader(object): + """Fake payload header for testing.""" + + def __init__(self, version, manifest_len, metadata_signature_len): + self.version = version + self.manifest_len = manifest_len + self.metadata_signature_len = metadata_signature_len + + @property + def size(self): + return (20 if self.version == cros_payload.MAJOR_PAYLOAD_VERSION_CHROMEOS + else 24) + + class FakePayload(object): """Fake payload for testing.""" - def __init__(self): - FakeHeader = collections.namedtuple('FakeHeader', - ['version', 'manifest_len']) - self._header = FakeHeader('111', 222) + def __init__(self, major_version): + self._header = FakeHeader(major_version, 222, 0) self.header = None - self._manifest = FakeManifest() + self._manifest = FakeManifest(major_version) self.manifest = None self._blobs = {} - self._signatures = update_metadata_pb2.Signatures() + self._payload_signatures = update_metadata_pb2.Signatures() + self._metadata_signatures = update_metadata_pb2.Signatures() def Init(self): """Fake Init that sets header and manifest. @@ -114,15 +139,28 @@ class FakePayload(object): 'actual: %d)' % (len(blob), length)) return blob - def AddSignature(self, **kwargs): - new_signature = self._signatures.signatures.add() + @staticmethod + def _AddSignatureToProto(proto, **kwargs): + """Add a new Signature element to the passed proto.""" + new_signature = proto.signatures.add() for key, val in kwargs.iteritems(): setattr(new_signature, key, val) - blob = self._signatures.SerializeToString() + + def AddPayloadSignature(self, **kwargs): + self._AddSignatureToProto(self._payload_signatures, **kwargs) + blob = self._payload_signatures.SerializeToString() self._manifest.signatures_offset = 1234 self._manifest.signatures_size = len(blob) self._blobs[self._manifest.signatures_offset] = blob + def AddMetadataSignature(self, **kwargs): + self._AddSignatureToProto(self._metadata_signatures, **kwargs) + if self._header.metadata_signature_len: + del self._blobs[-self._header.metadata_signature_len] + blob = self._metadata_signatures.SerializeToString() + self._header.metadata_signature_len = len(blob) + self._blobs[-len(blob)] = blob + class PayloadCommandTest(cros_test_lib.MockOutputTestCase): """Test class for our PayloadCommand class.""" @@ -137,13 +175,14 @@ class PayloadCommandTest(cros_test_lib.MockOutputTestCase): def testRun(self): """Verify that Run parses and displays the payload like we expect.""" payload_cmd = cros_payload.PayloadCommand(FakeOption(action='show')) - self.PatchObject(update_payload, 'Payload', return_value=FakePayload()) + self.PatchObject(update_payload, 'Payload', return_value=FakePayload( + cros_payload.MAJOR_PAYLOAD_VERSION_CHROMEOS)) with self.OutputCapturer() as output: payload_cmd.Run() stdout = output.GetStdout() - expected_out = """Payload version: 111 + expected_out = """Payload version: 1 Manifest length: 222 Number of operations: 1 Number of kernel ops: 1 @@ -152,17 +191,18 @@ Minor version: 4 """ self.assertEquals(stdout, expected_out) - def testListOps(self): + def testListOpsOnVersion1(self): """Verify that the --list_ops option gives the correct output.""" payload_cmd = cros_payload.PayloadCommand(FakeOption(list_ops=True, action='show')) - self.PatchObject(update_payload, 'Payload', return_value=FakePayload()) + self.PatchObject(update_payload, 'Payload', return_value=FakePayload( + cros_payload.MAJOR_PAYLOAD_VERSION_CHROMEOS)) with self.OutputCapturer() as output: payload_cmd.Run() stdout = output.GetStdout() - expected_out = """Payload version: 111 + expected_out = """Payload version: 1 Manifest length: 222 Number of operations: 1 Number of kernel ops: 1 @@ -185,17 +225,53 @@ Kernel install operations: """ self.assertEquals(stdout, expected_out) - def testStats(self): + def testListOpsOnVersion2(self): + """Verify that the --list_ops option gives the correct output.""" + payload_cmd = cros_payload.PayloadCommand(FakeOption(list_ops=True, + action='show')) + self.PatchObject(update_payload, 'Payload', return_value=FakePayload( + cros_payload.MAJOR_PAYLOAD_VERSION_BRILLO)) + + with self.OutputCapturer() as output: + payload_cmd.Run() + + stdout = output.GetStdout() + expected_out = """Payload version: 2 +Manifest length: 222 +Number of partitions: 2 + Number of "rootfs" ops: 1 + Number of "kernel" ops: 1 +Block size: 4096 +Minor version: 4 + +rootfs install operations: + 0: REPLACE_BZ + Data offset: 1 + Data length: 1 + Destination: 2 extents (3 blocks) + (1,1) (2,2) +kernel install operations: + 0: SOURCE_COPY + Source: 1 extent (1 block) + (1,1) + Destination: 20 extents (190 blocks) + (0,0) (1,1) (2,2) (3,3) (4,4) (5,5) (6,6) (7,7) (8,8) (9,9) (10,10) + (11,11) (12,12) (13,13) (14,14) (15,15) (16,16) (17,17) (18,18) (19,19) +""" + self.assertEquals(stdout, expected_out) + + def testStatsOnVersion1(self): """Verify that the --stats option works correctly.""" payload_cmd = cros_payload.PayloadCommand(FakeOption(stats=True, action='show')) - self.PatchObject(update_payload, 'Payload', return_value=FakePayload()) + self.PatchObject(update_payload, 'Payload', return_value=FakePayload( + cros_payload.MAJOR_PAYLOAD_VERSION_CHROMEOS)) with self.OutputCapturer() as output: payload_cmd.Run() stdout = output.GetStdout() - expected_out = """Payload version: 111 + expected_out = """Payload version: 1 Manifest length: 222 Number of operations: 1 Number of kernel ops: 1 @@ -207,23 +283,49 @@ Seeks when writing: 18 """ self.assertEquals(stdout, expected_out) + def testStatsOnVersion2(self): + """Verify that the --stats option works correctly on version 2.""" + payload_cmd = cros_payload.PayloadCommand(FakeOption(stats=True, + action='show')) + self.PatchObject(update_payload, 'Payload', return_value=FakePayload( + cros_payload.MAJOR_PAYLOAD_VERSION_BRILLO)) + + with self.OutputCapturer() as output: + payload_cmd.Run() + + stdout = output.GetStdout() + expected_out = """Payload version: 2 +Manifest length: 222 +Number of partitions: 2 + Number of "rootfs" ops: 1 + Number of "kernel" ops: 1 +Block size: 4096 +Minor version: 4 +Blocks read: 11 +Blocks written: 193 +Seeks when writing: 18 +""" + self.assertEquals(stdout, expected_out) + def testEmptySignatures(self): """Verify that the --signatures option works with unsigned payloads.""" payload_cmd = cros_payload.PayloadCommand( FakeOption(action='show', signatures=True)) - self.PatchObject(update_payload, 'Payload', return_value=FakePayload()) + self.PatchObject(update_payload, 'Payload', return_value=FakePayload( + cros_payload.MAJOR_PAYLOAD_VERSION_CHROMEOS)) with self.OutputCapturer() as output: payload_cmd.Run() stdout = output.GetStdout() - expected_out = """Payload version: 111 + expected_out = """Payload version: 1 Manifest length: 222 Number of operations: 1 Number of kernel ops: 1 Block size: 4096 Minor version: 4 -No signatures stored in the payload +No metadata signatures stored in the payload +No payload signatures stored in the payload """ self.assertEquals(stdout, expected_out) @@ -232,22 +334,29 @@ No signatures stored in the payload """Verify that the --signatures option shows the present signatures.""" payload_cmd = cros_payload.PayloadCommand( FakeOption(action='show', signatures=True)) - payload = FakePayload() - payload.AddSignature(version=1, data='12345678abcdefgh\x00\x01\x02\x03') - payload.AddSignature(data='I am a signature so access is yes.') + payload = FakePayload(cros_payload.MAJOR_PAYLOAD_VERSION_BRILLO) + payload.AddPayloadSignature(version=1, + data='12345678abcdefgh\x00\x01\x02\x03') + payload.AddPayloadSignature(data='I am a signature so access is yes.') + payload.AddMetadataSignature(data='\x00\x0a\x0c') self.PatchObject(update_payload, 'Payload', return_value=payload) with self.OutputCapturer() as output: payload_cmd.Run() stdout = output.GetStdout() - expected_out = """Payload version: 111 + expected_out = """Payload version: 2 Manifest length: 222 -Number of operations: 1 -Number of kernel ops: 1 +Number of partitions: 2 + Number of "rootfs" ops: 1 + Number of "kernel" ops: 1 Block size: 4096 Minor version: 4 -Signature blob: offset=1234 (64 bytes) +Metadata signatures blob: file_offset=246 (7 bytes) +Metadata signatures: (1 entries) + version=None, hex_data: (3 bytes) + 00 0a 0c | ... +Payload signatures blob: blob_offset=1234 (64 bytes) Payload signatures: (2 entries) version=1, hex_data: (20 bytes) 31 32 33 34 35 36 37 38 61 62 63 64 65 66 67 68 | 12345678abcdefgh |