Index: third_party/protobuf/python/google/protobuf/internal/generator_test.py |
=================================================================== |
--- third_party/protobuf/python/google/protobuf/internal/generator_test.py (revision 216642) |
+++ third_party/protobuf/python/google/protobuf/internal/generator_test.py (working copy) |
@@ -42,8 +42,10 @@ |
__author__ = 'robinson@google.com (Will Robinson)' |
import unittest |
+from google.protobuf.internal import test_bad_identifiers_pb2 |
from google.protobuf import unittest_custom_options_pb2 |
from google.protobuf import unittest_import_pb2 |
+from google.protobuf import unittest_import_public_pb2 |
from google.protobuf import unittest_mset_pb2 |
from google.protobuf import unittest_pb2 |
from google.protobuf import unittest_no_generic_services_pb2 |
@@ -239,6 +241,29 @@ |
unittest_pb2._TESTALLTYPES_NESTEDMESSAGE.name in |
file_type.message_types_by_name) |
+ def testPublicImports(self): |
+ # Test public imports as embedded message. |
+ all_type_proto = unittest_pb2.TestAllTypes() |
+ self.assertEqual(0, all_type_proto.optional_public_import_message.e) |
+ # PublicImportMessage is actually defined in unittest_import_public_pb2 |
+ # module, and is public imported by unittest_import_pb2 module. |
+ public_import_proto = unittest_import_pb2.PublicImportMessage() |
+ self.assertEqual(0, public_import_proto.e) |
+ self.assertTrue(unittest_import_public_pb2.PublicImportMessage is |
+ unittest_import_pb2.PublicImportMessage) |
+ |
+ def testBadIdentifiers(self): |
+ # We're just testing that the code was imported without problems. |
+ message = test_bad_identifiers_pb2.TestBadIdentifiers() |
+ self.assertEqual(message.Extensions[test_bad_identifiers_pb2.message], |
+ "foo") |
+ self.assertEqual(message.Extensions[test_bad_identifiers_pb2.descriptor], |
+ "bar") |
+ self.assertEqual(message.Extensions[test_bad_identifiers_pb2.reflection], |
+ "baz") |
+ self.assertEqual(message.Extensions[test_bad_identifiers_pb2.service], |
+ "qux") |
+ |
if __name__ == '__main__': |
unittest.main() |