def testMergeActiveShadowFileMissing(self): """MergeActiveShadowFile recovers missing messages from shadow.""" active_proto = descriptor_pb2.FileDescriptorProto() shadow_proto = descriptor_pb2.FileDescriptorProto() shadow_proto.message_type.add().name = 'foo' target_proto = descriptor_pb2.DescriptorProto() target_proto = merge_active_shadow.MergeActiveShadowFile(active_proto, shadow_proto) self.assertEqual(target_proto.message_type[0].name, 'foo')
def testMergeActiveShadowFileNoShadowEnum(self): """MergeActiveShadowFile doesn't require a shadow enum for new active enums.""" active_proto = descriptor_pb2.FileDescriptorProto() shadow_proto = descriptor_pb2.FileDescriptorProto() active_proto.enum_type.add().name = 'foo' target_proto = descriptor_pb2.DescriptorProto() target_proto = merge_active_shadow.MergeActiveShadowFile(active_proto, shadow_proto) self.assertEqual(target_proto.enum_type[0].name, 'foo')