Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
P
protobuf
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
protobuf
Commits
a2abc947
Commit
a2abc947
authored
Jan 13, 2015
by
Tres Seaver
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Straddle Python 2.6, 2.7, 3.3, 3.4.
parent
3d5460dd
Hide whitespace changes
Inline
Side-by-side
Showing
10 changed files
with
123 additions
and
123 deletions
+123
-123
descriptor_database_test.py
python/google/protobuf/internal/descriptor_database_test.py
+6
-6
descriptor_pool_test.py
python/google/protobuf/internal/descriptor_pool_test.py
+50
-50
descriptor_test.py
python/google/protobuf/internal/descriptor_test.py
+1
-1
generator_test.py
python/google/protobuf/internal/generator_test.py
+8
-8
message_factory_test.py
python/google/protobuf/internal/message_factory_test.py
+4
-4
message_test.py
python/google/protobuf/internal/message_test.py
+9
-9
reflection_test.py
python/google/protobuf/internal/reflection_test.py
+21
-21
symbol_database_test.py
python/google/protobuf/internal/symbol_database_test.py
+11
-11
text_encoding_test.py
python/google/protobuf/internal/text_encoding_test.py
+4
-4
text_format_test.py
python/google/protobuf/internal/text_format_test.py
+9
-9
No files found.
python/google/protobuf/internal/descriptor_database_test.py
View file @
a2abc947
...
@@ -49,17 +49,17 @@ class DescriptorDatabaseTest(unittest.TestCase):
...
@@ -49,17 +49,17 @@ class DescriptorDatabaseTest(unittest.TestCase):
factory_test2_pb2
.
DESCRIPTOR
.
serialized_pb
)
factory_test2_pb2
.
DESCRIPTOR
.
serialized_pb
)
db
.
Add
(
file_desc_proto
)
db
.
Add
(
file_desc_proto
)
self
.
assertEqual
s
(
file_desc_proto
,
db
.
FindFileByName
(
self
.
assertEqual
(
file_desc_proto
,
db
.
FindFileByName
(
'google/protobuf/internal/factory_test2.proto'
))
'google/protobuf/internal/factory_test2.proto'
))
self
.
assertEqual
s
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
self
.
assertEqual
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
'google.protobuf.python.internal.Factory2Message'
))
'google.protobuf.python.internal.Factory2Message'
))
self
.
assertEqual
s
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
self
.
assertEqual
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
'google.protobuf.python.internal.Factory2Message.NestedFactory2Message'
))
'google.protobuf.python.internal.Factory2Message.NestedFactory2Message'
))
self
.
assertEqual
s
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
self
.
assertEqual
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
'google.protobuf.python.internal.Factory2Enum'
))
'google.protobuf.python.internal.Factory2Enum'
))
self
.
assertEqual
s
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
self
.
assertEqual
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
'google.protobuf.python.internal.Factory2Message.NestedFactory2Enum'
))
'google.protobuf.python.internal.Factory2Message.NestedFactory2Enum'
))
self
.
assertEqual
s
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
self
.
assertEqual
(
file_desc_proto
,
db
.
FindFileContainingSymbol
(
'google.protobuf.python.internal.MessageWithNestedEnumOnly.NestedEnum'
))
'google.protobuf.python.internal.MessageWithNestedEnumOnly.NestedEnum'
))
if
__name__
==
'__main__'
:
if
__name__
==
'__main__'
:
...
...
python/google/protobuf/internal/descriptor_pool_test.py
View file @
a2abc947
...
@@ -65,15 +65,15 @@ class DescriptorPoolTest(unittest.TestCase):
...
@@ -65,15 +65,15 @@ class DescriptorPoolTest(unittest.TestCase):
name1
=
'google/protobuf/internal/factory_test1.proto'
name1
=
'google/protobuf/internal/factory_test1.proto'
file_desc1
=
self
.
pool
.
FindFileByName
(
name1
)
file_desc1
=
self
.
pool
.
FindFileByName
(
name1
)
self
.
assertIsInstance
(
file_desc1
,
descriptor
.
FileDescriptor
)
self
.
assertIsInstance
(
file_desc1
,
descriptor
.
FileDescriptor
)
self
.
assertEqual
s
(
name1
,
file_desc1
.
name
)
self
.
assertEqual
(
name1
,
file_desc1
.
name
)
self
.
assertEqual
s
(
'google.protobuf.python.internal'
,
file_desc1
.
package
)
self
.
assertEqual
(
'google.protobuf.python.internal'
,
file_desc1
.
package
)
self
.
assertIn
(
'Factory1Message'
,
file_desc1
.
message_types_by_name
)
self
.
assertIn
(
'Factory1Message'
,
file_desc1
.
message_types_by_name
)
name2
=
'google/protobuf/internal/factory_test2.proto'
name2
=
'google/protobuf/internal/factory_test2.proto'
file_desc2
=
self
.
pool
.
FindFileByName
(
name2
)
file_desc2
=
self
.
pool
.
FindFileByName
(
name2
)
self
.
assertIsInstance
(
file_desc2
,
descriptor
.
FileDescriptor
)
self
.
assertIsInstance
(
file_desc2
,
descriptor
.
FileDescriptor
)
self
.
assertEqual
s
(
name2
,
file_desc2
.
name
)
self
.
assertEqual
(
name2
,
file_desc2
.
name
)
self
.
assertEqual
s
(
'google.protobuf.python.internal'
,
file_desc2
.
package
)
self
.
assertEqual
(
'google.protobuf.python.internal'
,
file_desc2
.
package
)
self
.
assertIn
(
'Factory2Message'
,
file_desc2
.
message_types_by_name
)
self
.
assertIn
(
'Factory2Message'
,
file_desc2
.
message_types_by_name
)
def
testFindFileByNameFailure
(
self
):
def
testFindFileByNameFailure
(
self
):
...
@@ -84,17 +84,17 @@ class DescriptorPoolTest(unittest.TestCase):
...
@@ -84,17 +84,17 @@ class DescriptorPoolTest(unittest.TestCase):
file_desc1
=
self
.
pool
.
FindFileContainingSymbol
(
file_desc1
=
self
.
pool
.
FindFileContainingSymbol
(
'google.protobuf.python.internal.Factory1Message'
)
'google.protobuf.python.internal.Factory1Message'
)
self
.
assertIsInstance
(
file_desc1
,
descriptor
.
FileDescriptor
)
self
.
assertIsInstance
(
file_desc1
,
descriptor
.
FileDescriptor
)
self
.
assertEqual
s
(
'google/protobuf/internal/factory_test1.proto'
,
self
.
assertEqual
(
'google/protobuf/internal/factory_test1.proto'
,
file_desc1
.
name
)
file_desc1
.
name
)
self
.
assertEqual
s
(
'google.protobuf.python.internal'
,
file_desc1
.
package
)
self
.
assertEqual
(
'google.protobuf.python.internal'
,
file_desc1
.
package
)
self
.
assertIn
(
'Factory1Message'
,
file_desc1
.
message_types_by_name
)
self
.
assertIn
(
'Factory1Message'
,
file_desc1
.
message_types_by_name
)
file_desc2
=
self
.
pool
.
FindFileContainingSymbol
(
file_desc2
=
self
.
pool
.
FindFileContainingSymbol
(
'google.protobuf.python.internal.Factory2Message'
)
'google.protobuf.python.internal.Factory2Message'
)
self
.
assertIsInstance
(
file_desc2
,
descriptor
.
FileDescriptor
)
self
.
assertIsInstance
(
file_desc2
,
descriptor
.
FileDescriptor
)
self
.
assertEqual
s
(
'google/protobuf/internal/factory_test2.proto'
,
self
.
assertEqual
(
'google/protobuf/internal/factory_test2.proto'
,
file_desc2
.
name
)
file_desc2
.
name
)
self
.
assertEqual
s
(
'google.protobuf.python.internal'
,
file_desc2
.
package
)
self
.
assertEqual
(
'google.protobuf.python.internal'
,
file_desc2
.
package
)
self
.
assertIn
(
'Factory2Message'
,
file_desc2
.
message_types_by_name
)
self
.
assertIn
(
'Factory2Message'
,
file_desc2
.
message_types_by_name
)
def
testFindFileContainingSymbolFailure
(
self
):
def
testFindFileContainingSymbolFailure
(
self
):
...
@@ -105,72 +105,72 @@ class DescriptorPoolTest(unittest.TestCase):
...
@@ -105,72 +105,72 @@ class DescriptorPoolTest(unittest.TestCase):
msg1
=
self
.
pool
.
FindMessageTypeByName
(
msg1
=
self
.
pool
.
FindMessageTypeByName
(
'google.protobuf.python.internal.Factory1Message'
)
'google.protobuf.python.internal.Factory1Message'
)
self
.
assertIsInstance
(
msg1
,
descriptor
.
Descriptor
)
self
.
assertIsInstance
(
msg1
,
descriptor
.
Descriptor
)
self
.
assertEqual
s
(
'Factory1Message'
,
msg1
.
name
)
self
.
assertEqual
(
'Factory1Message'
,
msg1
.
name
)
self
.
assertEqual
s
(
'google.protobuf.python.internal.Factory1Message'
,
self
.
assertEqual
(
'google.protobuf.python.internal.Factory1Message'
,
msg1
.
full_name
)
msg1
.
full_name
)
self
.
assertEqual
s
(
None
,
msg1
.
containing_type
)
self
.
assertEqual
(
None
,
msg1
.
containing_type
)
nested_msg1
=
msg1
.
nested_types
[
0
]
nested_msg1
=
msg1
.
nested_types
[
0
]
self
.
assertEqual
s
(
'NestedFactory1Message'
,
nested_msg1
.
name
)
self
.
assertEqual
(
'NestedFactory1Message'
,
nested_msg1
.
name
)
self
.
assertEqual
s
(
msg1
,
nested_msg1
.
containing_type
)
self
.
assertEqual
(
msg1
,
nested_msg1
.
containing_type
)
nested_enum1
=
msg1
.
enum_types
[
0
]
nested_enum1
=
msg1
.
enum_types
[
0
]
self
.
assertEqual
s
(
'NestedFactory1Enum'
,
nested_enum1
.
name
)
self
.
assertEqual
(
'NestedFactory1Enum'
,
nested_enum1
.
name
)
self
.
assertEqual
s
(
msg1
,
nested_enum1
.
containing_type
)
self
.
assertEqual
(
msg1
,
nested_enum1
.
containing_type
)
self
.
assertEqual
s
(
nested_msg1
,
msg1
.
fields_by_name
[
self
.
assertEqual
(
nested_msg1
,
msg1
.
fields_by_name
[
'nested_factory_1_message'
]
.
message_type
)
'nested_factory_1_message'
]
.
message_type
)
self
.
assertEqual
s
(
nested_enum1
,
msg1
.
fields_by_name
[
self
.
assertEqual
(
nested_enum1
,
msg1
.
fields_by_name
[
'nested_factory_1_enum'
]
.
enum_type
)
'nested_factory_1_enum'
]
.
enum_type
)
msg2
=
self
.
pool
.
FindMessageTypeByName
(
msg2
=
self
.
pool
.
FindMessageTypeByName
(
'google.protobuf.python.internal.Factory2Message'
)
'google.protobuf.python.internal.Factory2Message'
)
self
.
assertIsInstance
(
msg2
,
descriptor
.
Descriptor
)
self
.
assertIsInstance
(
msg2
,
descriptor
.
Descriptor
)
self
.
assertEqual
s
(
'Factory2Message'
,
msg2
.
name
)
self
.
assertEqual
(
'Factory2Message'
,
msg2
.
name
)
self
.
assertEqual
s
(
'google.protobuf.python.internal.Factory2Message'
,
self
.
assertEqual
(
'google.protobuf.python.internal.Factory2Message'
,
msg2
.
full_name
)
msg2
.
full_name
)
self
.
assertIsNone
(
msg2
.
containing_type
)
self
.
assertIsNone
(
msg2
.
containing_type
)
nested_msg2
=
msg2
.
nested_types
[
0
]
nested_msg2
=
msg2
.
nested_types
[
0
]
self
.
assertEqual
s
(
'NestedFactory2Message'
,
nested_msg2
.
name
)
self
.
assertEqual
(
'NestedFactory2Message'
,
nested_msg2
.
name
)
self
.
assertEqual
s
(
msg2
,
nested_msg2
.
containing_type
)
self
.
assertEqual
(
msg2
,
nested_msg2
.
containing_type
)
nested_enum2
=
msg2
.
enum_types
[
0
]
nested_enum2
=
msg2
.
enum_types
[
0
]
self
.
assertEqual
s
(
'NestedFactory2Enum'
,
nested_enum2
.
name
)
self
.
assertEqual
(
'NestedFactory2Enum'
,
nested_enum2
.
name
)
self
.
assertEqual
s
(
msg2
,
nested_enum2
.
containing_type
)
self
.
assertEqual
(
msg2
,
nested_enum2
.
containing_type
)
self
.
assertEqual
s
(
nested_msg2
,
msg2
.
fields_by_name
[
self
.
assertEqual
(
nested_msg2
,
msg2
.
fields_by_name
[
'nested_factory_2_message'
]
.
message_type
)
'nested_factory_2_message'
]
.
message_type
)
self
.
assertEqual
s
(
nested_enum2
,
msg2
.
fields_by_name
[
self
.
assertEqual
(
nested_enum2
,
msg2
.
fields_by_name
[
'nested_factory_2_enum'
]
.
enum_type
)
'nested_factory_2_enum'
]
.
enum_type
)
self
.
assertTrue
(
msg2
.
fields_by_name
[
'int_with_default'
]
.
has_default_value
)
self
.
assertTrue
(
msg2
.
fields_by_name
[
'int_with_default'
]
.
has_default_value
)
self
.
assertEqual
s
(
self
.
assertEqual
(
1776
,
msg2
.
fields_by_name
[
'int_with_default'
]
.
default_value
)
1776
,
msg2
.
fields_by_name
[
'int_with_default'
]
.
default_value
)
self
.
assertTrue
(
self
.
assertTrue
(
msg2
.
fields_by_name
[
'double_with_default'
]
.
has_default_value
)
msg2
.
fields_by_name
[
'double_with_default'
]
.
has_default_value
)
self
.
assertEqual
s
(
self
.
assertEqual
(
9.99
,
msg2
.
fields_by_name
[
'double_with_default'
]
.
default_value
)
9.99
,
msg2
.
fields_by_name
[
'double_with_default'
]
.
default_value
)
self
.
assertTrue
(
self
.
assertTrue
(
msg2
.
fields_by_name
[
'string_with_default'
]
.
has_default_value
)
msg2
.
fields_by_name
[
'string_with_default'
]
.
has_default_value
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'hello world'
,
msg2
.
fields_by_name
[
'string_with_default'
]
.
default_value
)
'hello world'
,
msg2
.
fields_by_name
[
'string_with_default'
]
.
default_value
)
self
.
assertTrue
(
msg2
.
fields_by_name
[
'bool_with_default'
]
.
has_default_value
)
self
.
assertTrue
(
msg2
.
fields_by_name
[
'bool_with_default'
]
.
has_default_value
)
self
.
assertFalse
(
msg2
.
fields_by_name
[
'bool_with_default'
]
.
default_value
)
self
.
assertFalse
(
msg2
.
fields_by_name
[
'bool_with_default'
]
.
default_value
)
self
.
assertTrue
(
msg2
.
fields_by_name
[
'enum_with_default'
]
.
has_default_value
)
self
.
assertTrue
(
msg2
.
fields_by_name
[
'enum_with_default'
]
.
has_default_value
)
self
.
assertEqual
s
(
self
.
assertEqual
(
1
,
msg2
.
fields_by_name
[
'enum_with_default'
]
.
default_value
)
1
,
msg2
.
fields_by_name
[
'enum_with_default'
]
.
default_value
)
msg3
=
self
.
pool
.
FindMessageTypeByName
(
msg3
=
self
.
pool
.
FindMessageTypeByName
(
'google.protobuf.python.internal.Factory2Message.NestedFactory2Message'
)
'google.protobuf.python.internal.Factory2Message.NestedFactory2Message'
)
self
.
assertEqual
s
(
nested_msg2
,
msg3
)
self
.
assertEqual
(
nested_msg2
,
msg3
)
self
.
assertTrue
(
msg2
.
fields_by_name
[
'bytes_with_default'
]
.
has_default_value
)
self
.
assertTrue
(
msg2
.
fields_by_name
[
'bytes_with_default'
]
.
has_default_value
)
self
.
assertEqual
s
(
self
.
assertEqual
(
b
'a
\xfb\x00
c'
,
b
'a
\xfb\x00
c'
,
msg2
.
fields_by_name
[
'bytes_with_default'
]
.
default_value
)
msg2
.
fields_by_name
[
'bytes_with_default'
]
.
default_value
)
...
@@ -190,29 +190,29 @@ class DescriptorPoolTest(unittest.TestCase):
...
@@ -190,29 +190,29 @@ class DescriptorPoolTest(unittest.TestCase):
enum1
=
self
.
pool
.
FindEnumTypeByName
(
enum1
=
self
.
pool
.
FindEnumTypeByName
(
'google.protobuf.python.internal.Factory1Enum'
)
'google.protobuf.python.internal.Factory1Enum'
)
self
.
assertIsInstance
(
enum1
,
descriptor
.
EnumDescriptor
)
self
.
assertIsInstance
(
enum1
,
descriptor
.
EnumDescriptor
)
self
.
assertEqual
s
(
0
,
enum1
.
values_by_name
[
'FACTORY_1_VALUE_0'
]
.
number
)
self
.
assertEqual
(
0
,
enum1
.
values_by_name
[
'FACTORY_1_VALUE_0'
]
.
number
)
self
.
assertEqual
s
(
1
,
enum1
.
values_by_name
[
'FACTORY_1_VALUE_1'
]
.
number
)
self
.
assertEqual
(
1
,
enum1
.
values_by_name
[
'FACTORY_1_VALUE_1'
]
.
number
)
nested_enum1
=
self
.
pool
.
FindEnumTypeByName
(
nested_enum1
=
self
.
pool
.
FindEnumTypeByName
(
'google.protobuf.python.internal.Factory1Message.NestedFactory1Enum'
)
'google.protobuf.python.internal.Factory1Message.NestedFactory1Enum'
)
self
.
assertIsInstance
(
nested_enum1
,
descriptor
.
EnumDescriptor
)
self
.
assertIsInstance
(
nested_enum1
,
descriptor
.
EnumDescriptor
)
self
.
assertEqual
s
(
self
.
assertEqual
(
0
,
nested_enum1
.
values_by_name
[
'NESTED_FACTORY_1_VALUE_0'
]
.
number
)
0
,
nested_enum1
.
values_by_name
[
'NESTED_FACTORY_1_VALUE_0'
]
.
number
)
self
.
assertEqual
s
(
self
.
assertEqual
(
1
,
nested_enum1
.
values_by_name
[
'NESTED_FACTORY_1_VALUE_1'
]
.
number
)
1
,
nested_enum1
.
values_by_name
[
'NESTED_FACTORY_1_VALUE_1'
]
.
number
)
enum2
=
self
.
pool
.
FindEnumTypeByName
(
enum2
=
self
.
pool
.
FindEnumTypeByName
(
'google.protobuf.python.internal.Factory2Enum'
)
'google.protobuf.python.internal.Factory2Enum'
)
self
.
assertIsInstance
(
enum2
,
descriptor
.
EnumDescriptor
)
self
.
assertIsInstance
(
enum2
,
descriptor
.
EnumDescriptor
)
self
.
assertEqual
s
(
0
,
enum2
.
values_by_name
[
'FACTORY_2_VALUE_0'
]
.
number
)
self
.
assertEqual
(
0
,
enum2
.
values_by_name
[
'FACTORY_2_VALUE_0'
]
.
number
)
self
.
assertEqual
s
(
1
,
enum2
.
values_by_name
[
'FACTORY_2_VALUE_1'
]
.
number
)
self
.
assertEqual
(
1
,
enum2
.
values_by_name
[
'FACTORY_2_VALUE_1'
]
.
number
)
nested_enum2
=
self
.
pool
.
FindEnumTypeByName
(
nested_enum2
=
self
.
pool
.
FindEnumTypeByName
(
'google.protobuf.python.internal.Factory2Message.NestedFactory2Enum'
)
'google.protobuf.python.internal.Factory2Message.NestedFactory2Enum'
)
self
.
assertIsInstance
(
nested_enum2
,
descriptor
.
EnumDescriptor
)
self
.
assertIsInstance
(
nested_enum2
,
descriptor
.
EnumDescriptor
)
self
.
assertEqual
s
(
self
.
assertEqual
(
0
,
nested_enum2
.
values_by_name
[
'NESTED_FACTORY_2_VALUE_0'
]
.
number
)
0
,
nested_enum2
.
values_by_name
[
'NESTED_FACTORY_2_VALUE_0'
]
.
number
)
self
.
assertEqual
s
(
self
.
assertEqual
(
1
,
nested_enum2
.
values_by_name
[
'NESTED_FACTORY_2_VALUE_1'
]
.
number
)
1
,
nested_enum2
.
values_by_name
[
'NESTED_FACTORY_2_VALUE_1'
]
.
number
)
def
testFindEnumTypeByNameFailure
(
self
):
def
testFindEnumTypeByNameFailure
(
self
):
...
@@ -274,8 +274,8 @@ class ProtoFile(object):
...
@@ -274,8 +274,8 @@ class ProtoFile(object):
def
CheckFile
(
self
,
test
,
pool
):
def
CheckFile
(
self
,
test
,
pool
):
file_desc
=
pool
.
FindFileByName
(
self
.
name
)
file_desc
=
pool
.
FindFileByName
(
self
.
name
)
test
.
assertEqual
s
(
self
.
name
,
file_desc
.
name
)
test
.
assertEqual
(
self
.
name
,
file_desc
.
name
)
test
.
assertEqual
s
(
self
.
package
,
file_desc
.
package
)
test
.
assertEqual
(
self
.
package
,
file_desc
.
package
)
dependencies_names
=
[
f
.
name
for
f
in
file_desc
.
dependencies
]
dependencies_names
=
[
f
.
name
for
f
in
file_desc
.
dependencies
]
test
.
assertEqual
(
self
.
dependencies
,
dependencies_names
)
test
.
assertEqual
(
self
.
dependencies
,
dependencies_names
)
for
name
,
msg_type
in
self
.
messages
.
items
():
for
name
,
msg_type
in
self
.
messages
.
items
():
...
@@ -430,7 +430,7 @@ class AddDescriptorTest(unittest.TestCase):
...
@@ -430,7 +430,7 @@ class AddDescriptorTest(unittest.TestCase):
def
_TestMessage
(
self
,
prefix
):
def
_TestMessage
(
self
,
prefix
):
pool
=
descriptor_pool
.
DescriptorPool
()
pool
=
descriptor_pool
.
DescriptorPool
()
pool
.
AddDescriptor
(
unittest_pb2
.
TestAllTypes
.
DESCRIPTOR
)
pool
.
AddDescriptor
(
unittest_pb2
.
TestAllTypes
.
DESCRIPTOR
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.TestAllTypes'
,
'protobuf_unittest.TestAllTypes'
,
pool
.
FindMessageTypeByName
(
pool
.
FindMessageTypeByName
(
prefix
+
'protobuf_unittest.TestAllTypes'
)
.
full_name
)
prefix
+
'protobuf_unittest.TestAllTypes'
)
.
full_name
)
...
@@ -441,18 +441,18 @@ class AddDescriptorTest(unittest.TestCase):
...
@@ -441,18 +441,18 @@ class AddDescriptorTest(unittest.TestCase):
prefix
+
'protobuf_unittest.TestAllTypes.NestedMessage'
)
prefix
+
'protobuf_unittest.TestAllTypes.NestedMessage'
)
pool
.
AddDescriptor
(
unittest_pb2
.
TestAllTypes
.
NestedMessage
.
DESCRIPTOR
)
pool
.
AddDescriptor
(
unittest_pb2
.
TestAllTypes
.
NestedMessage
.
DESCRIPTOR
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.TestAllTypes.NestedMessage'
,
'protobuf_unittest.TestAllTypes.NestedMessage'
,
pool
.
FindMessageTypeByName
(
pool
.
FindMessageTypeByName
(
prefix
+
'protobuf_unittest.TestAllTypes.NestedMessage'
)
.
full_name
)
prefix
+
'protobuf_unittest.TestAllTypes.NestedMessage'
)
.
full_name
)
# Files are implicitly also indexed when messages are added.
# Files are implicitly also indexed when messages are added.
self
.
assertEqual
s
(
self
.
assertEqual
(
'google/protobuf/unittest.proto'
,
'google/protobuf/unittest.proto'
,
pool
.
FindFileByName
(
pool
.
FindFileByName
(
'google/protobuf/unittest.proto'
)
.
name
)
'google/protobuf/unittest.proto'
)
.
name
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'google/protobuf/unittest.proto'
,
'google/protobuf/unittest.proto'
,
pool
.
FindFileContainingSymbol
(
pool
.
FindFileContainingSymbol
(
prefix
+
'protobuf_unittest.TestAllTypes.NestedMessage'
)
.
name
)
prefix
+
'protobuf_unittest.TestAllTypes.NestedMessage'
)
.
name
)
...
@@ -464,7 +464,7 @@ class AddDescriptorTest(unittest.TestCase):
...
@@ -464,7 +464,7 @@ class AddDescriptorTest(unittest.TestCase):
def
_TestEnum
(
self
,
prefix
):
def
_TestEnum
(
self
,
prefix
):
pool
=
descriptor_pool
.
DescriptorPool
()
pool
=
descriptor_pool
.
DescriptorPool
()
pool
.
AddEnumDescriptor
(
unittest_pb2
.
ForeignEnum
.
DESCRIPTOR
)
pool
.
AddEnumDescriptor
(
unittest_pb2
.
ForeignEnum
.
DESCRIPTOR
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.ForeignEnum'
,
'protobuf_unittest.ForeignEnum'
,
pool
.
FindEnumTypeByName
(
pool
.
FindEnumTypeByName
(
prefix
+
'protobuf_unittest.ForeignEnum'
)
.
full_name
)
prefix
+
'protobuf_unittest.ForeignEnum'
)
.
full_name
)
...
@@ -475,18 +475,18 @@ class AddDescriptorTest(unittest.TestCase):
...
@@ -475,18 +475,18 @@ class AddDescriptorTest(unittest.TestCase):
prefix
+
'protobuf_unittest.ForeignEnum.NestedEnum'
)
prefix
+
'protobuf_unittest.ForeignEnum.NestedEnum'
)
pool
.
AddEnumDescriptor
(
unittest_pb2
.
TestAllTypes
.
NestedEnum
.
DESCRIPTOR
)
pool
.
AddEnumDescriptor
(
unittest_pb2
.
TestAllTypes
.
NestedEnum
.
DESCRIPTOR
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.TestAllTypes.NestedEnum'
,
'protobuf_unittest.TestAllTypes.NestedEnum'
,
pool
.
FindEnumTypeByName
(
pool
.
FindEnumTypeByName
(
prefix
+
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
full_name
)
prefix
+
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
full_name
)
# Files are implicitly also indexed when enums are added.
# Files are implicitly also indexed when enums are added.
self
.
assertEqual
s
(
self
.
assertEqual
(
'google/protobuf/unittest.proto'
,
'google/protobuf/unittest.proto'
,
pool
.
FindFileByName
(
pool
.
FindFileByName
(
'google/protobuf/unittest.proto'
)
.
name
)
'google/protobuf/unittest.proto'
)
.
name
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'google/protobuf/unittest.proto'
,
'google/protobuf/unittest.proto'
,
pool
.
FindFileContainingSymbol
(
pool
.
FindFileContainingSymbol
(
prefix
+
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
name
)
prefix
+
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
name
)
...
@@ -498,7 +498,7 @@ class AddDescriptorTest(unittest.TestCase):
...
@@ -498,7 +498,7 @@ class AddDescriptorTest(unittest.TestCase):
def
testFile
(
self
):
def
testFile
(
self
):
pool
=
descriptor_pool
.
DescriptorPool
()
pool
=
descriptor_pool
.
DescriptorPool
()
pool
.
AddFileDescriptor
(
unittest_pb2
.
DESCRIPTOR
)
pool
.
AddFileDescriptor
(
unittest_pb2
.
DESCRIPTOR
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'google/protobuf/unittest.proto'
,
'google/protobuf/unittest.proto'
,
pool
.
FindFileByName
(
pool
.
FindFileByName
(
'google/protobuf/unittest.proto'
)
.
name
)
'google/protobuf/unittest.proto'
)
.
name
)
...
...
python/google/protobuf/internal/descriptor_test.py
View file @
a2abc947
...
@@ -673,7 +673,7 @@ class MakeDescriptorTest(unittest.TestCase):
...
@@ -673,7 +673,7 @@ class MakeDescriptorTest(unittest.TestCase):
reformed_descriptor
=
descriptor
.
MakeDescriptor
(
descriptor_proto
)
reformed_descriptor
=
descriptor
.
MakeDescriptor
(
descriptor_proto
)
options
=
reformed_descriptor
.
GetOptions
()
options
=
reformed_descriptor
.
GetOptions
()
self
.
assertEqual
s
(
101
,
self
.
assertEqual
(
101
,
options
.
Extensions
[
unittest_custom_options_pb2
.
msgopt
]
.
i
)
options
.
Extensions
[
unittest_custom_options_pb2
.
msgopt
]
.
i
)
if
__name__
==
'__main__'
:
if
__name__
==
'__main__'
:
...
...
python/google/protobuf/internal/generator_test.py
View file @
a2abc947
...
@@ -154,7 +154,7 @@ class GeneratorTest(unittest.TestCase):
...
@@ -154,7 +154,7 @@ class GeneratorTest(unittest.TestCase):
# extension and for its value to be set to -789.
# extension and for its value to be set to -789.
def
testNestedTypes
(
self
):
def
testNestedTypes
(
self
):
self
.
assertEqual
s
(
self
.
assertEqual
(
set
(
unittest_pb2
.
TestAllTypes
.
DESCRIPTOR
.
nested_types
),
set
(
unittest_pb2
.
TestAllTypes
.
DESCRIPTOR
.
nested_types
),
set
([
set
([
unittest_pb2
.
TestAllTypes
.
NestedMessage
.
DESCRIPTOR
,
unittest_pb2
.
TestAllTypes
.
NestedMessage
.
DESCRIPTOR
,
...
@@ -306,36 +306,36 @@ class SymbolDatabaseRegistrationTest(unittest.TestCase):
...
@@ -306,36 +306,36 @@ class SymbolDatabaseRegistrationTest(unittest.TestCase):
"""Checks that messages, enums and files are correctly registered."""
"""Checks that messages, enums and files are correctly registered."""
def
testGetSymbol
(
self
):
def
testGetSymbol
(
self
):
self
.
assertEqual
s
(
self
.
assertEqual
(
unittest_pb2
.
TestAllTypes
,
symbol_database
.
Default
()
.
GetSymbol
(
unittest_pb2
.
TestAllTypes
,
symbol_database
.
Default
()
.
GetSymbol
(
'protobuf_unittest.TestAllTypes'
))
'protobuf_unittest.TestAllTypes'
))
self
.
assertEqual
s
(
self
.
assertEqual
(
unittest_pb2
.
TestAllTypes
.
NestedMessage
,
unittest_pb2
.
TestAllTypes
.
NestedMessage
,
symbol_database
.
Default
()
.
GetSymbol
(
symbol_database
.
Default
()
.
GetSymbol
(
'protobuf_unittest.TestAllTypes.NestedMessage'
))
'protobuf_unittest.TestAllTypes.NestedMessage'
))
with
self
.
assertRaises
(
KeyError
):
with
self
.
assertRaises
(
KeyError
):
symbol_database
.
Default
()
.
GetSymbol
(
'protobuf_unittest.NestedMessage'
)
symbol_database
.
Default
()
.
GetSymbol
(
'protobuf_unittest.NestedMessage'
)
self
.
assertEqual
s
(
self
.
assertEqual
(
unittest_pb2
.
TestAllTypes
.
OptionalGroup
,
unittest_pb2
.
TestAllTypes
.
OptionalGroup
,
symbol_database
.
Default
()
.
GetSymbol
(
symbol_database
.
Default
()
.
GetSymbol
(
'protobuf_unittest.TestAllTypes.OptionalGroup'
))
'protobuf_unittest.TestAllTypes.OptionalGroup'
))
self
.
assertEqual
s
(
self
.
assertEqual
(
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
,
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
,
symbol_database
.
Default
()
.
GetSymbol
(
symbol_database
.
Default
()
.
GetSymbol
(
'protobuf_unittest.TestAllTypes.RepeatedGroup'
))
'protobuf_unittest.TestAllTypes.RepeatedGroup'
))
def
testEnums
(
self
):
def
testEnums
(
self
):
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.ForeignEnum'
,
'protobuf_unittest.ForeignEnum'
,
symbol_database
.
Default
()
.
pool
.
FindEnumTypeByName
(
symbol_database
.
Default
()
.
pool
.
FindEnumTypeByName
(
'protobuf_unittest.ForeignEnum'
)
.
full_name
)
'protobuf_unittest.ForeignEnum'
)
.
full_name
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.TestAllTypes.NestedEnum'
,
'protobuf_unittest.TestAllTypes.NestedEnum'
,
symbol_database
.
Default
()
.
pool
.
FindEnumTypeByName
(
symbol_database
.
Default
()
.
pool
.
FindEnumTypeByName
(
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
full_name
)
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
full_name
)
def
testFindFileByName
(
self
):
def
testFindFileByName
(
self
):
self
.
assertEqual
s
(
self
.
assertEqual
(
'google/protobuf/unittest.proto'
,
'google/protobuf/unittest.proto'
,
symbol_database
.
Default
()
.
pool
.
FindFileByName
(
symbol_database
.
Default
()
.
pool
.
FindFileByName
(
'google/protobuf/unittest.proto'
)
.
name
)
'google/protobuf/unittest.proto'
)
.
name
)
...
...
python/google/protobuf/internal/message_factory_test.py
View file @
a2abc947
...
@@ -82,9 +82,9 @@ class MessageFactoryTest(unittest.TestCase):
...
@@ -82,9 +82,9 @@ class MessageFactoryTest(unittest.TestCase):
serialized
=
msg
.
SerializeToString
()
serialized
=
msg
.
SerializeToString
()
converted
=
factory_test2_pb2
.
Factory2Message
.
FromString
(
serialized
)
converted
=
factory_test2_pb2
.
Factory2Message
.
FromString
(
serialized
)
reserialized
=
converted
.
SerializeToString
()
reserialized
=
converted
.
SerializeToString
()
self
.
assertEqual
s
(
serialized
,
reserialized
)
self
.
assertEqual
(
serialized
,
reserialized
)
result
=
cls
.
FromString
(
reserialized
)
result
=
cls
.
FromString
(
reserialized
)
self
.
assertEqual
s
(
msg
,
result
)
self
.
assertEqual
(
msg
,
result
)
def
testGetPrototype
(
self
):
def
testGetPrototype
(
self
):
db
=
descriptor_database
.
DescriptorDatabase
()
db
=
descriptor_database
.
DescriptorDatabase
()
...
@@ -125,8 +125,8 @@ class MessageFactoryTest(unittest.TestCase):
...
@@ -125,8 +125,8 @@ class MessageFactoryTest(unittest.TestCase):
'google.protobuf.python.internal.another_field'
]
'google.protobuf.python.internal.another_field'
]
msg1
.
Extensions
[
ext1
]
=
'test1'
msg1
.
Extensions
[
ext1
]
=
'test1'
msg1
.
Extensions
[
ext2
]
=
'test2'
msg1
.
Extensions
[
ext2
]
=
'test2'
self
.
assertEqual
s
(
'test1'
,
msg1
.
Extensions
[
ext1
])
self
.
assertEqual
(
'test1'
,
msg1
.
Extensions
[
ext1
])
self
.
assertEqual
s
(
'test2'
,
msg1
.
Extensions
[
ext2
])
self
.
assertEqual
(
'test2'
,
msg1
.
Extensions
[
ext2
])
if
__name__
==
'__main__'
:
if
__name__
==
'__main__'
:
...
...
python/google/protobuf/internal/message_test.py
View file @
a2abc947
...
@@ -98,7 +98,7 @@ class MessageTest(unittest.TestCase):
...
@@ -98,7 +98,7 @@ class MessageTest(unittest.TestCase):
golden_message
.
ParseFromString
(
golden_data
)
golden_message
.
ParseFromString
(
golden_data
)
all_set
=
unittest_pb2
.
TestAllExtensions
()
all_set
=
unittest_pb2
.
TestAllExtensions
()
test_util
.
SetAllExtensions
(
all_set
)
test_util
.
SetAllExtensions
(
all_set
)
self
.
assertEqual
s
(
all_set
,
golden_message
)
self
.
assertEqual
(
all_set
,
golden_message
)
self
.
assertEqual
(
golden_data
,
golden_message
.
SerializeToString
())
self
.
assertEqual
(
golden_data
,
golden_message
.
SerializeToString
())
golden_copy
=
copy
.
deepcopy
(
golden_message
)
golden_copy
=
copy
.
deepcopy
(
golden_message
)
self
.
assertEqual
(
golden_data
,
golden_copy
.
SerializeToString
())
self
.
assertEqual
(
golden_data
,
golden_copy
.
SerializeToString
())
...
@@ -109,7 +109,7 @@ class MessageTest(unittest.TestCase):
...
@@ -109,7 +109,7 @@ class MessageTest(unittest.TestCase):
golden_message
.
ParseFromString
(
golden_data
)
golden_message
.
ParseFromString
(
golden_data
)
all_set
=
unittest_pb2
.
TestPackedTypes
()
all_set
=
unittest_pb2
.
TestPackedTypes
()
test_util
.
SetAllPackedFields
(
all_set
)
test_util
.
SetAllPackedFields
(
all_set
)
self
.
assertEqual
s
(
all_set
,
golden_message
)
self
.
assertEqual
(
all_set
,
golden_message
)
self
.
assertEqual
(
golden_data
,
all_set
.
SerializeToString
())
self
.
assertEqual
(
golden_data
,
all_set
.
SerializeToString
())
golden_copy
=
copy
.
deepcopy
(
golden_message
)
golden_copy
=
copy
.
deepcopy
(
golden_message
)
self
.
assertEqual
(
golden_data
,
golden_copy
.
SerializeToString
())
self
.
assertEqual
(
golden_data
,
golden_copy
.
SerializeToString
())
...
@@ -120,7 +120,7 @@ class MessageTest(unittest.TestCase):
...
@@ -120,7 +120,7 @@ class MessageTest(unittest.TestCase):
golden_message
.
ParseFromString
(
golden_data
)
golden_message
.
ParseFromString
(
golden_data
)
all_set
=
unittest_pb2
.
TestPackedExtensions
()
all_set
=
unittest_pb2
.
TestPackedExtensions
()
test_util
.
SetAllPackedExtensions
(
all_set
)
test_util
.
SetAllPackedExtensions
(
all_set
)
self
.
assertEqual
s
(
all_set
,
golden_message
)
self
.
assertEqual
(
all_set
,
golden_message
)
self
.
assertEqual
(
golden_data
,
all_set
.
SerializeToString
())
self
.
assertEqual
(
golden_data
,
all_set
.
SerializeToString
())
golden_copy
=
copy
.
deepcopy
(
golden_message
)
golden_copy
=
copy
.
deepcopy
(
golden_message
)
self
.
assertEqual
(
golden_data
,
golden_copy
.
SerializeToString
())
self
.
assertEqual
(
golden_data
,
golden_copy
.
SerializeToString
())
...
@@ -132,7 +132,7 @@ class MessageTest(unittest.TestCase):
...
@@ -132,7 +132,7 @@ class MessageTest(unittest.TestCase):
pickled_message
=
pickle
.
dumps
(
golden_message
)
pickled_message
=
pickle
.
dumps
(
golden_message
)
unpickled_message
=
pickle
.
loads
(
pickled_message
)
unpickled_message
=
pickle
.
loads
(
pickled_message
)
self
.
assertEqual
s
(
unpickled_message
,
golden_message
)
self
.
assertEqual
(
unpickled_message
,
golden_message
)
def
testPickleIncompleteProto
(
self
):
def
testPickleIncompleteProto
(
self
):
...
@@ -140,8 +140,8 @@ class MessageTest(unittest.TestCase):
...
@@ -140,8 +140,8 @@ class MessageTest(unittest.TestCase):
pickled_message
=
pickle
.
dumps
(
golden_message
)
pickled_message
=
pickle
.
dumps
(
golden_message
)
unpickled_message
=
pickle
.
loads
(
pickled_message
)
unpickled_message
=
pickle
.
loads
(
pickled_message
)
self
.
assertEqual
s
(
unpickled_message
,
golden_message
)
self
.
assertEqual
(
unpickled_message
,
golden_message
)
self
.
assertEqual
s
(
unpickled_message
.
a
,
1
)
self
.
assertEqual
(
unpickled_message
.
a
,
1
)
# This is still an incomplete proto - so serializing should fail
# This is still an incomplete proto - so serializing should fail
self
.
assertRaises
(
message
.
EncodeError
,
unpickled_message
.
SerializeToString
)
self
.
assertRaises
(
message
.
EncodeError
,
unpickled_message
.
SerializeToString
)
...
@@ -344,11 +344,11 @@ class MessageTest(unittest.TestCase):
...
@@ -344,11 +344,11 @@ class MessageTest(unittest.TestCase):
msg
.
repeated_nested_message
.
add
(
bb
=
3
)
msg
.
repeated_nested_message
.
add
(
bb
=
3
)
msg
.
repeated_nested_message
.
add
(
bb
=
4
)
msg
.
repeated_nested_message
.
add
(
bb
=
4
)
self
.
assertEqual
s
([
1
,
2
,
3
,
4
],
self
.
assertEqual
([
1
,
2
,
3
,
4
],
[
m
.
bb
for
m
in
msg
.
repeated_nested_message
])
[
m
.
bb
for
m
in
msg
.
repeated_nested_message
])
self
.
assertEqual
s
([
4
,
3
,
2
,
1
],
self
.
assertEqual
([
4
,
3
,
2
,
1
],
[
m
.
bb
for
m
in
reversed
(
msg
.
repeated_nested_message
)])
[
m
.
bb
for
m
in
reversed
(
msg
.
repeated_nested_message
)])
self
.
assertEqual
s
([
4
,
3
,
2
,
1
],
self
.
assertEqual
([
4
,
3
,
2
,
1
],
[
m
.
bb
for
m
in
msg
.
repeated_nested_message
[::
-
1
]])
[
m
.
bb
for
m
in
msg
.
repeated_nested_message
[::
-
1
]])
def
testSortingRepeatedScalarFieldsDefaultComparator
(
self
):
def
testSortingRepeatedScalarFieldsDefaultComparator
(
self
):
...
...
python/google/protobuf/internal/reflection_test.py
View file @
a2abc947
...
@@ -141,10 +141,10 @@ class ReflectionTest(unittest.TestCase):
...
@@ -141,10 +141,10 @@ class ReflectionTest(unittest.TestCase):
repeated_bool
=
[
True
,
False
,
False
],
repeated_bool
=
[
True
,
False
,
False
],
repeated_string
=
[
"optional_string"
])
repeated_string
=
[
"optional_string"
])
self
.
assertEqual
s
([
1
,
2
,
3
,
4
],
list
(
proto
.
repeated_int32
))
self
.
assertEqual
([
1
,
2
,
3
,
4
],
list
(
proto
.
repeated_int32
))
self
.
assertEqual
s
([
1.23
,
54.321
],
list
(
proto
.
repeated_double
))
self
.
assertEqual
([
1.23
,
54.321
],
list
(
proto
.
repeated_double
))
self
.
assertEqual
s
([
True
,
False
,
False
],
list
(
proto
.
repeated_bool
))
self
.
assertEqual
([
True
,
False
,
False
],
list
(
proto
.
repeated_bool
))
self
.
assertEqual
s
([
"optional_string"
],
list
(
proto
.
repeated_string
))
self
.
assertEqual
([
"optional_string"
],
list
(
proto
.
repeated_string
))
def
testRepeatedCompositeConstructor
(
self
):
def
testRepeatedCompositeConstructor
(
self
):
# Constructor with only repeated composite types should succeed.
# Constructor with only repeated composite types should succeed.
...
@@ -163,18 +163,18 @@ class ReflectionTest(unittest.TestCase):
...
@@ -163,18 +163,18 @@ class ReflectionTest(unittest.TestCase):
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(
a
=
1
),
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(
a
=
1
),
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(
a
=
2
)])
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(
a
=
2
)])
self
.
assertEqual
s
(
self
.
assertEqual
(
[
unittest_pb2
.
TestAllTypes
.
NestedMessage
(
[
unittest_pb2
.
TestAllTypes
.
NestedMessage
(
bb
=
unittest_pb2
.
TestAllTypes
.
FOO
),
bb
=
unittest_pb2
.
TestAllTypes
.
FOO
),
unittest_pb2
.
TestAllTypes
.
NestedMessage
(
unittest_pb2
.
TestAllTypes
.
NestedMessage
(
bb
=
unittest_pb2
.
TestAllTypes
.
BAR
)],
bb
=
unittest_pb2
.
TestAllTypes
.
BAR
)],
list
(
proto
.
repeated_nested_message
))
list
(
proto
.
repeated_nested_message
))
self
.
assertEqual
s
(
self
.
assertEqual
(
[
unittest_pb2
.
ForeignMessage
(
c
=-
43
),
[
unittest_pb2
.
ForeignMessage
(
c
=-
43
),
unittest_pb2
.
ForeignMessage
(
c
=
45324
),
unittest_pb2
.
ForeignMessage
(
c
=
45324
),
unittest_pb2
.
ForeignMessage
(
c
=
12
)],
unittest_pb2
.
ForeignMessage
(
c
=
12
)],
list
(
proto
.
repeated_foreign_message
))
list
(
proto
.
repeated_foreign_message
))
self
.
assertEqual
s
(
self
.
assertEqual
(
[
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(),
[
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(),
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(
a
=
1
),
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(
a
=
1
),
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(
a
=
2
)],
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
(
a
=
2
)],
...
@@ -199,15 +199,15 @@ class ReflectionTest(unittest.TestCase):
...
@@ -199,15 +199,15 @@ class ReflectionTest(unittest.TestCase):
self
.
assertEqual
(
24
,
proto
.
optional_int32
)
self
.
assertEqual
(
24
,
proto
.
optional_int32
)
self
.
assertEqual
(
'optional_string'
,
proto
.
optional_string
)
self
.
assertEqual
(
'optional_string'
,
proto
.
optional_string
)
self
.
assertEqual
s
([
1.23
,
54.321
],
list
(
proto
.
repeated_double
))
self
.
assertEqual
([
1.23
,
54.321
],
list
(
proto
.
repeated_double
))
self
.
assertEqual
s
([
True
,
False
,
False
],
list
(
proto
.
repeated_bool
))
self
.
assertEqual
([
True
,
False
,
False
],
list
(
proto
.
repeated_bool
))
self
.
assertEqual
s
(
self
.
assertEqual
(
[
unittest_pb2
.
TestAllTypes
.
NestedMessage
(
[
unittest_pb2
.
TestAllTypes
.
NestedMessage
(
bb
=
unittest_pb2
.
TestAllTypes
.
FOO
),
bb
=
unittest_pb2
.
TestAllTypes
.
FOO
),
unittest_pb2
.
TestAllTypes
.
NestedMessage
(
unittest_pb2
.
TestAllTypes
.
NestedMessage
(
bb
=
unittest_pb2
.
TestAllTypes
.
BAR
)],
bb
=
unittest_pb2
.
TestAllTypes
.
BAR
)],
list
(
proto
.
repeated_nested_message
))
list
(
proto
.
repeated_nested_message
))
self
.
assertEqual
s
(
self
.
assertEqual
(
[
unittest_pb2
.
ForeignMessage
(
c
=-
43
),
[
unittest_pb2
.
ForeignMessage
(
c
=-
43
),
unittest_pb2
.
ForeignMessage
(
c
=
45324
),
unittest_pb2
.
ForeignMessage
(
c
=
45324
),
unittest_pb2
.
ForeignMessage
(
c
=
12
)],
unittest_pb2
.
ForeignMessage
(
c
=
12
)],
...
@@ -235,18 +235,18 @@ class ReflectionTest(unittest.TestCase):
...
@@ -235,18 +235,18 @@ class ReflectionTest(unittest.TestCase):
def
testConstructorInvalidatesCachedByteSize
(
self
):
def
testConstructorInvalidatesCachedByteSize
(
self
):
message
=
unittest_pb2
.
TestAllTypes
(
optional_int32
=
12
)
message
=
unittest_pb2
.
TestAllTypes
(
optional_int32
=
12
)
self
.
assertEqual
s
(
2
,
message
.
ByteSize
())
self
.
assertEqual
(
2
,
message
.
ByteSize
())
message
=
unittest_pb2
.
TestAllTypes
(
message
=
unittest_pb2
.
TestAllTypes
(
optional_nested_message
=
unittest_pb2
.
TestAllTypes
.
NestedMessage
())
optional_nested_message
=
unittest_pb2
.
TestAllTypes
.
NestedMessage
())
self
.
assertEqual
s
(
3
,
message
.
ByteSize
())
self
.
assertEqual
(
3
,
message
.
ByteSize
())
message
=
unittest_pb2
.
TestAllTypes
(
repeated_int32
=
[
12
])
message
=
unittest_pb2
.
TestAllTypes
(
repeated_int32
=
[
12
])
self
.
assertEqual
s
(
3
,
message
.
ByteSize
())
self
.
assertEqual
(
3
,
message
.
ByteSize
())
message
=
unittest_pb2
.
TestAllTypes
(
message
=
unittest_pb2
.
TestAllTypes
(
repeated_nested_message
=
[
unittest_pb2
.
TestAllTypes
.
NestedMessage
()])
repeated_nested_message
=
[
unittest_pb2
.
TestAllTypes
.
NestedMessage
()])
self
.
assertEqual
s
(
3
,
message
.
ByteSize
())
self
.
assertEqual
(
3
,
message
.
ByteSize
())
def
testSimpleHasBits
(
self
):
def
testSimpleHasBits
(
self
):
# Test a scalar.
# Test a scalar.
...
@@ -1188,7 +1188,7 @@ class ReflectionTest(unittest.TestCase):
...
@@ -1188,7 +1188,7 @@ class ReflectionTest(unittest.TestCase):
self
.
assertTrue
(
1
in
unittest_pb2
.
TestAllExtensions
.
_extensions_by_number
)
self
.
assertTrue
(
1
in
unittest_pb2
.
TestAllExtensions
.
_extensions_by_number
)
# Make sure extensions haven't been registered into types that shouldn't
# Make sure extensions haven't been registered into types that shouldn't
# have any.
# have any.
self
.
assertEqual
s
(
0
,
len
(
unittest_pb2
.
TestAllTypes
.
_extensions_by_name
))
self
.
assertEqual
(
0
,
len
(
unittest_pb2
.
TestAllTypes
.
_extensions_by_name
))
# If message A directly contains message B, and
# If message A directly contains message B, and
# a.HasField('b') is currently False, then mutating any
# a.HasField('b') is currently False, then mutating any
...
@@ -1512,18 +1512,18 @@ class ReflectionTest(unittest.TestCase):
...
@@ -1512,18 +1512,18 @@ class ReflectionTest(unittest.TestCase):
test_util
.
SetAllNonLazyFields
(
proto
)
test_util
.
SetAllNonLazyFields
(
proto
)
# Clear the message.
# Clear the message.
proto
.
Clear
()
proto
.
Clear
()
self
.
assertEqual
s
(
proto
.
ByteSize
(),
0
)
self
.
assertEqual
(
proto
.
ByteSize
(),
0
)
empty_proto
=
unittest_pb2
.
TestAllTypes
()
empty_proto
=
unittest_pb2
.
TestAllTypes
()
self
.
assertEqual
s
(
proto
,
empty_proto
)
self
.
assertEqual
(
proto
,
empty_proto
)
# Test if extensions which were set are cleared.
# Test if extensions which were set are cleared.
proto
=
unittest_pb2
.
TestAllExtensions
()
proto
=
unittest_pb2
.
TestAllExtensions
()
test_util
.
SetAllExtensions
(
proto
)
test_util
.
SetAllExtensions
(
proto
)
# Clear the message.
# Clear the message.
proto
.
Clear
()
proto
.
Clear
()
self
.
assertEqual
s
(
proto
.
ByteSize
(),
0
)
self
.
assertEqual
(
proto
.
ByteSize
(),
0
)
empty_proto
=
unittest_pb2
.
TestAllExtensions
()
empty_proto
=
unittest_pb2
.
TestAllExtensions
()
self
.
assertEqual
s
(
proto
,
empty_proto
)
self
.
assertEqual
(
proto
,
empty_proto
)
def
testDisconnectingBeforeClear
(
self
):
def
testDisconnectingBeforeClear
(
self
):
proto
=
unittest_pb2
.
TestAllTypes
()
proto
=
unittest_pb2
.
TestAllTypes
()
...
@@ -2481,7 +2481,7 @@ class SerializationTest(unittest.TestCase):
...
@@ -2481,7 +2481,7 @@ class SerializationTest(unittest.TestCase):
# Check that the message parsed well.
# Check that the message parsed well.
extension_message1
=
unittest_mset_pb2
.
TestMessageSetExtension1
extension_message1
=
unittest_mset_pb2
.
TestMessageSetExtension1
extension1
=
extension_message1
.
message_set_extension
extension1
=
extension_message1
.
message_set_extension
self
.
assertEqual
s
(
12345
,
proto
.
Extensions
[
extension1
]
.
i
)
self
.
assertEqual
(
12345
,
proto
.
Extensions
[
extension1
]
.
i
)
def
testUnknownFields
(
self
):
def
testUnknownFields
(
self
):
proto
=
unittest_pb2
.
TestAllTypes
()
proto
=
unittest_pb2
.
TestAllTypes
()
...
...
python/google/protobuf/internal/symbol_database_test.py
View file @
a2abc947
...
@@ -65,53 +65,53 @@ class SymbolDatabaseTest(unittest.TestCase):
...
@@ -65,53 +65,53 @@ class SymbolDatabaseTest(unittest.TestCase):
messages
[
'protobuf_unittest.TestAllTypes'
])
messages
[
'protobuf_unittest.TestAllTypes'
])
def
testGetSymbol
(
self
):
def
testGetSymbol
(
self
):
self
.
assertEqual
s
(
self
.
assertEqual
(
unittest_pb2
.
TestAllTypes
,
self
.
_Database
()
.
GetSymbol
(
unittest_pb2
.
TestAllTypes
,
self
.
_Database
()
.
GetSymbol
(
'protobuf_unittest.TestAllTypes'
))
'protobuf_unittest.TestAllTypes'
))
self
.
assertEqual
s
(
self
.
assertEqual
(
unittest_pb2
.
TestAllTypes
.
NestedMessage
,
self
.
_Database
()
.
GetSymbol
(
unittest_pb2
.
TestAllTypes
.
NestedMessage
,
self
.
_Database
()
.
GetSymbol
(
'protobuf_unittest.TestAllTypes.NestedMessage'
))
'protobuf_unittest.TestAllTypes.NestedMessage'
))
self
.
assertEqual
s
(
self
.
assertEqual
(
unittest_pb2
.
TestAllTypes
.
OptionalGroup
,
self
.
_Database
()
.
GetSymbol
(
unittest_pb2
.
TestAllTypes
.
OptionalGroup
,
self
.
_Database
()
.
GetSymbol
(
'protobuf_unittest.TestAllTypes.OptionalGroup'
))
'protobuf_unittest.TestAllTypes.OptionalGroup'
))
self
.
assertEqual
s
(
self
.
assertEqual
(
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
,
self
.
_Database
()
.
GetSymbol
(
unittest_pb2
.
TestAllTypes
.
RepeatedGroup
,
self
.
_Database
()
.
GetSymbol
(
'protobuf_unittest.TestAllTypes.RepeatedGroup'
))
'protobuf_unittest.TestAllTypes.RepeatedGroup'
))
def
testEnums
(
self
):
def
testEnums
(
self
):
# Check registration of types in the pool.
# Check registration of types in the pool.
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.ForeignEnum'
,
'protobuf_unittest.ForeignEnum'
,
self
.
_Database
()
.
pool
.
FindEnumTypeByName
(
self
.
_Database
()
.
pool
.
FindEnumTypeByName
(
'protobuf_unittest.ForeignEnum'
)
.
full_name
)
'protobuf_unittest.ForeignEnum'
)
.
full_name
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.TestAllTypes.NestedEnum'
,
'protobuf_unittest.TestAllTypes.NestedEnum'
,
self
.
_Database
()
.
pool
.
FindEnumTypeByName
(
self
.
_Database
()
.
pool
.
FindEnumTypeByName
(
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
full_name
)
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
full_name
)
def
testFindMessageTypeByName
(
self
):
def
testFindMessageTypeByName
(
self
):
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.TestAllTypes'
,
'protobuf_unittest.TestAllTypes'
,
self
.
_Database
()
.
pool
.
FindMessageTypeByName
(
self
.
_Database
()
.
pool
.
FindMessageTypeByName
(
'protobuf_unittest.TestAllTypes'
)
.
full_name
)
'protobuf_unittest.TestAllTypes'
)
.
full_name
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'protobuf_unittest.TestAllTypes.NestedMessage'
,
'protobuf_unittest.TestAllTypes.NestedMessage'
,
self
.
_Database
()
.
pool
.
FindMessageTypeByName
(
self
.
_Database
()
.
pool
.
FindMessageTypeByName
(
'protobuf_unittest.TestAllTypes.NestedMessage'
)
.
full_name
)
'protobuf_unittest.TestAllTypes.NestedMessage'
)
.
full_name
)
def
testFindFindContainingSymbol
(
self
):
def
testFindFindContainingSymbol
(
self
):
# Lookup based on either enum or message.
# Lookup based on either enum or message.
self
.
assertEqual
s
(
self
.
assertEqual
(
'google/protobuf/unittest.proto'
,
'google/protobuf/unittest.proto'
,
self
.
_Database
()
.
pool
.
FindFileContainingSymbol
(
self
.
_Database
()
.
pool
.
FindFileContainingSymbol
(
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
name
)
'protobuf_unittest.TestAllTypes.NestedEnum'
)
.
name
)
self
.
assertEqual
s
(
self
.
assertEqual
(
'google/protobuf/unittest.proto'
,
'google/protobuf/unittest.proto'
,
self
.
_Database
()
.
pool
.
FindFileContainingSymbol
(
self
.
_Database
()
.
pool
.
FindFileContainingSymbol
(
'protobuf_unittest.TestAllTypes'
)
.
name
)
'protobuf_unittest.TestAllTypes'
)
.
name
)
def
testFindFileByName
(
self
):
def
testFindFileByName
(
self
):
self
.
assertEqual
s
(
self
.
assertEqual
(
'google/protobuf/unittest.proto'
,
'google/protobuf/unittest.proto'
,
self
.
_Database
()
.
pool
.
FindFileByName
(
self
.
_Database
()
.
pool
.
FindFileByName
(
'google/protobuf/unittest.proto'
)
.
name
)
'google/protobuf/unittest.proto'
)
.
name
)
...
...
python/google/protobuf/internal/text_encoding_test.py
View file @
a2abc947
...
@@ -54,15 +54,15 @@ TEST_VALUES = [
...
@@ -54,15 +54,15 @@ TEST_VALUES = [
class
TextEncodingTestCase
(
unittest
.
TestCase
):
class
TextEncodingTestCase
(
unittest
.
TestCase
):
def
testCEscape
(
self
):
def
testCEscape
(
self
):
for
escaped
,
escaped_utf8
,
unescaped
in
TEST_VALUES
:
for
escaped
,
escaped_utf8
,
unescaped
in
TEST_VALUES
:
self
.
assertEqual
s
(
escaped
,
self
.
assertEqual
(
escaped
,
text_encoding
.
CEscape
(
unescaped
,
as_utf8
=
False
))
text_encoding
.
CEscape
(
unescaped
,
as_utf8
=
False
))
self
.
assertEqual
s
(
escaped_utf8
,
self
.
assertEqual
(
escaped_utf8
,
text_encoding
.
CEscape
(
unescaped
,
as_utf8
=
True
))
text_encoding
.
CEscape
(
unescaped
,
as_utf8
=
True
))
def
testCUnescape
(
self
):
def
testCUnescape
(
self
):
for
escaped
,
escaped_utf8
,
unescaped
in
TEST_VALUES
:
for
escaped
,
escaped_utf8
,
unescaped
in
TEST_VALUES
:
self
.
assertEqual
s
(
unescaped
,
text_encoding
.
CUnescape
(
escaped
))
self
.
assertEqual
(
unescaped
,
text_encoding
.
CUnescape
(
escaped
))
self
.
assertEqual
s
(
unescaped
,
text_encoding
.
CUnescape
(
escaped_utf8
))
self
.
assertEqual
(
unescaped
,
text_encoding
.
CUnescape
(
escaped_utf8
))
if
__name__
==
"__main__"
:
if
__name__
==
"__main__"
:
...
...
python/google/protobuf/internal/text_format_test.py
View file @
a2abc947
...
@@ -233,7 +233,7 @@ class TextFormatTest(unittest.TestCase):
...
@@ -233,7 +233,7 @@ class TextFormatTest(unittest.TestCase):
parsed_message
=
unittest_pb2
.
TestAllTypes
()
parsed_message
=
unittest_pb2
.
TestAllTypes
()
r
=
text_format
.
Parse
(
wire_text
,
parsed_message
)
r
=
text_format
.
Parse
(
wire_text
,
parsed_message
)
self
.
assertIs
(
r
,
parsed_message
)
self
.
assertIs
(
r
,
parsed_message
)
self
.
assertEqual
s
(
message
,
parsed_message
)
self
.
assertEqual
(
message
,
parsed_message
)
# Test as_utf8 = True.
# Test as_utf8 = True.
wire_text
=
text_format
.
MessageToString
(
wire_text
=
text_format
.
MessageToString
(
...
@@ -241,7 +241,7 @@ class TextFormatTest(unittest.TestCase):
...
@@ -241,7 +241,7 @@ class TextFormatTest(unittest.TestCase):
parsed_message
=
unittest_pb2
.
TestAllTypes
()
parsed_message
=
unittest_pb2
.
TestAllTypes
()
r
=
text_format
.
Parse
(
wire_text
,
parsed_message
)
r
=
text_format
.
Parse
(
wire_text
,
parsed_message
)
self
.
assertIs
(
r
,
parsed_message
)
self
.
assertIs
(
r
,
parsed_message
)
self
.
assertEqual
s
(
message
,
parsed_message
,
self
.
assertEqual
(
message
,
parsed_message
,
'
\n
%
s !=
%
s'
%
(
message
,
parsed_message
))
'
\n
%
s !=
%
s'
%
(
message
,
parsed_message
))
def
testPrintRawUtf8String
(
self
):
def
testPrintRawUtf8String
(
self
):
...
@@ -251,7 +251,7 @@ class TextFormatTest(unittest.TestCase):
...
@@ -251,7 +251,7 @@ class TextFormatTest(unittest.TestCase):
self
.
CompareToGoldenText
(
text
,
'repeated_string: "
\303\274\352\234\237
"
\n
'
)
self
.
CompareToGoldenText
(
text
,
'repeated_string: "
\303\274\352\234\237
"
\n
'
)
parsed_message
=
unittest_pb2
.
TestAllTypes
()
parsed_message
=
unittest_pb2
.
TestAllTypes
()
text_format
.
Parse
(
text
,
parsed_message
)
text_format
.
Parse
(
text
,
parsed_message
)
self
.
assertEqual
s
(
message
,
parsed_message
,
self
.
assertEqual
(
message
,
parsed_message
,
'
\n
%
s !=
%
s'
%
(
message
,
parsed_message
))
'
\n
%
s !=
%
s'
%
(
message
,
parsed_message
))
def
testPrintFloatFormat
(
self
):
def
testPrintFloatFormat
(
self
):
...
@@ -309,7 +309,7 @@ class TextFormatTest(unittest.TestCase):
...
@@ -309,7 +309,7 @@ class TextFormatTest(unittest.TestCase):
message
=
unittest_pb2
.
TestAllTypes
()
message
=
unittest_pb2
.
TestAllTypes
()
test_util
.
SetAllFields
(
message
)
test_util
.
SetAllFields
(
message
)
self
.
assertEqual
s
(
message
,
parsed_message
)
self
.
assertEqual
(
message
,
parsed_message
)
def
testParseGoldenExtensions
(
self
):
def
testParseGoldenExtensions
(
self
):
golden_text
=
'
\n
'
.
join
(
self
.
ReadGolden
(
golden_text
=
'
\n
'
.
join
(
self
.
ReadGolden
(
...
@@ -319,7 +319,7 @@ class TextFormatTest(unittest.TestCase):
...
@@ -319,7 +319,7 @@ class TextFormatTest(unittest.TestCase):
message
=
unittest_pb2
.
TestAllExtensions
()
message
=
unittest_pb2
.
TestAllExtensions
()
test_util
.
SetAllExtensions
(
message
)
test_util
.
SetAllExtensions
(
message
)
self
.
assertEqual
s
(
message
,
parsed_message
)
self
.
assertEqual
(
message
,
parsed_message
)
def
testParseAllFields
(
self
):
def
testParseAllFields
(
self
):
message
=
unittest_pb2
.
TestAllTypes
()
message
=
unittest_pb2
.
TestAllTypes
()
...
@@ -360,8 +360,8 @@ class TextFormatTest(unittest.TestCase):
...
@@ -360,8 +360,8 @@ class TextFormatTest(unittest.TestCase):
text_format
.
Parse
(
text
,
message
)
text_format
.
Parse
(
text
,
message
)
ext1
=
unittest_mset_pb2
.
TestMessageSetExtension1
.
message_set_extension
ext1
=
unittest_mset_pb2
.
TestMessageSetExtension1
.
message_set_extension
ext2
=
unittest_mset_pb2
.
TestMessageSetExtension2
.
message_set_extension
ext2
=
unittest_mset_pb2
.
TestMessageSetExtension2
.
message_set_extension
self
.
assertEqual
s
(
23
,
message
.
message_set
.
Extensions
[
ext1
]
.
i
)
self
.
assertEqual
(
23
,
message
.
message_set
.
Extensions
[
ext1
]
.
i
)
self
.
assertEqual
s
(
'foo'
,
message
.
message_set
.
Extensions
[
ext2
]
.
str
)
self
.
assertEqual
(
'foo'
,
message
.
message_set
.
Extensions
[
ext2
]
.
str
)
def
testParseExotic
(
self
):
def
testParseExotic
(
self
):
message
=
unittest_pb2
.
TestAllTypes
()
message
=
unittest_pb2
.
TestAllTypes
()
...
@@ -408,7 +408,7 @@ class TextFormatTest(unittest.TestCase):
...
@@ -408,7 +408,7 @@ class TextFormatTest(unittest.TestCase):
message
=
unittest_pb2
.
TestAllTypes
()
message
=
unittest_pb2
.
TestAllTypes
()
text
=
''
text
=
''
text_format
.
Parse
(
text
,
message
)
text_format
.
Parse
(
text
,
message
)
self
.
assertEqual
s
(
unittest_pb2
.
TestAllTypes
(),
message
)
self
.
assertEqual
(
unittest_pb2
.
TestAllTypes
(),
message
)
def
testParseInvalidUtf8
(
self
):
def
testParseInvalidUtf8
(
self
):
message
=
unittest_pb2
.
TestAllTypes
()
message
=
unittest_pb2
.
TestAllTypes
()
...
@@ -581,7 +581,7 @@ class TextFormatTest(unittest.TestCase):
...
@@ -581,7 +581,7 @@ class TextFormatTest(unittest.TestCase):
message
=
unittest_pb2
.
TestAllTypes
()
message
=
unittest_pb2
.
TestAllTypes
()
test_util
.
SetAllFields
(
message
)
test_util
.
SetAllFields
(
message
)
self
.
assertEqual
s
(
message
,
parsed_message
)
self
.
assertEqual
(
message
,
parsed_message
)
def
testMergeLinesGolden
(
self
):
def
testMergeLinesGolden
(
self
):
opened
=
self
.
ReadGolden
(
'text_format_unittest_data.txt'
)
opened
=
self
.
ReadGolden
(
'text_format_unittest_data.txt'
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment