def test_keyset_info(self): keyset = tink_pb2.Keyset(primary_key_id=1) keyset.key.extend([ helper.fake_key( value=b'v1', type_url='t1', key_id=1, status=tink_pb2.ENABLED, output_prefix_type=tink_pb2.TINK), helper.fake_key( value=b'v2', type_url='t2', key_id=2, status=tink_pb2.DESTROYED, output_prefix_type=tink_pb2.RAW) ]) handle = _keyset_handle(keyset) expected_keyset_info = tink_pb2.KeysetInfo(primary_key_id=1) info1 = expected_keyset_info.key_info.add() info1.type_url = 't1' info1.status = tink_pb2.ENABLED info1.output_prefix_type = tink_pb2.TINK info1.key_id = 1 info2 = expected_keyset_info.key_info.add() info2.type_url = 't2' info2.status = tink_pb2.DESTROYED info2.output_prefix_type = tink_pb2.RAW info2.key_id = 2 self.assertEqual(expected_keyset_info, handle.keyset_info())
def _keyset_info(keyset: tink_pb2.Keyset) -> tink_pb2.KeysetInfo: keyset_info = tink_pb2.KeysetInfo(primary_key_id=keyset.primary_key_id) for key in keyset.key: key_info = keyset_info.key_info.add() key_info.type_url = key.key_data.type_url key_info.status = key.status key_info.output_prefix_type = key.output_prefix_type key_info.key_id = key.key_id return keyset_info