| Index: third_party/protobuf/python/google/protobuf/internal/symbol_database_test.py
|
| diff --git a/third_party/protobuf/python/google/protobuf/internal/symbol_database_test.py b/third_party/protobuf/python/google/protobuf/internal/symbol_database_test.py
|
| index 4f5173b2a26d8ce46a563a93bd9da401af02e466..c99b426dc4cd915b441d29b801d9b60555f9f6fc 100644
|
| --- a/third_party/protobuf/python/google/protobuf/internal/symbol_database_test.py
|
| +++ b/third_party/protobuf/python/google/protobuf/internal/symbol_database_test.py
|
| @@ -39,28 +39,26 @@ except ImportError:
|
|
|
| from google.protobuf import unittest_pb2
|
| from google.protobuf import descriptor
|
| -from google.protobuf import descriptor_pool
|
| from google.protobuf import symbol_database
|
|
|
| -
|
| class SymbolDatabaseTest(unittest.TestCase):
|
|
|
| def _Database(self):
|
| + # TODO(b/17734095): Remove this difference when the C++ implementation
|
| + # supports multiple databases.
|
| if descriptor._USE_C_DESCRIPTORS:
|
| - # The C++ implementation does not allow mixing descriptors from
|
| - # different pools.
|
| - db = symbol_database.SymbolDatabase(pool=descriptor_pool.Default())
|
| + return symbol_database.Default()
|
| else:
|
| db = symbol_database.SymbolDatabase()
|
| - # Register representative types from unittest_pb2.
|
| - db.RegisterFileDescriptor(unittest_pb2.DESCRIPTOR)
|
| - db.RegisterMessage(unittest_pb2.TestAllTypes)
|
| - db.RegisterMessage(unittest_pb2.TestAllTypes.NestedMessage)
|
| - db.RegisterMessage(unittest_pb2.TestAllTypes.OptionalGroup)
|
| - db.RegisterMessage(unittest_pb2.TestAllTypes.RepeatedGroup)
|
| - db.RegisterEnumDescriptor(unittest_pb2.ForeignEnum.DESCRIPTOR)
|
| - db.RegisterEnumDescriptor(unittest_pb2.TestAllTypes.NestedEnum.DESCRIPTOR)
|
| - return db
|
| + # Register representative types from unittest_pb2.
|
| + db.RegisterFileDescriptor(unittest_pb2.DESCRIPTOR)
|
| + db.RegisterMessage(unittest_pb2.TestAllTypes)
|
| + db.RegisterMessage(unittest_pb2.TestAllTypes.NestedMessage)
|
| + db.RegisterMessage(unittest_pb2.TestAllTypes.OptionalGroup)
|
| + db.RegisterMessage(unittest_pb2.TestAllTypes.RepeatedGroup)
|
| + db.RegisterEnumDescriptor(unittest_pb2.ForeignEnum.DESCRIPTOR)
|
| + db.RegisterEnumDescriptor(unittest_pb2.TestAllTypes.NestedEnum.DESCRIPTOR)
|
| + return db
|
|
|
| def testGetPrototype(self):
|
| instance = self._Database().GetPrototype(
|
|
|