Index: third_party/protobuf/python/google/protobuf/internal/unknown_fields_test.py |
diff --git a/third_party/protobuf/python/google/protobuf/internal/unknown_fields_test.py b/third_party/protobuf/python/google/protobuf/internal/unknown_fields_test.py |
index 9685b8b4f031f2c5b7a925d07916caefbd341a94..84073f1c5344f934d4def999fa209700fc7dd39d 100755 |
--- a/third_party/protobuf/python/google/protobuf/internal/unknown_fields_test.py |
+++ b/third_party/protobuf/python/google/protobuf/internal/unknown_fields_test.py |
@@ -36,7 +36,7 @@ |
__author__ = 'bohdank@google.com (Bohdan Koval)' |
try: |
- import unittest2 as unittest |
+ import unittest2 as unittest #PY26 |
except ImportError: |
import unittest |
from google.protobuf import unittest_mset_pb2 |
@@ -119,6 +119,26 @@ class UnknownFieldsTest(unittest.TestCase): |
message.ParseFromString(self.all_fields.SerializeToString()) |
self.assertNotEqual(self.empty_message, message) |
+ def testDiscardUnknownFields(self): |
+ self.empty_message.DiscardUnknownFields() |
+ self.assertEqual(b'', self.empty_message.SerializeToString()) |
+ # Test message field and repeated message field. |
+ message = unittest_pb2.TestAllTypes() |
+ other_message = unittest_pb2.TestAllTypes() |
+ other_message.optional_string = 'discard' |
+ message.optional_nested_message.ParseFromString( |
+ other_message.SerializeToString()) |
+ message.repeated_nested_message.add().ParseFromString( |
+ other_message.SerializeToString()) |
+ self.assertNotEqual( |
+ b'', message.optional_nested_message.SerializeToString()) |
+ self.assertNotEqual( |
+ b'', message.repeated_nested_message[0].SerializeToString()) |
+ message.DiscardUnknownFields() |
+ self.assertEqual(b'', message.optional_nested_message.SerializeToString()) |
+ self.assertEqual( |
+ b'', message.repeated_nested_message[0].SerializeToString()) |
+ |
class UnknownFieldsAccessorsTest(unittest.TestCase): |