# Protocol Buffers - Google's data interchange format
# Copyright 2008 Google Inc.  All rights reserved.
#
# Use of this source code is governed by a BSD-style
# license that can be found in the LICENSE file or at
# https://developers.google.com/open-source/licenses/bsd

"""Unittest for google.protobuf.internal.descriptor."""

__author__ = 'robinson@google.com (Will Robinson)'

import unittest
import warnings

from google.protobuf import descriptor
from google.protobuf import descriptor_pb2
from google.protobuf import descriptor_pool
from google.protobuf import message
from google.protobuf import symbol_database
from google.protobuf import text_format
from google.protobuf.internal import api_implementation
from google.protobuf.internal import test_proto2_pb2
from google.protobuf.internal import test_util
from google.protobuf.internal import testing_refleaks

from absl.testing import parameterized
from google.protobuf import unittest_custom_options_pb2
from google.protobuf import unittest_features_pb2
from google.protobuf import unittest_import_pb2
from google.protobuf import unittest_legacy_features_pb2
from google.protobuf import unittest_pb2
from google.protobuf import unittest_proto3_extensions_pb2
from google.protobuf import unittest_proto3_pb2

TEST_EMPTY_MESSAGE_DESCRIPTOR_ASCII = """
name: 'TestEmptyMessage'
"""

TEST_FILE_DESCRIPTOR_DEBUG = """syntax = "proto2";

package proto2_unittest;

message NestedMessage {
  enum ForeignEnum {
    FOREIGN_FOO = 4;
    FOREIGN_BAR = 5;
    FOREIGN_BAZ = 6;
    FOREIGN_BAX = 32;
  }
  optional int32 bb = 1;
}

message ResponseMessage {
}

service DescriptorTestService {
  rpc CallMethod(.proto2_unittest.NestedMessage) returns (.proto2_unittest.ResponseMessage);
}

"""


warnings.simplefilter('error', DeprecationWarning)


@testing_refleaks.TestCase
class DescriptorTest(unittest.TestCase):

  def setUp(self):
    file_proto = descriptor_pb2.FileDescriptorProto(
        name='some/filename/some.proto', package='proto2_unittest'
    )
    message_proto = file_proto.message_type.add(name='NestedMessage')
    message_proto.field.add(
        name='bb',
        number=1,
        type=descriptor_pb2.FieldDescriptorProto.TYPE_INT32,
        label=descriptor_pb2.FieldDescriptorProto.LABEL_OPTIONAL,
    )
    enum_proto = message_proto.enum_type.add(name='ForeignEnum')
    enum_proto.value.add(name='FOREIGN_FOO', number=4)
    enum_proto.value.add(name='FOREIGN_BAR', number=5)
    enum_proto.value.add(name='FOREIGN_BAZ', number=6)
    enum_proto.value.add(name='FOREIGN_BAX', number=32)

    file_proto.message_type.add(name='ResponseMessage')
    service_proto = file_proto.service.add(name='DescriptorTestService')
    method_proto = service_proto.method.add(
        name='CallMethod',
        input_type='.proto2_unittest.NestedMessage',
        output_type='.proto2_unittest.ResponseMessage',
    )

    # Note: Calling DescriptorPool.Add() multiple times with the same file only
    # works if the input is canonical; in particular, all type names must be
    # fully qualified.
    self.pool = self.GetDescriptorPool()
    self.pool.Add(file_proto)
    self.my_file = self.pool.FindFileByName(file_proto.name)
    self.my_message = self.my_file.message_types_by_name[message_proto.name]
    self.my_enum = self.my_message.enum_types_by_name[enum_proto.name]
    self.my_service = self.my_file.services_by_name[service_proto.name]
    self.my_method = self.my_service.methods_by_name[method_proto.name]

  def GetDescriptorPool(self):
    return symbol_database.Default().pool

  def testMissingPackage(self):
    file_proto = descriptor_pb2.FileDescriptorProto(
        name='some/filename/some.proto'
    )
    serialized = file_proto.SerializeToString()
    pool = descriptor_pool.DescriptorPool()
    file_descriptor = pool.AddSerializedFile(serialized)
    self.assertEqual('', file_descriptor.package)

  def testEmptyPackage(self):
    file_proto = descriptor_pb2.FileDescriptorProto(
        name='some/filename/some.proto', package=''
    )
    serialized = file_proto.SerializeToString()
    pool = descriptor_pool.DescriptorPool()
    file_descriptor = pool.AddSerializedFile(serialized)
    self.assertEqual('', file_descriptor.package)

  @unittest.skipIf(
      api_implementation.Type() in ('upb', 'cpp'),
      "Currently broken in OSS, will be fixed in the next breaking release"
  )
  def testBadEqOnFieldsByName(self):
    fields = unittest_pb2.TestAllTypes.DESCRIPTOR.fields_by_name

    class BadEq:

      def __eq__(self, other):
        raise RuntimeError('comparison error')

      def __iter__(self):
        return iter([])

      def __len__(self):
        return 0

    with self.assertRaises(RuntimeError):
      fields == BadEq()

  def testReservedName(self):
    text = """
      name: "foo.proto"
      message_type {
        name: "BrokenMessageFoo"
        reserved_name: "is_deprecated"
      }
      """

    fdp = text_format.Parse(text, descriptor_pb2.FileDescriptorProto())
    serialized = fdp.SerializeToString()
    # AddSerializedFile() will allow duplicate adds but only if the descriptors
    # are identical and can round-trip through a FileDescriptor losslessly.
    desc1 = descriptor_pool.Default().AddSerializedFile(serialized)
    desc2 = descriptor_pool.Default().AddSerializedFile(serialized)
    self.assertEqual(desc1, desc2)

  def testReservedRange(self):
    text = """
      name: "bar.proto"
      message_type {
        name: "BrokenMessageBar"
        reserved_range {
          start: 101
          end: 102
        }
      }
      """

    fdp = text_format.Parse(text, descriptor_pb2.FileDescriptorProto())
    serialized = fdp.SerializeToString()
    # AddSerializedFile() will allow duplicate adds but only if the descriptors
    # are identical and can round-trip through a FileDescriptor losslessly.
    desc1 = descriptor_pool.Default().AddSerializedFile(serialized)
    desc2 = descriptor_pool.Default().AddSerializedFile(serialized)
    self.assertEqual(desc1, desc2)

  def testReservedNameEnum(self):
    text = """
      name: "baz.proto"
      enum_type {
        name: "BrokenMessageBaz"
        value: <
          name: 'ENUM_BAZ'
          number: 114
        >
        reserved_name: "is_deprecated"
      }
      """

    fdp = text_format.Parse(text, descriptor_pb2.FileDescriptorProto())
    serialized = fdp.SerializeToString()
    # AddSerializedFile() will allow duplicate adds but only if the descriptors
    # are identical and can round-trip through a FileDescriptor losslessly.
    desc1 = descriptor_pool.Default().AddSerializedFile(serialized)
    desc2 = descriptor_pool.Default().AddSerializedFile(serialized)
    self.assertEqual(desc1, desc2)

  def testReservedRangeEnum(self):
    text = """
      name: "bat.proto"
      enum_type {
        name: "BrokenMessageBat"
        value: <
          name: 'ENUM_BAT'
          number: 115
        >
        reserved_range {
          start: 1001
          end: 1002
        }
      }
      """

    fdp = text_format.Parse(text, descriptor_pb2.FileDescriptorProto())
    serialized = fdp.SerializeToString()
    # AddSerializedFile() will allow duplicate adds but only if the descriptors
    # are identical and can round-trip through a FileDescriptor losslessly.
    desc1 = descriptor_pool.Default().AddSerializedFile(serialized)
    desc2 = descriptor_pool.Default().AddSerializedFile(serialized)
    self.assertEqual(desc1, desc2)

  def testFindMethodByName(self):
    service_descriptor = (
        unittest_custom_options_pb2.TestServiceWithCustomOptions.DESCRIPTOR
    )
    method_descriptor = service_descriptor.FindMethodByName('Foo')
    self.assertEqual(method_descriptor.name, 'Foo')
    with self.assertRaises(KeyError):
      service_descriptor.FindMethodByName('MethodDoesNotExist')

  def testEnumValueName(self):
    self.assertEqual(
        self.my_message.EnumValueName('ForeignEnum', 4), 'FOREIGN_FOO'
    )

    self.assertEqual(
        self.my_message.enum_types_by_name['ForeignEnum']
        .values_by_number[4]
        .name,
        self.my_message.EnumValueName('ForeignEnum', 4),
    )
    with self.assertRaises(KeyError):
      self.my_message.EnumValueName('ForeignEnum', 999)
    with self.assertRaises(KeyError):
      self.my_message.EnumValueName('NoneEnum', 999)
    with self.assertRaises(TypeError):
      self.my_message.EnumValueName()

  def testEnumFixups(self):
    self.assertEqual(self.my_enum, self.my_enum.values[0].type)

  def testContainingTypeFixups(self):
    self.assertEqual(self.my_message, self.my_message.fields[0].containing_type)
    self.assertEqual(self.my_message, self.my_enum.containing_type)

  def testContainingServiceFixups(self):
    self.assertEqual(self.my_service, self.my_method.containing_service)

  def testGetOptions(self):
    self.assertEqual(self.my_enum.GetOptions(), descriptor_pb2.EnumOptions())
    self.assertEqual(
        self.my_enum.values[0].GetOptions(), descriptor_pb2.EnumValueOptions()
    )
    self.assertEqual(
        self.my_message.GetOptions(), descriptor_pb2.MessageOptions()
    )
    self.assertEqual(
        self.my_message.fields[0].GetOptions(), descriptor_pb2.FieldOptions()
    )
    self.assertEqual(
        self.my_method.GetOptions(), descriptor_pb2.MethodOptions()
    )
    self.assertEqual(
        self.my_service.GetOptions(), descriptor_pb2.ServiceOptions()
    )

  @unittest.skipIf(
      api_implementation.Type() == 'upb',
      'Needs to wait for a breaking change release in OSS'
  )
  @unittest.skipIf(
      api_implementation.Type() == 'cpp',
      'Needs to wait for a breaking change release in OSS'
  )
  @unittest.skipIf(
      api_implementation.Type() == 'python',
      'Needs to wait for a breaking change release in OSS'
  )
  def testModifyFrozenMessage(self):
    immutability_error = message.FrozenInstanceError
    message_options = self.my_message.GetOptions()
    other_options = descriptor_pb2.MessageOptions()
    other_options.deprecated = True

    # Singular field mutation
    with self.assertRaises(immutability_error):
      message_options.deprecated = True

    # Clear methods
    with self.assertRaises(immutability_error):
      message_options.Clear()
    with self.assertRaises(immutability_error):
      message_options.ClearField('deprecated')

    # Merge/Copy
    with self.assertRaises(immutability_error):
      message_options.MergeFrom(other_options)
    with self.assertRaises(immutability_error):
      message_options.CopyFrom(other_options)

    # Repeated field mutations
    repeated_field = message_options.uninterpreted_option
    with self.assertRaises(immutability_error):
      repeated_field.add()
    with self.assertRaises(immutability_error):
      repeated_field.append(descriptor_pb2.UninterpretedOption())
    with self.assertRaises(immutability_error):
      repeated_field.extend([descriptor_pb2.UninterpretedOption()])
    with self.assertRaises(immutability_error):
      repeated_field.insert(0, descriptor_pb2.UninterpretedOption())
    with self.assertRaises(immutability_error):
      repeated_field.remove(descriptor_pb2.UninterpretedOption())
    with self.assertRaises(immutability_error):
      repeated_field.pop()
    with self.assertRaises(immutability_error):
      repeated_field.sort()
    with self.assertRaises(immutability_error):
      repeated_field.reverse()
    with self.assertRaises(immutability_error):
      del repeated_field[:]

    # Unset submessage mutation
    complex_opt1 = unittest_custom_options_pb2.complex_opt1
    stub_submsg = unittest_pb2.TestAllTypes.DESCRIPTOR.GetOptions().Extensions[
        complex_opt1
    ]
    with self.assertRaises(immutability_error):
      stub_submsg.foo = 5

    # Non-empty repeated field mutation
    complex_options_msg = (
        unittest_custom_options_pb2.VariousComplexOptions.DESCRIPTOR.GetOptions()
    )
    non_empty_repeated = complex_options_msg.Extensions[complex_opt1].foo4
    self.assertEqual(len(non_empty_repeated), 2)
    with self.assertRaises(immutability_error):
      non_empty_repeated.clear()
    with self.assertRaises(immutability_error):
      non_empty_repeated.sort()
    with self.assertRaises(immutability_error):
      non_empty_repeated.remove(99)
    with self.assertRaises(immutability_error):
      non_empty_repeated.pop()
    with self.assertRaises(immutability_error):
      non_empty_repeated.reverse()

    # Non-empty repeated composite field item access
    complex_opt2 = unittest_custom_options_pb2.complex_opt2
    non_empty_repeated_composite = complex_options_msg.Extensions[
        complex_opt2
    ].barney
    self.assertEqual(len(non_empty_repeated_composite), 2)
    first_barney = non_empty_repeated_composite[0]
    self.assertEqual(first_barney.waldo, 101)
    with self.assertRaises(immutability_error):
      first_barney.waldo = 999

    # Extension dict mutation
    with self.assertRaises(immutability_error):
      message_options.Extensions[complex_opt1] = descriptor_pb2.MessageOptions()

    message_opt1 = unittest_custom_options_pb2.message_opt1
    with self.assertRaises(immutability_error):
      message_options.Extensions[message_opt1] = -56

    with self.assertRaises(immutability_error):
      del message_options.Extensions[complex_opt1]

    with self.assertRaises(immutability_error):
      message_options.ClearExtension(complex_opt1)

    # Map field mutations
    map_field = stub_submsg.my_map
    with self.assertRaises(immutability_error):
      map_field['key'] = 123
    with self.assertRaises(immutability_error):
      del map_field['key']
    with self.assertRaises(immutability_error):
      map_field.setdefault('key', 123)
    with self.assertRaises(immutability_error):
      map_field.update({'key': 123})
    with self.assertRaises(immutability_error):
      map_field.MergeFrom(map_field)

  def testModifyOptions(self):
    # We unfortunately allow modification of options returned from GetOptions().
    # This is not intended, and has negative consequences:
    # - It makes the results of GetOptions() invalid if the options are
    #   modified.
    # - It has an efficiency cost from copying the options.
    #
    # This will be fixed in a breaking change release.
    message_options = self.my_message.GetOptions()
    message_options.deprecated = True
    self.assertTrue(
        message_options.deprecated,
        f'Modification ignored on {api_implementation.Type()}',
    )

    # Modification is (unfortunately) reflected in the descriptor.
    self.assertTrue(self.my_message.GetOptions().deprecated)

  def testImmutableMapLookup(self):
    complex_opt1 = unittest_custom_options_pb2.complex_opt1
    complex_options_msg = (
        unittest_custom_options_pb2.VariousComplexOptions.DESCRIPTOR.GetOptions()
    )
    immutable_map = complex_options_msg.Extensions[complex_opt1].my_map

    # Test lookups.
    self.assertEqual(immutable_map['key'], 123)
    self.assertEqual(immutable_map['other_key'], 456)
    self.assertIn('key', immutable_map)
    self.assertIn('other_key', immutable_map)
    self.assertNotIn('nonexistent_key', immutable_map)
    self.assertEqual(len(immutable_map), 2)

    # Test lookups via bytes.
    self.assertEqual(immutable_map[b'key'], 123)
    self.assertEqual(immutable_map[b'other_key'], 456)
    self.assertIn(b'key', immutable_map)
    self.assertIn(b'other_key', immutable_map)
    self.assertNotIn(b'nonexistent_key', immutable_map)
    self.assertEqual(len(immutable_map), 2)

    # Test iteration.
    self.assertEqual(set(immutable_map.keys()), {'key', 'other_key'})
    self.assertEqual(set(immutable_map.values()), {123, 456})
    self.assertEqual(
        set(immutable_map.items()), {('key', 123), ('other_key', 456)}
    )

    # # Test get().
    self.assertEqual(immutable_map.get('key'), 123)
    self.assertEqual(immutable_map.get('nonexistent_key'), None)
    self.assertEqual(immutable_map.get('nonexistent_key', 999), 999)

  def testSimpleCustomOptions(self):
    file_descriptor = unittest_custom_options_pb2.DESCRIPTOR
    message_descriptor = (
        unittest_custom_options_pb2.TestMessageWithCustomOptions.DESCRIPTOR
    )
    field_descriptor = message_descriptor.fields_by_name['field1']
    oneof_descriptor = message_descriptor.oneofs_by_name['AnOneof']
    enum_descriptor = message_descriptor.enum_types_by_name['AnEnum']
    enum_value_descriptor = message_descriptor.enum_values_by_name[
        'ANENUM_VAL2'
    ]
    other_enum_value_descriptor = message_descriptor.enum_values_by_name[
        'ANENUM_VAL1'
    ]
    service_descriptor = (
        unittest_custom_options_pb2.TestServiceWithCustomOptions.DESCRIPTOR
    )
    method_descriptor = service_descriptor.FindMethodByName('Foo')

    file_options = file_descriptor.GetOptions()
    file_opt1 = unittest_custom_options_pb2.file_opt1
    self.assertEqual(9876543210, file_options.Extensions[file_opt1])
    message_options = message_descriptor.GetOptions()
    message_opt1 = unittest_custom_options_pb2.message_opt1
    self.assertEqual(-56, message_options.Extensions[message_opt1])
    field_options = field_descriptor.GetOptions()
    field_opt1 = unittest_custom_options_pb2.field_opt1
    self.assertEqual(8765432109, field_options.Extensions[field_opt1])
    field_opt2 = unittest_custom_options_pb2.field_opt2
    self.assertEqual(42, field_options.Extensions[field_opt2])
    oneof_options = oneof_descriptor.GetOptions()
    oneof_opt1 = unittest_custom_options_pb2.oneof_opt1
    self.assertEqual(-99, oneof_options.Extensions[oneof_opt1])
    enum_options = enum_descriptor.GetOptions()
    enum_opt1 = unittest_custom_options_pb2.enum_opt1
    self.assertEqual(-789, enum_options.Extensions[enum_opt1])
    enum_value_options = enum_value_descriptor.GetOptions()
    enum_value_opt1 = unittest_custom_options_pb2.enum_value_opt1
    self.assertEqual(123, enum_value_options.Extensions[enum_value_opt1])

    service_options = service_descriptor.GetOptions()
    service_opt1 = unittest_custom_options_pb2.service_opt1
    self.assertEqual(-9876543210, service_options.Extensions[service_opt1])
    self.assertTrue(method_descriptor.has_options)
    method_options = method_descriptor.GetOptions()
    method_opt1 = unittest_custom_options_pb2.method_opt1
    self.assertEqual(
        unittest_custom_options_pb2.METHODOPT1_VAL2,
        method_options.Extensions[method_opt1],
    )

    message_descriptor = (
        unittest_custom_options_pb2.DummyMessageContainingEnum.DESCRIPTOR
    )
    self.assertTrue(file_descriptor.has_options)
    self.assertFalse(message_descriptor.has_options)
    self.assertTrue(field_descriptor.has_options)
    self.assertTrue(oneof_descriptor.has_options)
    self.assertTrue(enum_descriptor.has_options)
    self.assertTrue(enum_value_descriptor.has_options)
    self.assertFalse(other_enum_value_descriptor.has_options)

  def testCustomOptionsCopyTo(self):
    message_descriptor = (
        unittest_custom_options_pb2.TestMessageWithCustomOptions.DESCRIPTOR
    )
    message_proto = descriptor_pb2.DescriptorProto()
    message_descriptor.CopyToProto(message_proto)
    self.assertEqual(len(message_proto.options.ListFields()), 2)

  def testDifferentCustomOptionTypes(self):
    kint32min = -(2**31)
    kint64min = -(2**63)
    kint32max = 2**31 - 1
    kint64max = 2**63 - 1
    kuint32max = 2**32 - 1
    kuint64max = 2**64 - 1

    message_descriptor = (
        unittest_custom_options_pb2.CustomOptionMinIntegerValues.DESCRIPTOR
    )
    message_options = message_descriptor.GetOptions()
    self.assertEqual(
        False, message_options.Extensions[unittest_custom_options_pb2.bool_opt]
    )
    self.assertEqual(
        kint32min,
        message_options.Extensions[unittest_custom_options_pb2.int32_opt],
    )
    self.assertEqual(
        kint64min,
        message_options.Extensions[unittest_custom_options_pb2.int64_opt],
    )
    self.assertEqual(
        0, message_options.Extensions[unittest_custom_options_pb2.uint32_opt]
    )
    self.assertEqual(
        0, message_options.Extensions[unittest_custom_options_pb2.uint64_opt]
    )
    self.assertEqual(
        kint32min,
        message_options.Extensions[unittest_custom_options_pb2.sint32_opt],
    )
    self.assertEqual(
        kint64min,
        message_options.Extensions[unittest_custom_options_pb2.sint64_opt],
    )
    self.assertEqual(
        0, message_options.Extensions[unittest_custom_options_pb2.fixed32_opt]
    )
    self.assertEqual(
        0, message_options.Extensions[unittest_custom_options_pb2.fixed64_opt]
    )
    self.assertEqual(
        kint32min,
        message_options.Extensions[unittest_custom_options_pb2.sfixed32_opt],
    )
    self.assertEqual(
        kint64min,
        message_options.Extensions[unittest_custom_options_pb2.sfixed64_opt],
    )

    message_descriptor = (
        unittest_custom_options_pb2.CustomOptionMaxIntegerValues.DESCRIPTOR
    )
    message_options = message_descriptor.GetOptions()
    self.assertEqual(
        True, message_options.Extensions[unittest_custom_options_pb2.bool_opt]
    )
    self.assertEqual(
        kint32max,
        message_options.Extensions[unittest_custom_options_pb2.int32_opt],
    )
    self.assertEqual(
        kint64max,
        message_options.Extensions[unittest_custom_options_pb2.int64_opt],
    )
    self.assertEqual(
        kuint32max,
        message_options.Extensions[unittest_custom_options_pb2.uint32_opt],
    )
    self.assertEqual(
        kuint64max,
        message_options.Extensions[unittest_custom_options_pb2.uint64_opt],
    )
    self.assertEqual(
        kint32max,
        message_options.Extensions[unittest_custom_options_pb2.sint32_opt],
    )
    self.assertEqual(
        kint64max,
        message_options.Extensions[unittest_custom_options_pb2.sint64_opt],
    )
    self.assertEqual(
        kuint32max,
        message_options.Extensions[unittest_custom_options_pb2.fixed32_opt],
    )
    self.assertEqual(
        kuint64max,
        message_options.Extensions[unittest_custom_options_pb2.fixed64_opt],
    )
    self.assertEqual(
        kint32max,
        message_options.Extensions[unittest_custom_options_pb2.sfixed32_opt],
    )
    self.assertEqual(
        kint64max,
        message_options.Extensions[unittest_custom_options_pb2.sfixed64_opt],
    )

    message_descriptor = (
        unittest_custom_options_pb2.CustomOptionOtherValues.DESCRIPTOR
    )
    message_options = message_descriptor.GetOptions()
    self.assertEqual(
        -100, message_options.Extensions[unittest_custom_options_pb2.int32_opt]
    )
    self.assertAlmostEqual(
        12.3456789,
        message_options.Extensions[unittest_custom_options_pb2.float_opt],
        6,
    )
    self.assertAlmostEqual(
        1.234567890123456789,
        message_options.Extensions[unittest_custom_options_pb2.double_opt],
    )
    self.assertEqual(
        'Hello, "World"',
        message_options.Extensions[unittest_custom_options_pb2.string_opt],
    )
    self.assertEqual(
        b'Hello\0World',
        message_options.Extensions[unittest_custom_options_pb2.bytes_opt],
    )
    dummy_enum = unittest_custom_options_pb2.DummyMessageContainingEnum
    self.assertEqual(
        dummy_enum.TEST_OPTION_ENUM_TYPE2,
        message_options.Extensions[unittest_custom_options_pb2.enum_opt],
    )

    message_descriptor = (
        unittest_custom_options_pb2.SettingRealsFromPositiveInts.DESCRIPTOR
    )
    message_options = message_descriptor.GetOptions()
    self.assertAlmostEqual(
        12, message_options.Extensions[unittest_custom_options_pb2.float_opt], 6
    )
    self.assertAlmostEqual(
        154, message_options.Extensions[unittest_custom_options_pb2.double_opt]
    )

    message_descriptor = (
        unittest_custom_options_pb2.SettingRealsFromNegativeInts.DESCRIPTOR
    )
    message_options = message_descriptor.GetOptions()
    self.assertAlmostEqual(
        -12,
        message_options.Extensions[unittest_custom_options_pb2.float_opt],
        6,
    )
    self.assertAlmostEqual(
        -154, message_options.Extensions[unittest_custom_options_pb2.double_opt]
    )

  def testComplexExtensionOptions(self):
    descriptor = unittest_custom_options_pb2.VariousComplexOptions.DESCRIPTOR
    options = descriptor.GetOptions()
    self.assertEqual(
        42, options.Extensions[unittest_custom_options_pb2.complex_opt1].foo
    )
    self.assertEqual(
        324,
        options.Extensions[unittest_custom_options_pb2.complex_opt1].Extensions[
            unittest_custom_options_pb2.mooo
        ],
    )
    self.assertEqual(
        876,
        options.Extensions[unittest_custom_options_pb2.complex_opt1]
        .Extensions[unittest_custom_options_pb2.corge]
        .moo,
    )
    self.assertEqual(
        987, options.Extensions[unittest_custom_options_pb2.complex_opt2].baz
    )
    self.assertEqual(
        654,
        options.Extensions[unittest_custom_options_pb2.complex_opt2].Extensions[
            unittest_custom_options_pb2.grault
        ],
    )
    self.assertEqual(
        743,
        options.Extensions[unittest_custom_options_pb2.complex_opt2].bar.foo,
    )
    self.assertEqual(
        1999,
        options.Extensions[
            unittest_custom_options_pb2.complex_opt2
        ].bar.Extensions[unittest_custom_options_pb2.mooo],
    )
    self.assertEqual(
        2008,
        options.Extensions[unittest_custom_options_pb2.complex_opt2]
        .bar.Extensions[unittest_custom_options_pb2.corge]
        .moo,
    )
    self.assertEqual(
        741,
        options.Extensions[unittest_custom_options_pb2.complex_opt2]
        .Extensions[unittest_custom_options_pb2.garply]
        .foo,
    )
    self.assertEqual(
        1998,
        options.Extensions[unittest_custom_options_pb2.complex_opt2]
        .Extensions[unittest_custom_options_pb2.garply]
        .Extensions[unittest_custom_options_pb2.mooo],
    )
    self.assertEqual(
        2121,
        options.Extensions[unittest_custom_options_pb2.complex_opt2]
        .Extensions[unittest_custom_options_pb2.garply]
        .Extensions[unittest_custom_options_pb2.corge]
        .moo,
    )
    self.assertEqual(
        1971,
        options.Extensions[
            unittest_custom_options_pb2.ComplexOptionType2.ComplexOptionType4.complex_opt4
        ].waldo,
    )
    self.assertEqual(
        321,
        options.Extensions[unittest_custom_options_pb2.complex_opt2].fred.waldo,
    )
    self.assertEqual(
        9, options.Extensions[unittest_custom_options_pb2.complex_opt3].moo
    )
    self.assertEqual(
        22,
        options.Extensions[
            unittest_custom_options_pb2.complex_opt3
        ].complexoptiontype5.plugh,
    )
    self.assertEqual(
        24, options.Extensions[unittest_custom_options_pb2.complexopt6].xyzzy
    )

  # Check that aggregate options were parsed and saved correctly in
  # the appropriate descriptors.
  def testAggregateOptions(self):
    file_descriptor = unittest_custom_options_pb2.DESCRIPTOR
    message_descriptor = unittest_custom_options_pb2.AggregateMessage.DESCRIPTOR
    field_descriptor = message_descriptor.fields_by_name['fieldname']
    enum_descriptor = unittest_custom_options_pb2.AggregateEnum.DESCRIPTOR
    enum_value_descriptor = enum_descriptor.values_by_name['VALUE']
    service_descriptor = unittest_custom_options_pb2.AggregateService.DESCRIPTOR
    method_descriptor = service_descriptor.FindMethodByName('Method')

    # Tests for the different types of data embedded in fileopt
    file_options = file_descriptor.GetOptions().Extensions[
        unittest_custom_options_pb2.fileopt
    ]
    self.assertEqual(100, file_options.i)
    self.assertEqual('FileAnnotation', file_options.s)
    self.assertEqual('NestedFileAnnotation', file_options.sub.s)
    self.assertEqual(
        'FileExtensionAnnotation',
        file_options.file.Extensions[unittest_custom_options_pb2.fileopt].s,
    )
    self.assertEqual(
        'EmbeddedMessageSetElement',
        file_options.mset.Extensions[
            unittest_custom_options_pb2.AggregateMessageSetElement.message_set_extension
        ].s,
    )

    # Simple tests for all the other types of annotations
    self.assertEqual(
        'MessageAnnotation',
        message_descriptor.GetOptions()
        .Extensions[unittest_custom_options_pb2.msgopt]
        .s,
    )
    self.assertEqual(
        'FieldAnnotation',
        field_descriptor.GetOptions()
        .Extensions[unittest_custom_options_pb2.fieldopt]
        .s,
    )
    self.assertEqual(
        'EnumAnnotation',
        enum_descriptor.GetOptions()
        .Extensions[unittest_custom_options_pb2.enumopt]
        .s,
    )
    self.assertEqual(
        'EnumValueAnnotation',
        enum_value_descriptor.GetOptions()
        .Extensions[unittest_custom_options_pb2.enumvalopt]
        .s,
    )
    self.assertEqual(
        'ServiceAnnotation',
        service_descriptor.GetOptions()
        .Extensions[unittest_custom_options_pb2.serviceopt]
        .s,
    )
    self.assertEqual(
        'MethodAnnotation',
        method_descriptor.GetOptions()
        .Extensions[unittest_custom_options_pb2.methodopt]
        .s,
    )

  def testNestedOptions(self):
    nested_message = (
        unittest_custom_options_pb2.NestedOptionType.NestedMessage.DESCRIPTOR
    )
    self.assertEqual(
        1001,
        nested_message.GetOptions().Extensions[
            unittest_custom_options_pb2.message_opt1
        ],
    )
    nested_field = nested_message.fields_by_name['nested_field']
    self.assertEqual(
        1002,
        nested_field.GetOptions().Extensions[
            unittest_custom_options_pb2.field_opt1
        ],
    )
    outer_message = unittest_custom_options_pb2.NestedOptionType.DESCRIPTOR
    nested_enum = outer_message.enum_types_by_name['NestedEnum']
    self.assertEqual(
        1003,
        nested_enum.GetOptions().Extensions[
            unittest_custom_options_pb2.enum_opt1
        ],
    )
    nested_enum_value = outer_message.enum_values_by_name['NESTED_ENUM_VALUE']
    self.assertEqual(
        1004,
        nested_enum_value.GetOptions().Extensions[
            unittest_custom_options_pb2.enum_value_opt1
        ],
    )
    nested_extension = outer_message.extensions_by_name['nested_extension']
    self.assertEqual(
        1005,
        nested_extension.GetOptions().Extensions[
            unittest_custom_options_pb2.field_opt2
        ],
    )

  def testFileDescriptorReferences(self):
    self.assertEqual(self.my_enum.file, self.my_file)
    self.assertEqual(self.my_message.file, self.my_file)

  def testFileDescriptor(self):
    self.assertEqual(self.my_file.name, 'some/filename/some.proto')
    self.assertEqual(self.my_file.package, 'proto2_unittest')
    self.assertEqual(self.my_file.pool, self.pool)
    self.assertFalse(self.my_file.has_options)
    file_proto = descriptor_pb2.FileDescriptorProto()
    self.my_file.CopyToProto(file_proto)
    self.assertEqual(self.my_file.serialized_pb, file_proto.SerializeToString())
    # Generated modules also belong to the default pool.
    self.assertEqual(unittest_pb2.DESCRIPTOR.pool, descriptor_pool.Default())

  @unittest.skipIf(
      api_implementation.Type() == 'python',
      'Immutability of descriptors is only enforced in c++ and upb backends',
  )
  def testImmutableCppDescriptor(self):
    file_descriptor = unittest_pb2.DESCRIPTOR
    message_descriptor = unittest_pb2.TestAllTypes.DESCRIPTOR
    field_descriptor = message_descriptor.fields_by_name['optional_int32']
    enum_descriptor = message_descriptor.enum_types_by_name['NestedEnum']
    oneof_descriptor = message_descriptor.oneofs_by_name['oneof_field']
    with self.assertRaises(AttributeError):
      message_descriptor.fields_by_name = None
    with self.assertRaises(TypeError):
      message_descriptor.fields_by_name['Another'] = None
    with self.assertRaises(TypeError):
      message_descriptor.fields.append(None)
    with self.assertRaises(AttributeError):
      field_descriptor.containing_type = message_descriptor
    with self.assertRaises(AttributeError):
      file_descriptor.has_options = False
    with self.assertRaises(AttributeError):
      field_descriptor.has_options = False
    with self.assertRaises(AttributeError):
      oneof_descriptor.has_options = False
    with self.assertRaises(AttributeError):
      enum_descriptor.has_options = False
    with self.assertRaises(AttributeError) as e:
      message_descriptor.has_options = True

    if api_implementation.Type() == 'cpp':
      self.assertEqual(
          'attribute is not writable: has_options', str(e.exception)
      )
    else:
      self.assertEqual(api_implementation.Type(), 'upb')
      self.assertEqual(
          "attribute 'has_options' of "
          "'google._upb._message.Descriptor' "
          'objects is not writable',
          str(e.exception),
      )

  def testDefault(self):
    message_descriptor = unittest_pb2.TestAllTypes.DESCRIPTOR
    field = message_descriptor.fields_by_name['repeated_int32']
    self.assertEqual(field.default_value, [])
    self.assertTrue(field.is_repeated)
    field = message_descriptor.fields_by_name['repeated_nested_message']
    self.assertEqual(field.default_value, [])
    self.assertTrue(field.is_repeated)
    field = message_descriptor.fields_by_name['optionalgroup']
    self.assertEqual(field.default_value, None)
    self.assertFalse(field.is_required)
    field = message_descriptor.fields_by_name['optional_nested_message']
    self.assertEqual(field.default_value, None)
    self.assertFalse(field.is_required)


class NewDescriptorTest(DescriptorTest):
  """Redo the same tests as above, but with a separate DescriptorPool."""

  def GetDescriptorPool(self):
    return descriptor_pool.DescriptorPool()


class GeneratedDescriptorTest(unittest.TestCase):
  """Tests for the properties of descriptors in generated code."""

  def CheckMessageDescriptor(self, message_descriptor):
    # Basic properties
    self.assertEqual(message_descriptor.name, 'TestAllTypes')
    self.assertEqual(
        message_descriptor.full_name, 'proto2_unittest.TestAllTypes'
    )
    # Test equality and hashability
    self.assertEqual(message_descriptor, message_descriptor)
    self.assertEqual(
        message_descriptor.fields[0].containing_type, message_descriptor
    )
    self.assertIn(message_descriptor, [message_descriptor])
    self.assertIn(message_descriptor, {message_descriptor: None})
    # Test field containers
    self.CheckDescriptorSequence(message_descriptor.fields)
    self.CheckDescriptorMapping(message_descriptor.fields_by_name)
    self.CheckDescriptorMapping(message_descriptor.fields_by_number)
    self.CheckDescriptorMapping(message_descriptor.fields_by_camelcase_name)
    self.CheckDescriptorMapping(message_descriptor.enum_types_by_name)
    self.CheckDescriptorMapping(message_descriptor.enum_values_by_name)
    self.CheckDescriptorMapping(message_descriptor.oneofs_by_name)
    self.CheckDescriptorMapping(message_descriptor.enum_types[0].values_by_name)
    # Test extension range
    self.assertEqual(message_descriptor.extension_ranges, [])

  def CheckFieldDescriptor(self, field_descriptor):
    # Basic properties
    self.assertEqual(field_descriptor.name, 'optional_int32')
    self.assertEqual(field_descriptor.camelcase_name, 'optionalInt32')
    self.assertEqual(
        field_descriptor.full_name,
        'proto2_unittest.TestAllTypes.optional_int32',
    )
    self.assertEqual(field_descriptor.containing_type.name, 'TestAllTypes')
    self.assertEqual(field_descriptor.file, unittest_pb2.DESCRIPTOR)
    # Test equality and hashability
    self.assertEqual(field_descriptor, field_descriptor)
    self.assertEqual(
        field_descriptor.containing_type.fields_by_name['optional_int32'],
        field_descriptor,
    )
    self.assertEqual(
        field_descriptor.containing_type.fields_by_camelcase_name[
            'optionalInt32'
        ],
        field_descriptor,
    )
    self.assertIn(field_descriptor, [field_descriptor])
    self.assertIn(field_descriptor, {field_descriptor: None})
    self.assertEqual(None, field_descriptor.extension_scope)
    self.assertEqual(None, field_descriptor.enum_type)
    self.assertFalse(field_descriptor.is_required)
    self.assertTrue(field_descriptor.has_presence)
    if api_implementation.Type() == 'cpp':
      # For test coverage only
      self.assertEqual(field_descriptor.id, field_descriptor.id)

  def CheckDescriptorSequence(self, sequence):
    # Verifies that a property like 'messageDescriptor.fields' has all the
    # properties of an immutable abc.Sequence.
    self.assertNotEqual(
        sequence, unittest_pb2.TestAllExtensions.DESCRIPTOR.fields
    )
    self.assertNotEqual(sequence, [])
    self.assertNotEqual(sequence, 1)
    self.assertFalse(sequence == 1)  # Only for cpp test coverage
    self.assertEqual(sequence, sequence)
    expected_list = list(sequence)
    self.assertEqual(expected_list, sequence)
    self.assertGreater(len(sequence), 0)  # Sized
    self.assertEqual(len(sequence), len(expected_list))  # Iterable
    self.assertEqual(sequence[len(sequence) - 1], sequence[-1])
    item = sequence[0]
    self.assertEqual(item, sequence[0])
    self.assertIn(item, sequence)  # Container
    self.assertEqual(sequence.index(item), 0)
    self.assertEqual(sequence.count(item), 1)
    other_item = unittest_pb2.NestedTestAllTypes.DESCRIPTOR.fields[0]
    self.assertNotIn(other_item, sequence)
    self.assertEqual(sequence.count(other_item), 0)
    self.assertRaises(ValueError, sequence.index, other_item)
    self.assertRaises(ValueError, sequence.index, [])
    reversed_iterator = reversed(sequence)
    self.assertEqual(list(reversed_iterator), list(sequence)[::-1])
    self.assertRaises(StopIteration, next, reversed_iterator)
    expected_list[0] = 'change value'
    self.assertNotEqual(expected_list, sequence)
    # TODO: Change __repr__ support for DescriptorSequence.
    if api_implementation.Type() == 'python':
      self.assertEqual(str(list(sequence)), str(sequence))
    else:
      self.assertEqual(str(sequence)[0], '<')

  def CheckDescriptorMapping(self, mapping):
    # Verifies that a property like 'messageDescriptor.fields' has all the
    # properties of an immutable abc.Mapping.
    iterated_keys = []
    for key in mapping:
      iterated_keys.append(key)
    self.assertEqual(len(iterated_keys), len(mapping))
    self.assertEqual(set(iterated_keys), set(mapping.keys()))

    self.assertNotEqual(
        mapping, unittest_pb2.TestAllExtensions.DESCRIPTOR.fields_by_name
    )
    self.assertNotEqual(mapping, {})
    self.assertNotEqual(mapping, 1)
    self.assertFalse(mapping == 1)  # Only for cpp test coverage
    excepted_dict = dict(mapping.items())
    self.assertEqual(mapping, excepted_dict)
    self.assertEqual(mapping, mapping)
    self.assertGreater(len(mapping), 0)  # Sized
    self.assertEqual(len(mapping), len(excepted_dict))  # Iterable
    key, item = next(iter(mapping.items()))
    self.assertIn(key, mapping)  # Container
    self.assertEqual(mapping.get(key), item)
    with self.assertRaises(TypeError):
      mapping.get()
    # TODO: Fix python and cpp extension diff.
    if api_implementation.Type() == 'cpp':
      self.assertEqual(None, mapping.get([]))
    else:
      self.assertRaises(TypeError, mapping.get, [])
      with self.assertRaises(TypeError):
        if [] in mapping:
          pass
      with self.assertRaises(TypeError):
        _ = mapping[[]]
    # keys(), iterkeys() &co
    item = (next(iter(mapping.keys())), next(iter(mapping.values())))
    self.assertEqual(item, next(iter(mapping.items())))
    excepted_dict[key] = 'change value'
    self.assertNotEqual(mapping, excepted_dict)
    del excepted_dict[key]
    excepted_dict['new_key'] = 'new'
    self.assertNotEqual(mapping, excepted_dict)
    self.assertRaises(KeyError, mapping.__getitem__, 'key_error')
    self.assertRaises(KeyError, mapping.__getitem__, len(mapping) * 2)
    # TODO: Add __repr__ support for DescriptorMapping.
    if api_implementation.Type() == 'cpp':
      self.assertEqual(str(mapping)[0], '<')
    else:
      print(str(dict(mapping.items()))[:100])
      print(str(mapping)[:100])
      self.assertEqual(len(str(dict(mapping.items()))), len(str(mapping)))

  def testDescriptor(self):
    message_descriptor = unittest_pb2.TestAllTypes.DESCRIPTOR
    self.CheckMessageDescriptor(message_descriptor)
    field_descriptor = message_descriptor.fields_by_name['optional_int32']
    self.CheckFieldDescriptor(field_descriptor)
    field_descriptor = message_descriptor.fields_by_camelcase_name[
        'optionalInt32'
    ]
    self.CheckFieldDescriptor(field_descriptor)
    enum_descriptor = unittest_pb2.DESCRIPTOR.enum_types_by_name['ForeignEnum']
    self.assertEqual(None, enum_descriptor.containing_type)
    # Test extension range
    self.assertEqual(
        unittest_pb2.TestAllExtensions.DESCRIPTOR.extension_ranges,
        [(1, 536870912)],
    )
    self.assertEqual(
        unittest_pb2.TestMultipleExtensionRanges.DESCRIPTOR.extension_ranges,
        [(42, 43), (4143, 4244), (65536, 536870912)],
    )

  def testCppDescriptorContainer(self):
    containing_file = unittest_pb2.DESCRIPTOR
    self.CheckDescriptorSequence(containing_file.dependencies)
    self.CheckDescriptorMapping(containing_file.message_types_by_name)
    self.CheckDescriptorMapping(containing_file.enum_types_by_name)
    self.CheckDescriptorMapping(containing_file.services_by_name)
    self.CheckDescriptorMapping(containing_file.extensions_by_name)
    self.CheckDescriptorMapping(
        unittest_pb2.TestNestedExtension.DESCRIPTOR.extensions_by_name
    )

  def testCppDescriptorContainer_Iterator(self):
    # Same test with the iterator
    enum = unittest_pb2.TestAllTypes.DESCRIPTOR.enum_types_by_name['NestedEnum']
    values_iter = iter(enum.values)
    del enum
    self.assertEqual('FOO', next(values_iter).name)

  def testDescriptorNestedTypesContainer(self):
    message_descriptor = unittest_pb2.TestAllTypes.DESCRIPTOR
    nested_message_descriptor = (
        unittest_pb2.TestAllTypes.NestedMessage.DESCRIPTOR
    )
    self.assertEqual(len(message_descriptor.nested_types), 3)
    self.assertFalse(None in message_descriptor.nested_types)
    self.assertTrue(
        nested_message_descriptor in message_descriptor.nested_types
    )

  def testServiceDescriptor(self):
    service_descriptor = unittest_pb2.DESCRIPTOR.services_by_name['TestService']
    self.assertEqual(service_descriptor.name, 'TestService')
    self.assertEqual(service_descriptor.methods[0].name, 'Foo')
    self.assertIs(service_descriptor.file, unittest_pb2.DESCRIPTOR)
    self.assertEqual(service_descriptor.index, 0)
    self.CheckDescriptorMapping(service_descriptor.methods_by_name)
    self.assertFalse(service_descriptor.has_options)

  def testOneofDescriptor(self):
    message_descriptor = unittest_pb2.TestAllTypes.DESCRIPTOR
    oneof_descriptor = message_descriptor.oneofs_by_name['oneof_field']
    self.assertFalse(oneof_descriptor.has_options)
    self.assertEqual(message_descriptor, oneof_descriptor.containing_type)
    self.assertEqual('oneof_field', oneof_descriptor.name)
    self.assertEqual(
        'proto2_unittest.TestAllTypes.oneof_field', oneof_descriptor.full_name
    )
    self.assertEqual(0, oneof_descriptor.index)

  def testDescriptorSlice(self):
    message_descriptor = unittest_pb2.TestAllTypes.DESCRIPTOR
    nested = message_descriptor.nested_types[:]
    self.assertEqual(message_descriptor.nested_types, nested)
    fields = message_descriptor.fields
    fields_list = list(fields)
    self.assertEqual(fields_list[:], fields[:])
    self.assertEqual(fields_list[2::2], fields[2::2])
    self.assertEqual(fields_list[3:19:3], fields[3:19:3])


class DescriptorCopyToProtoTest(unittest.TestCase):
  """Tests for CopyTo functions of Descriptor."""

  def _AssertProtoEqual(self, actual_proto, expected_class, expected_ascii):
    expected_proto = expected_class()
    text_format.Merge(expected_ascii, expected_proto)

    self.assertEqual(
        actual_proto,
        expected_proto,
        'Not equal,\nActual:\n%s\nExpected:\n%s\n'
        % (str(actual_proto), str(expected_proto)),
    )

  def _InternalTestCopyToProto(
      self, desc, expected_proto_class, expected_proto_ascii
  ):
    actual = expected_proto_class()
    desc.CopyToProto(actual)
    self._AssertProtoEqual(actual, expected_proto_class, expected_proto_ascii)

  def testCopyToProto_EmptyMessage(self):
    self._InternalTestCopyToProto(
        unittest_pb2.TestEmptyMessage.DESCRIPTOR,
        descriptor_pb2.DescriptorProto,
        TEST_EMPTY_MESSAGE_DESCRIPTOR_ASCII,
    )

  def testCopyToProto_NestedMessage(self):
    TEST_NESTED_MESSAGE_ASCII = """
      name: 'NestedMessage'
      field: <
        name: 'bb'
        number: 1
        label: 1  # Optional
        type: 5  # TYPE_INT32
      >
      """

    self._InternalTestCopyToProto(
        unittest_pb2.TestAllTypes.NestedMessage.DESCRIPTOR,
        descriptor_pb2.DescriptorProto,
        TEST_NESTED_MESSAGE_ASCII,
    )

  def testCopyToProto_ForeignNestedMessage(self):
    TEST_FOREIGN_NESTED_ASCII = """
      name: 'TestForeignNested'
      field: <
        name: 'foreign_nested'
        number: 1
        label: 1  # Optional
        type: 11  # TYPE_MESSAGE
        type_name: '.proto2_unittest.TestAllTypes.NestedMessage'
      >
      """

    self._InternalTestCopyToProto(
        unittest_pb2.TestForeignNested.DESCRIPTOR,
        descriptor_pb2.DescriptorProto,
        TEST_FOREIGN_NESTED_ASCII,
    )

  def testCopyToProto_ForeignEnum(self):
    TEST_FOREIGN_ENUM_ASCII = """
      name: 'ForeignEnum'
      value: <
        name: 'FOREIGN_FOO'
        number: 4
      >
      value: <
        name: 'FOREIGN_BAR'
        number: 5
      >
      value: <
        name: 'FOREIGN_BAZ'
        number: 6
      >
      value: <
        name: 'FOREIGN_BAX'
        number: 32
      >
      value: <
        name: 'FOREIGN_LARGE'
        number: 123456
      >
      """

    self._InternalTestCopyToProto(
        unittest_pb2.ForeignEnum.DESCRIPTOR,
        descriptor_pb2.EnumDescriptorProto,
        TEST_FOREIGN_ENUM_ASCII,
    )

  def testCopyToProto_Options(self):
    TEST_DEPRECATED_FIELDS_ASCII = """
      name: 'TestDeprecatedFields'
      field: <
        name: 'deprecated_int32'
        number: 1
        label: 1  # Optional
        type: 5  # TYPE_INT32
        options: <
          deprecated: true
        >
      >
      field: {
        name: 'deprecated_repeated_string'
        number: 4
        label: LABEL_REPEATED
        type: TYPE_STRING
        options: {
          deprecated: true
        }
      }
      field {
        name: "deprecated_message"
        number: 3
        label: LABEL_OPTIONAL
        type: TYPE_MESSAGE
        type_name: ".proto2_unittest.TestAllTypes.NestedMessage"
        options {
          deprecated: true
        }
      }
      field {
        name: "deprecated_int32_in_oneof"
        number: 2
        label: LABEL_OPTIONAL
        type: TYPE_INT32
        options {
          deprecated: true
        }
        oneof_index: 0
      }
      field {
        name: "nested"
        number: 5
        label: LABEL_OPTIONAL
        type: TYPE_MESSAGE
        type_name: ".proto2_unittest.TestDeprecatedFields"
      }
      oneof_decl {
        name: "oneof_fields"
      }
      """

    self._InternalTestCopyToProto(
        unittest_pb2.TestDeprecatedFields.DESCRIPTOR,
        descriptor_pb2.DescriptorProto,
        TEST_DEPRECATED_FIELDS_ASCII,
    )

  def testCopyToProto_AllExtensions(self):
    TEST_EMPTY_MESSAGE_WITH_EXTENSIONS_ASCII = """
      name: 'TestEmptyMessageWithExtensions'
      extension_range: <
        start: 1
        end: 536870912
      >
      """

    self._InternalTestCopyToProto(
        unittest_pb2.TestEmptyMessageWithExtensions.DESCRIPTOR,
        descriptor_pb2.DescriptorProto,
        TEST_EMPTY_MESSAGE_WITH_EXTENSIONS_ASCII,
    )

  def testCopyToProto_SeveralExtensions(self):
    TEST_MESSAGE_WITH_SEVERAL_EXTENSIONS_ASCII = """
      name: 'TestMultipleExtensionRanges'
      extension_range: <
        start: 42
        end: 43
      >
      extension_range: <
        start: 4143
        end: 4244
      >
      extension_range: <
        start: 65536
        end: 536870912
      >
      """

    self._InternalTestCopyToProto(
        unittest_pb2.TestMultipleExtensionRanges.DESCRIPTOR,
        descriptor_pb2.DescriptorProto,
        TEST_MESSAGE_WITH_SEVERAL_EXTENSIONS_ASCII,
    )

  def testCopyToProto_FileDescriptor(self):
    UNITTEST_IMPORT_FILE_DESCRIPTOR_ASCII = (
        """
      name: 'google/protobuf/unittest_import.proto'
      package: 'proto2_unittest_import'
      dependency: 'google/protobuf/unittest_import_public.proto'
      message_type: <
        name: 'ImportMessage'
        field: <
          name: 'd'
          number: 1
          label: 1  # Optional
          type: 5  # TYPE_INT32
        >
      >
      """
        +
        """enum_type: <
        name: 'ImportEnum'
        value: <
          name: 'IMPORT_FOO'
          number: 7
        >
        value: <
          name: 'IMPORT_BAR'
          number: 8
        >
        value: <
          name: 'IMPORT_BAZ'
          number: 9
        >
      >
      enum_type: <
        name: 'ImportEnumForMap'
        value: <
          name: 'UNKNOWN'
          number: 0
        >
        value: <
          name: 'FOO'
          number: 1
        >
        value: <
          name: 'BAR'
          number: 2
        >
      >
      options: <
        java_package: 'com.google.protobuf.test'
        optimize_for: 1  # SPEED
      """
        +
        """
        cc_enable_arenas: true
      >
      public_dependency: 0
    """
    )
    self._InternalTestCopyToProto(
        unittest_import_pb2.DESCRIPTOR,
        descriptor_pb2.FileDescriptorProto,
        UNITTEST_IMPORT_FILE_DESCRIPTOR_ASCII,
    )

  def testCopyToProto_ServiceDescriptor(self):
    TEST_SERVICE_ASCII = """
      name: 'TestService'
      method: <
        name: 'Foo'
        input_type: '.proto2_unittest.FooRequest'
        output_type: '.proto2_unittest.FooResponse'
      >
      method: <
        name: 'Bar'
        input_type: '.proto2_unittest.BarRequest'
        output_type: '.proto2_unittest.BarResponse'
      >
      """
    self._InternalTestCopyToProto(
        unittest_pb2.TestService.DESCRIPTOR,
        descriptor_pb2.ServiceDescriptorProto,
        TEST_SERVICE_ASCII,
    )

  def testCopyToProto_MethodDescriptor(self):
    expected_ascii = """
      name: 'Foo'
      input_type: '.proto2_unittest.FooRequest'
      output_type: '.proto2_unittest.FooResponse'
    """
    method_descriptor = unittest_pb2.TestService.DESCRIPTOR.FindMethodByName(
        'Foo'
    )
    self._InternalTestCopyToProto(
        method_descriptor, descriptor_pb2.MethodDescriptorProto, expected_ascii
    )

  def testCopyToProto_TypeError(self):
    file_proto = descriptor_pb2.FileDescriptorProto()
    self.assertRaises(
        TypeError,
        unittest_pb2.TestEmptyMessage.DESCRIPTOR.CopyToProto,
        file_proto,
    )
    self.assertRaises(
        TypeError, unittest_pb2.ForeignEnum.DESCRIPTOR.CopyToProto, file_proto
    )
    self.assertRaises(
        TypeError, unittest_pb2.TestService.DESCRIPTOR.CopyToProto, file_proto
    )
    self.assertRaises(
        TypeError,
        unittest_pb2.TestService.DESCRIPTOR.FindMethodByName('Foo').CopyToProto,
        file_proto,
    )
    proto = descriptor_pb2.DescriptorProto()
    self.assertRaises(
        TypeError, unittest_import_pb2.DESCRIPTOR.CopyToProto, proto
    )


class MakeDescriptorTest(unittest.TestCase):

  def testMakeDescriptorWithNestedFields(self):
    file_descriptor_proto = descriptor_pb2.FileDescriptorProto()
    file_descriptor_proto.name = 'Foo2'
    message_type = file_descriptor_proto.message_type.add()
    message_type.name = file_descriptor_proto.name
    nested_type = message_type.nested_type.add()
    nested_type.name = 'Sub'
    enum_type = nested_type.enum_type.add()
    enum_type.name = 'FOO'
    enum_type_val = enum_type.value.add()
    enum_type_val.name = 'BAR'
    enum_type_val.number = 3
    field = message_type.field.add()
    field.number = 1
    field.name = 'uint64_field'
    field.label = descriptor.FieldDescriptor.LABEL_REQUIRED
    field.type = descriptor.FieldDescriptor.TYPE_UINT64
    field = message_type.field.add()
    field.number = 2
    field.name = 'nested_message_field'
    field.label = descriptor.FieldDescriptor.LABEL_REQUIRED
    field.type = descriptor.FieldDescriptor.TYPE_MESSAGE
    field.type_name = 'Sub'
    enum_field = nested_type.field.add()
    enum_field.number = 2
    enum_field.name = 'bar_field'
    enum_field.label = descriptor.FieldDescriptor.LABEL_REQUIRED
    enum_field.type = descriptor.FieldDescriptor.TYPE_ENUM
    enum_field.type_name = 'Foo2.Sub.FOO'

    result = descriptor.MakeDescriptor(message_type)
    self.assertEqual(
        result.fields[0].cpp_type, descriptor.FieldDescriptor.CPPTYPE_UINT64
    )
    self.assertEqual(result.fields[0].cpp_type, result.fields[0].CPPTYPE_UINT64)
    self.assertEqual(
        result.fields[1].cpp_type, descriptor.FieldDescriptor.CPPTYPE_MESSAGE
    )
    self.assertEqual(
        result.fields[1].cpp_type, result.fields[1].CPPTYPE_MESSAGE
    )
    self.assertEqual(result.fields[1].message_type.containing_type, result)
    self.assertEqual(
        result.nested_types[0].fields[0].full_name, 'Foo2.Sub.bar_field'
    )
    self.assertEqual(
        result.nested_types[0].fields[0].enum_type,
        result.nested_types[0].enum_types[0],
    )
    self.assertFalse(result.has_options)
    self.assertFalse(result.fields[0].has_options)
    if api_implementation.Type() == 'cpp':
      with self.assertRaises(AttributeError):
        result.fields[0].has_options = False

  def testMakeDescriptorWithUnsignedIntField(self):
    file_descriptor_proto = descriptor_pb2.FileDescriptorProto()
    file_descriptor_proto.name = 'Foo'
    message_type = file_descriptor_proto.message_type.add()
    message_type.name = file_descriptor_proto.name
    enum_type = message_type.enum_type.add()
    enum_type.name = 'FOO'
    enum_type_val = enum_type.value.add()
    enum_type_val.name = 'BAR'
    enum_type_val.number = 3
    field = message_type.field.add()
    field.number = 1
    field.name = 'uint64_field'
    field.label = descriptor.FieldDescriptor.LABEL_REQUIRED
    field.type = descriptor.FieldDescriptor.TYPE_UINT64
    enum_field = message_type.field.add()
    enum_field.number = 2
    enum_field.name = 'bar_field'
    enum_field.label = descriptor.FieldDescriptor.LABEL_REQUIRED
    enum_field.type = descriptor.FieldDescriptor.TYPE_ENUM
    enum_field.type_name = 'Foo.FOO'

    result = descriptor.MakeDescriptor(message_type)
    self.assertEqual(
        result.fields[0].cpp_type, descriptor.FieldDescriptor.CPPTYPE_UINT64
    )

  def testMakeDescriptorWithOptions(self):
    descriptor_proto = descriptor_pb2.DescriptorProto()
    aggregate_message = unittest_custom_options_pb2.AggregateMessage
    aggregate_message.DESCRIPTOR.CopyToProto(descriptor_proto)
    reformed_descriptor = descriptor.MakeDescriptor(descriptor_proto)

    options = reformed_descriptor.GetOptions()
    self.assertEqual(
        101, options.Extensions[unittest_custom_options_pb2.msgopt].i
    )

  def testCamelcaseName(self):
    descriptor_proto = descriptor_pb2.DescriptorProto()
    descriptor_proto.options.deprecated_legacy_json_field_conflicts = True
    descriptor_proto.name = 'Bar'
    names = ['foo_foo', 'FooBar', 'fooBaz', 'fooFoo', 'foobar']
    camelcase_names = ['fooFoo', 'fooBar', 'fooBaz', 'fooFoo', 'foobar']
    for index in range(len(names)):
      field = descriptor_proto.field.add()
      field.number = index + 1
      field.name = names[index]
    result = descriptor.MakeDescriptor(descriptor_proto)
    for index in range(len(camelcase_names)):
      self.assertEqual(
          result.fields[index].camelcase_name, camelcase_names[index]
      )

  def testJsonName(self):
    descriptor_proto = descriptor_pb2.DescriptorProto()
    descriptor_proto.options.deprecated_legacy_json_field_conflicts = True
    descriptor_proto.name = 'TestJsonName'
    names = [
        'field_name',
        'fieldName',
        'FieldName',
        '_field_name',
        'FIELD_NAME',
        'json_name',
    ]
    json_names = [
        'fieldName',
        'fieldName',
        'FieldName',
        'FieldName',
        'FIELDNAME',
        '@type',
    ]
    for index in range(len(names)):
      field = descriptor_proto.field.add()
      field.number = index + 1
      field.name = names[index]
    field.json_name = '@type'
    result = descriptor.MakeDescriptor(descriptor_proto)
    for index in range(len(json_names)):
      self.assertEqual(result.fields[index].json_name, json_names[index])


@testing_refleaks.TestCase
class FeaturesTest(parameterized.TestCase):

  @parameterized.named_parameters([
      ('File', lambda: descriptor_pb2.DESCRIPTOR),
      ('Message', lambda: descriptor_pb2.FeatureSet.DESCRIPTOR),
      (
          'Enum',
          lambda: descriptor_pb2.FeatureSet.FieldPresence.DESCRIPTOR,
      ),
      (
          'Field',
          lambda: descriptor_pb2.FeatureSet.DESCRIPTOR.fields_by_name[
              'enum_type'
          ],
      ),
  ])
  def testDescriptorProtoDefaultFeatures(self, desc):
    self.assertEqual(
        desc()._GetFeatures().field_presence,
        descriptor_pb2.FeatureSet.FieldPresence.EXPLICIT,
    )
    self.assertEqual(
        desc()._GetFeatures().enum_type,
        descriptor_pb2.FeatureSet.EnumType.CLOSED,
    )
    self.assertEqual(
        desc()._GetFeatures().repeated_field_encoding,
        descriptor_pb2.FeatureSet.RepeatedFieldEncoding.EXPANDED,
    )

  def testDescriptorProtoOverrideFeatures(self):
    desc = descriptor_pb2.SourceCodeInfo.Location.DESCRIPTOR.fields_by_name[
        'path'
    ]
    self.assertEqual(
        desc._GetFeatures().field_presence,
        descriptor_pb2.FeatureSet.FieldPresence.EXPLICIT,
    )
    self.assertEqual(
        desc._GetFeatures().enum_type,
        descriptor_pb2.FeatureSet.EnumType.CLOSED,
    )
    self.assertEqual(
        desc._GetFeatures().repeated_field_encoding,
        descriptor_pb2.FeatureSet.RepeatedFieldEncoding.PACKED,
    )

  def testFeaturesStripped(self):
    desc = unittest_legacy_features_pb2.TestEditionsMessage.DESCRIPTOR.fields_by_name[
        'required_field'
    ]
    self.assertFalse(desc.GetOptions().HasField('features'))

  def testLegacyRequiredTransform(self):
    desc = unittest_legacy_features_pb2.TestEditionsMessage.DESCRIPTOR
    self.assertTrue(desc.fields_by_name['required_field'].is_required)

  def testLegacyGroupTransform(self):
    desc = unittest_legacy_features_pb2.TestEditionsMessage.DESCRIPTOR
    self.assertEqual(
        desc.fields_by_name['delimited_field'].type,
        descriptor.FieldDescriptor.TYPE_GROUP,
    )

  def testLegacyInferRequired(self):
    desc = unittest_pb2.TestRequired.DESCRIPTOR.fields_by_name['a']
    self.assertEqual(
        desc._GetFeatures().field_presence,
        descriptor_pb2.FeatureSet.FieldPresence.LEGACY_REQUIRED,
    )

  def testLegacyInferGroup(self):
    desc = unittest_pb2.TestAllTypes.DESCRIPTOR.fields_by_name['optionalgroup']
    self.assertEqual(
        desc._GetFeatures().message_encoding,
        descriptor_pb2.FeatureSet.MessageEncoding.DELIMITED,
    )

  def testLegacyInferProto2Packed(self):
    desc = unittest_pb2.TestPackedTypes.DESCRIPTOR.fields_by_name[
        'packed_int32'
    ]
    self.assertEqual(
        desc._GetFeatures().repeated_field_encoding,
        descriptor_pb2.FeatureSet.RepeatedFieldEncoding.PACKED,
    )

  def testLegacyInferProto3Expanded(self):
    desc = unittest_proto3_pb2.TestUnpackedTypes.DESCRIPTOR.fields_by_name[
        'repeated_int32'
    ]
    self.assertEqual(
        desc._GetFeatures().repeated_field_encoding,
        descriptor_pb2.FeatureSet.RepeatedFieldEncoding.EXPANDED,
    )

  def testProto2Defaults(self):
    features = test_proto2_pb2.TestProto2.DESCRIPTOR.fields_by_name[
        'optional_int32'
    ]._GetFeatures()
    fs = descriptor_pb2.FeatureSet
    self.assertEqual(features.field_presence, fs.FieldPresence.EXPLICIT)
    self.assertEqual(features.enum_type, fs.EnumType.CLOSED)
    self.assertEqual(
        features.repeated_field_encoding, fs.RepeatedFieldEncoding.EXPANDED
    )
    self.assertEqual(features.utf8_validation, fs.Utf8Validation.NONE)
    self.assertEqual(
        features.message_encoding, fs.MessageEncoding.LENGTH_PREFIXED
    )
    self.assertEqual(features.json_format, fs.JsonFormat.LEGACY_BEST_EFFORT)

  def testProto3Defaults(self):
    features = unittest_proto3_pb2.TestAllTypes.DESCRIPTOR.fields_by_name[
        'optional_int32'
    ]._GetFeatures()
    fs = descriptor_pb2.FeatureSet
    self.assertEqual(features.field_presence, fs.FieldPresence.IMPLICIT)
    self.assertEqual(features.enum_type, fs.EnumType.OPEN)
    self.assertEqual(
        features.repeated_field_encoding, fs.RepeatedFieldEncoding.PACKED
    )
    self.assertEqual(features.utf8_validation, fs.Utf8Validation.VERIFY)
    self.assertEqual(
        features.message_encoding, fs.MessageEncoding.LENGTH_PREFIXED
    )
    self.assertEqual(features.json_format, fs.JsonFormat.ALLOW)

  def testProto3ExtensionPresence(self):
    ext = unittest_proto3_extensions_pb2.Proto3FileExtensions.singular_int
    file = descriptor_pb2.FileDescriptorProto()

    self.assertFalse(file.options.HasExtension(ext))

    file.options.Extensions[ext] = 1

    self.assertTrue(file.options.HasExtension(ext))

  def testProto3ExtensionHasPresence(self):
    exts = unittest_proto3_extensions_pb2.Proto3FileExtensions
    self.assertTrue(exts.singular_int.has_presence)
    self.assertFalse(exts.repeated_int.has_presence)


def GetTestFeature(desc):
  return (
      desc._GetFeatures()
      .Extensions[unittest_features_pb2.test]
      .multiple_feature
  )


def SetTestFeature(proto, value):
  proto.options.features.Extensions[
      unittest_features_pb2.test
  ].multiple_feature = value


@testing_refleaks.TestCase
class FeatureInheritanceTest(unittest.TestCase):

  def setUp(self):
    super(FeatureInheritanceTest, self).setUp()
    self.file_proto = descriptor_pb2.FileDescriptorProto(
        name='some/filename/some.proto',
        package='proto2_unittest',
        edition=descriptor_pb2.Edition.EDITION_2023,
        syntax='editions',
    )
    self.top_extension_proto = self.file_proto.extension.add(
        name='top_extension',
        number=10,
        type=descriptor_pb2.FieldDescriptorProto.TYPE_INT32,
        label=descriptor_pb2.FieldDescriptorProto.LABEL_OPTIONAL,
        extendee='.proto2_unittest.TopMessage',
    )
    self.top_enum_proto = self.file_proto.enum_type.add(name='TopEnum')
    self.enum_value_proto = self.top_enum_proto.value.add(
        name='TOP_VALUE', number=0
    )
    self.top_message_proto = self.file_proto.message_type.add(name='TopMessage')
    self.field_proto = self.top_message_proto.field.add(
        name='field',
        number=1,
        type=descriptor_pb2.FieldDescriptorProto.TYPE_INT32,
        label=descriptor_pb2.FieldDescriptorProto.LABEL_OPTIONAL,
    )
    self.top_message_proto.extension_range.add(start=10, end=20)
    self.nested_extension_proto = self.top_message_proto.extension.add(
        name='nested_extension',
        number=11,
        type=descriptor_pb2.FieldDescriptorProto.TYPE_INT32,
        label=descriptor_pb2.FieldDescriptorProto.LABEL_OPTIONAL,
        extendee='.proto2_unittest.TopMessage',
    )
    self.nested_message_proto = self.top_message_proto.nested_type.add(
        name='NestedMessage'
    )
    self.nested_enum_proto = self.top_message_proto.enum_type.add(
        name='NestedEnum'
    )
    self.nested_enum_proto.value.add(name='NESTED_VALUE', number=0)
    self.oneof_proto = self.top_message_proto.oneof_decl.add(name='Oneof')
    self.oneof_field_proto = self.top_message_proto.field.add(
        name='oneof_field',
        number=2,
        type=descriptor_pb2.FieldDescriptorProto.TYPE_INT32,
        label=descriptor_pb2.FieldDescriptorProto.LABEL_OPTIONAL,
        oneof_index=0,
    )

    self.service_proto = self.file_proto.service.add(name='TestService')
    self.method_proto = self.service_proto.method.add(
        name='CallMethod',
        input_type='.proto2_unittest.TopMessage',
        output_type='.proto2_unittest.TopMessage',
    )

  def BuildPool(self):

    # These can't be put onto the fixture without breaking the refleak checks.
    class ReturnObject:
      pass

    ret = ReturnObject()
    ret.pool = descriptor_pool.DescriptorPool()

    defaults = descriptor_pb2.FeatureSetDefaults(
        defaults=[
            descriptor_pb2.FeatureSetDefaults.FeatureSetEditionDefault(
                edition=descriptor_pb2.Edition.EDITION_PROTO2,
                overridable_features=unittest_pb2.TestAllTypes.DESCRIPTOR._GetFeatures(),
            )
        ],
        minimum_edition=descriptor_pb2.Edition.EDITION_PROTO2,
        maximum_edition=descriptor_pb2.Edition.EDITION_2026,
    )
    defaults.defaults[0].overridable_features.Extensions[
        unittest_features_pb2.test
    ].multiple_feature = 1
    ret.pool.SetFeatureSetDefaults(defaults)

    # Add dependencies
    file = descriptor_pb2.FileDescriptorProto()
    descriptor_pb2.DESCRIPTOR.CopyToProto(file)
    ret.pool.Add(file)
    file.Clear()
    unittest_features_pb2.DESCRIPTOR.CopyToProto(file)
    ret.pool.Add(file)

    ret.file = ret.pool.AddSerializedFile(self.file_proto.SerializeToString())
    ret.top_message = ret.pool.FindMessageTypeByName(
        'proto2_unittest.TopMessage'
    )
    ret.top_enum = ret.pool.FindEnumTypeByName('proto2_unittest.TopEnum')
    ret.top_extension = ret.pool.FindExtensionByName(
        'proto2_unittest.top_extension'
    )
    ret.nested_message = ret.top_message.nested_types_by_name['NestedMessage']
    ret.nested_enum = ret.top_message.enum_types_by_name['NestedEnum']
    ret.nested_extension = ret.top_message.extensions_by_name[
        'nested_extension'
    ]
    ret.field = ret.top_message.fields_by_name['field']
    ret.oneof = ret.top_message.oneofs_by_name['Oneof']
    ret.oneof_field = ret.top_message.fields_by_name['oneof_field']
    ret.enum_value = ret.top_enum.values_by_name['TOP_VALUE']
    ret.service = ret.pool.FindServiceByName('proto2_unittest.TestService')
    ret.method = ret.service.methods_by_name['CallMethod']
    return ret

  def testFileDefaults(self):
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.file), 1)

  def testFileOverride(self):
    SetTestFeature(self.file_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.file), 3)

  def testFileMessageInherit(self):
    SetTestFeature(self.file_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.top_message), 3)

  def testFileMessageOverride(self):
    SetTestFeature(self.file_proto, 3)
    SetTestFeature(self.top_message_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.top_message), 5)

  def testFileEnumInherit(self):
    SetTestFeature(self.file_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.top_enum), 3)

  def testFileEnumOverride(self):
    SetTestFeature(self.file_proto, 3)
    SetTestFeature(self.top_enum_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.top_enum), 5)

  def testFileExtensionInherit(self):
    SetTestFeature(self.file_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.top_extension), 3)

  def testFileExtensionOverride(self):
    SetTestFeature(self.file_proto, 3)
    SetTestFeature(self.top_extension_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.top_extension), 5)

  def testFileServiceInherit(self):
    SetTestFeature(self.file_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.service), 3)

  def testFileServiceOverride(self):
    SetTestFeature(self.file_proto, 3)
    SetTestFeature(self.service_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.service), 5)

  def testMessageFieldInherit(self):
    SetTestFeature(self.top_message_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.field), 3)

  def testMessageFieldOverride(self):
    SetTestFeature(self.top_message_proto, 3)
    SetTestFeature(self.field_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.field), 5)

  def testMessageEnumInherit(self):
    SetTestFeature(self.top_message_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.nested_enum), 3)

  def testMessageEnumOverride(self):
    SetTestFeature(self.top_message_proto, 3)
    SetTestFeature(self.nested_enum_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.nested_enum), 5)

  def testMessageMessageInherit(self):
    SetTestFeature(self.top_message_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.nested_message), 3)

  def testMessageMessageOverride(self):
    SetTestFeature(self.top_message_proto, 3)
    SetTestFeature(self.nested_message_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.nested_message), 5)

  def testMessageExtensionInherit(self):
    SetTestFeature(self.top_message_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.nested_extension), 3)

  def testMessageExtensionOverride(self):
    SetTestFeature(self.top_message_proto, 3)
    SetTestFeature(self.nested_extension_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.nested_extension), 5)

  def testMessageOneofInherit(self):
    SetTestFeature(self.top_message_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.oneof), 3)

  def testMessageOneofOverride(self):
    SetTestFeature(self.top_message_proto, 3)
    SetTestFeature(self.oneof_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.oneof), 5)

  def testOneofFieldInherit(self):
    SetTestFeature(self.oneof_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.oneof_field), 3)

  def testOneofFieldOverride(self):
    SetTestFeature(self.oneof_proto, 3)
    SetTestFeature(self.oneof_field_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.oneof_field), 5)

  def testEnumValueInherit(self):
    SetTestFeature(self.top_enum_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.enum_value), 3)

  def testEnumValueOverride(self):
    SetTestFeature(self.top_enum_proto, 3)
    SetTestFeature(self.enum_value_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.enum_value), 5)

  def testServiceMethodInherit(self):
    SetTestFeature(self.service_proto, 3)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.method), 3)

  def testServiceMethodOverride(self):
    SetTestFeature(self.service_proto, 3)
    SetTestFeature(self.method_proto, 5)
    pool = self.BuildPool()
    self.assertEqual(GetTestFeature(pool.method), 5)


@testing_refleaks.TestCase
class ExtensionRangesRefleakTest(unittest.TestCase):

  def testExtensionRangesRefleak(self):
    _ = unittest_pb2.TestAllExtensions.DESCRIPTOR.extension_ranges
    _ = unittest_pb2.TestMultipleExtensionRanges.DESCRIPTOR.extension_ranges


if __name__ == '__main__':
  unittest.main()
