Example #1
0
 def testMergeAllExtensions(self):
     message = unittest_pb2.TestAllExtensions()
     test_util.SetAllExtensions(message)
     ascii_text = text_format.MessageToString(message)
     parsed_message = unittest_pb2.TestAllExtensions()
     text_format.Merge(ascii_text, parsed_message)
     self.assertEqual(message, parsed_message)
Example #2
0
 def testGoldenExtensions(self):
     golden_data = test_util.GoldenFile('golden_message').read()
     golden_message = unittest_pb2.TestAllExtensions()
     golden_message.ParseFromString(golden_data)
     all_set = unittest_pb2.TestAllExtensions()
     test_util.SetAllExtensions(all_set)
     self.assertEquals(all_set, golden_message)
     self.assertTrue(golden_message.SerializeToString() == golden_data)
Example #3
0
 def testExtensionToDictAndBackWithScalar(self):
     message = unittest_pb2.TestAllExtensions()
     ext1 = unittest_pb2.TestNestedExtension.test
     message.Extensions[ext1] = 'data'
     message_dict = json_format.MessageToDict(message)
     parsed_message = unittest_pb2.TestAllExtensions()
     json_format.ParseDict(message_dict, parsed_message)
     self.assertEqual(message, parsed_message)
  def testParseGoldenExtensions(self):
    golden_text = '\n'.join(self.ReadGolden(
        'text_format_unittest_extensions_data.txt'))
    parsed_message = unittest_pb2.TestAllExtensions()
    text_format.Parse(golden_text, parsed_message)

    message = unittest_pb2.TestAllExtensions()
    test_util.SetAllExtensions(message)
    self.assertEquals(message, parsed_message)
 def testGoldenExtensions(self):
   golden_data = test_util.GoldenFileData('golden_message')
   golden_message = unittest_pb2.TestAllExtensions()
   golden_message.ParseFromString(golden_data)
   all_set = unittest_pb2.TestAllExtensions()
   test_util.SetAllExtensions(all_set)
   self.assertEqual(all_set, golden_message)
   self.assertEqual(golden_data, golden_message.SerializeToString())
   golden_copy = copy.deepcopy(golden_message)
   self.assertEqual(golden_data, golden_copy.SerializeToString())
 def testPrintAllExtensionsPointy(self):
   message = unittest_pb2.TestAllExtensions()
   test_util.SetAllExtensions(message)
   self.CompareToGoldenFile(
       self.RemoveRedundantZeros(text_format.MessageToString(
           message, pointy_brackets=True)),
       'text_format_unittest_extensions_data_pointy.txt')
Example #7
0
 def testMergeDuplicateExtensionScalars(self):
     message = unittest_pb2.TestAllExtensions()
     text = ('[protobuf_unittest.optional_int32_extension]: 42 '
             '[protobuf_unittest.optional_int32_extension]: 67')
     text_format.Merge(text, message)
     self.assertEqual(
         67, message.Extensions[unittest_pb2.optional_int32_extension])
 def testMergeBadExtension(self):
     message = unittest_pb2.TestAllExtensions()
     text = '[unknown_extension]: 8\n'
     self.assertRaises(text_format.ParseError, text_format.Merge, text,
                       message)
     message = unittest_pb2.TestAllTypes()
     self.assertRaisesWithMessage(text_format.ParseError, (
         '1:2 : Message type "protobuf_unittest.TestAllTypes" does not have '
         'extensions.'), text_format.Merge, text, message)
 def testParseRepeatedExtensionScalars(self):
   message = unittest_pb2.TestAllExtensions()
   text = ('[protobuf_unittest.optional_int32_extension]: 42 '
           '[protobuf_unittest.optional_int32_extension]: 67')
   self.assertRaisesWithLiteralMatch(
       text_format.ParseError,
       ('1:96 : Message type "protobuf_unittest.TestAllExtensions" '
        'should not have multiple '
        '"protobuf_unittest.optional_int32_extension" extensions.'),
       text_format.Parse, text, message)
Example #10
0
 def testParseDuplicateExtensionScalars(self):
     message = unittest_pb2.TestAllExtensions()
     text = ('[protobuf_unittest.optional_int32_extension]: 42 '
             '[protobuf_unittest.optional_int32_extension]: 67')
     six.assertRaisesRegex(
         self, text_format.ParseError,
         ('1:96 : Message type "protobuf_unittest.TestAllExtensions" '
          'should not have multiple '
          '"protobuf_unittest.optional_int32_extension" extensions.'),
         text_format.Parse, text, message)
Example #11
0
 def testPrintBadEnumValueExtensions(self):
     message = unittest_pb2.TestAllExtensions()
     message.Extensions[unittest_pb2.optional_nested_enum_extension] = 100
     message.Extensions[unittest_pb2.optional_foreign_enum_extension] = 101
     message.Extensions[unittest_pb2.optional_import_enum_extension] = 102
     self.CompareToGoldenText(
         text_format.MessageToString(message),
         '[protobuf_unittest.optional_nested_enum_extension]: 100\n'
         '[protobuf_unittest.optional_foreign_enum_extension]: 101\n'
         '[protobuf_unittest.optional_import_enum_extension]: 102\n')
Example #12
0
 def testParseBadExtension(self):
     message = unittest_pb2.TestAllExtensions()
     text = '[unknown_extension]: 8\n'
     six.assertRaisesRegex(
         self, text_format.ParseError,
         '1:2 : Extension "unknown_extension" not registered.',
         text_format.Parse, text, message)
     message = unittest_pb2.TestAllTypes()
     six.assertRaisesRegex(self, text_format.ParseError, (
         '1:2 : Message type "protobuf_unittest.TestAllTypes" does not have '
         'extensions.'), text_format.Parse, text, message)
Example #13
0
 def testPrintAllExtensions(self):
     message = unittest_pb2.TestAllExtensions()
     test_util.SetAllExtensions(message)
     self.CompareToGoldenFile(text_format.MessageToString(message),
                              'text_format_unittest_extensions_data.txt')