unknown_field_set_unittest.cc revision 5821806d5e7f356e8fa4b058a389a808ea183019
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Protocol Buffers - Google's data interchange format 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright 2008 Google Inc. All rights reserved. 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// http://code.google.com/p/protobuf/ 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Redistribution and use in source and binary forms, with or without 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// modification, are permitted provided that the following conditions are 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// met: 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Redistributions of source code must retain the above copyright 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// notice, this list of conditions and the following disclaimer. 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Redistributions in binary form must reproduce the above 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// copyright notice, this list of conditions and the following disclaimer 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// in the documentation and/or other materials provided with the 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// distribution. 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Neither the name of Google Inc. nor the names of its 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// contributors may be used to endorse or promote products derived from 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// this software without specific prior written permission. 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Author: kenton@google.com (Kenton Varda) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Based on original Protocol Buffers design by 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Sanjay Ghemawat, Jeff Dean, and others. 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This test is testing a lot more than just the UnknownFieldSet class. It 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// tests handling of unknown fields throughout the system. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/unknown_field_set.h> 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/descriptor.h> 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/io/coded_stream.h> 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/io/zero_copy_stream_impl.h> 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/wire_format.h> 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/unittest.pb.h> 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/test_util.h> 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/stubs/common.h> 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/testing/googletest.h> 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <gtest/gtest.h> 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/stubs/stl_util-inl.h> 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace google { 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace protobuf { 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using internal::WireFormat; 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class UnknownFieldSetTest : public testing::Test { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUp() { 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) descriptor_ = unittest::TestAllTypes::descriptor(); 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestUtil::SetAllFields(&all_fields_); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) all_fields_.SerializeToString(&all_fields_data_); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(empty_message_.ParseFromString(all_fields_data_)); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields_ = empty_message_.mutable_unknown_fields(); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownField* GetField(const string& name) { 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FieldDescriptor* field = descriptor_->FindFieldByName(name); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (field == NULL) return NULL; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < unknown_fields_->field_count(); i++) { 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (unknown_fields_->field(i).number() == field->number()) { 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return &unknown_fields_->field(i); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Constructs a protocol buffer which contains fields with all the same 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // numbers as all_fields_data_ except that each field is some other wire 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // type. 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string GetBizarroData() { 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage bizarro_message; 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnknownFieldSet* bizarro_unknown_fields = 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bizarro_message.mutable_unknown_fields(); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < unknown_fields_->field_count(); i++) { 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownField& unknown_field = unknown_fields_->field(i); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (unknown_field.type() == UnknownField::TYPE_VARINT) { 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bizarro_unknown_fields->AddFixed32(unknown_field.number(), 1); 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bizarro_unknown_fields->AddVarint(unknown_field.number(), 1); 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string data; 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(bizarro_message.SerializeToString(&data)); 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return data; 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Descriptor* descriptor_; 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestAllTypes all_fields_; 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string all_fields_data_; 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // An empty message that has been parsed from all_fields_data_. So, it has 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // unknown fields of every type. 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage empty_message_; 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnknownFieldSet* unknown_fields_; 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, AllFieldsPresent) { 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // All fields of TestAllTypes should be present, in numeric order (because 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // that's the order we parsed them in). Fields that are not valid field 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // numbers of TestAllTypes should NOT be present. 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pos = 0; 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < 1000; i++) { 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FieldDescriptor* field = descriptor_->FindFieldByNumber(i); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (field != NULL) { 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_LT(pos, unknown_fields_->field_count()); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(i, unknown_fields_->field(pos++).number()); 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (field->is_repeated()) { 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Should have a second instance. 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_LT(pos, unknown_fields_->field_count()); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(i, unknown_fields_->field(pos++).number()); 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(unknown_fields_->field_count(), pos); 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, Varint) { 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownField* field = GetField("optional_int32"); 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(field != NULL); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_VARINT, field->type()); 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(all_fields_.optional_int32(), field->varint()); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, Fixed32) { 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownField* field = GetField("optional_fixed32"); 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(field != NULL); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_FIXED32, field->type()); 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(all_fields_.optional_fixed32(), field->fixed32()); 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, Fixed64) { 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownField* field = GetField("optional_fixed64"); 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(field != NULL); 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_FIXED64, field->type()); 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(all_fields_.optional_fixed64(), field->fixed64()); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, LengthDelimited) { 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownField* field = GetField("optional_string"); 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(field != NULL); 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_LENGTH_DELIMITED, field->type()); 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(all_fields_.optional_string(), field->length_delimited()); 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, Group) { 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownField* field = GetField("optionalgroup"); 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(field != NULL); 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_GROUP, field->type()); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(1, field->group().field_count()); 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownField& nested_field = field->group().field(0); 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FieldDescriptor* nested_field_descriptor = 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestAllTypes::OptionalGroup::descriptor()->FindFieldByName("a"); 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(nested_field_descriptor != NULL); 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(nested_field_descriptor->number(), nested_field.number()); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_VARINT, nested_field.type()); 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(all_fields_.optionalgroup().a(), nested_field.varint()); 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, SerializeFastAndSlowAreEquivalent) { 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int size = WireFormat::ComputeUnknownFieldsSize( 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) empty_message_.unknown_fields()); 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string slow_buffer; 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string fast_buffer; 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) slow_buffer.resize(size); 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fast_buffer.resize(size); 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint8* target = reinterpret_cast<uint8*>(string_as_array(&fast_buffer)); 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint8* result = WireFormat::SerializeUnknownFieldsToArray( 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) empty_message_.unknown_fields(), target); 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(size, result - target); 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::ArrayOutputStream raw_stream(string_as_array(&slow_buffer), size, 1); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::CodedOutputStream output_stream(&raw_stream); 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WireFormat::SerializeUnknownFields(empty_message_.unknown_fields(), 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &output_stream); 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_FALSE(output_stream.HadError()); 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(fast_buffer == slow_buffer); 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, Serialize) { 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Check that serializing the UnknownFieldSet produces the original data 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // again. 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string data; 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) empty_message_.SerializeToString(&data); 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Don't use EXPECT_EQ because we don't want to dump raw binary data to 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // stdout. 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(data == all_fields_data_); 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, ParseViaReflection) { 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make sure fields are properly parsed to the UnknownFieldSet when parsing 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // via reflection. 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage message; 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::ArrayInputStream raw_input(all_fields_data_.data(), 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) all_fields_data_.size()); 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::CodedInputStream input(&raw_input); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(WireFormat::ParseAndMergePartial(&input, &message)); 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(message.DebugString(), empty_message_.DebugString()); 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, SerializeViaReflection) { 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make sure fields are properly written from the UnknownFieldSet when 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // serializing via reflection. 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string data; 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::StringOutputStream raw_output(&data); 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::CodedOutputStream output(&raw_output); 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int size = WireFormat::ByteSize(empty_message_); 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WireFormat::SerializeWithCachedSizes(empty_message_, size, &output); 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_FALSE(output.HadError()); 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Don't use EXPECT_EQ because we don't want to dump raw binary data to 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // stdout. 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(data == all_fields_data_); 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, CopyFrom) { 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage message; 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message.CopyFrom(empty_message_); 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(empty_message_.DebugString(), message.DebugString()); 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, Swap) { 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage other_message; 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(other_message.ParseFromString(GetBizarroData())); 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_GT(empty_message_.unknown_fields().field_count(), 0); 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_GT(other_message.unknown_fields().field_count(), 0); 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string debug_string = empty_message_.DebugString(); 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string other_debug_string = other_message.DebugString(); 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_NE(debug_string, other_debug_string); 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) empty_message_.Swap(&other_message); 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(debug_string, other_message.DebugString()); 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(other_debug_string, empty_message_.DebugString()); 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, SwapWithSelf) { 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string debug_string = empty_message_.DebugString(); 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_GT(empty_message_.unknown_fields().field_count(), 0); 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) empty_message_.Swap(&empty_message_); 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_GT(empty_message_.unknown_fields().field_count(), 0); 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(debug_string, empty_message_.DebugString()); 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, MergeFrom) { 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage source, destination; 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) destination.mutable_unknown_fields()->AddVarint(1, 1); 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) destination.mutable_unknown_fields()->AddVarint(3, 2); 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) source.mutable_unknown_fields()->AddVarint(2, 3); 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) source.mutable_unknown_fields()->AddVarint(3, 4); 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) destination.MergeFrom(source); 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ( 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: The ordering of fields here depends on the ordering of adds 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and merging, above. 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "1: 1\n" 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "3: 2\n" 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "2: 3\n" 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "3: 4\n", 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) destination.DebugString()); 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, Clear) { 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Clear the set. 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) empty_message_.Clear(); 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, unknown_fields_->field_count()); 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, ParseKnownAndUnknown) { 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Test mixing known and unknown fields when parsing. 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage source; 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) source.mutable_unknown_fields()->AddVarint(123456, 654321); 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string data; 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(source.SerializeToString(&data)); 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestAllTypes destination; 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(destination.ParseFromString(all_fields_data_ + data)); 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestUtil::ExpectAllFieldsSet(destination); 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(1, destination.unknown_fields().field_count()); 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_VARINT, 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) destination.unknown_fields().field(0).type()); 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(654321, destination.unknown_fields().field(0).varint()); 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, WrongTypeTreatedAsUnknown) { 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Test that fields of the wrong wire type are treated like unknown fields 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // when parsing. 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestAllTypes all_types_message; 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage empty_message; 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string bizarro_data = GetBizarroData(); 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(all_types_message.ParseFromString(bizarro_data)); 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(empty_message.ParseFromString(bizarro_data)); 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // All fields should have been interpreted as unknown, so the debug strings 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // should be the same. 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(empty_message.DebugString(), all_types_message.DebugString()); 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, WrongTypeTreatedAsUnknownViaReflection) { 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Same as WrongTypeTreatedAsUnknown but via the reflection interface. 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestAllTypes all_types_message; 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage empty_message; 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string bizarro_data = GetBizarroData(); 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::ArrayInputStream raw_input(bizarro_data.data(), bizarro_data.size()); 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::CodedInputStream input(&raw_input); 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(WireFormat::ParseAndMergePartial(&input, &all_types_message)); 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(empty_message.ParseFromString(bizarro_data)); 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(empty_message.DebugString(), all_types_message.DebugString()); 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, UnknownExtensions) { 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make sure fields are properly parsed to the UnknownFieldSet even when 3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // they are declared as extension numbers. 3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessageWithExtensions message; 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(message.ParseFromString(all_fields_data_)); 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(message.DebugString(), empty_message_.DebugString()); 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, UnknownExtensionsReflection) { 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Same as UnknownExtensions except parsing via reflection. 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessageWithExtensions message; 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::ArrayInputStream raw_input(all_fields_data_.data(), 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) all_fields_data_.size()); 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io::CodedInputStream input(&raw_input); 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(WireFormat::ParseAndMergePartial(&input, &message)); 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(message.DebugString(), empty_message_.DebugString()); 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, WrongExtensionTypeTreatedAsUnknown) { 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Test that fields of the wrong wire type are treated like unknown fields 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // when parsing extensions. 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestAllExtensions all_extensions_message; 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage empty_message; 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string bizarro_data = GetBizarroData(); 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(all_extensions_message.ParseFromString(bizarro_data)); 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(empty_message.ParseFromString(bizarro_data)); 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // All fields should have been interpreted as unknown, so the debug strings 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // should be the same. 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(empty_message.DebugString(), all_extensions_message.DebugString()); 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, UnknownEnumValue) { 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) using unittest::TestAllTypes; 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) using unittest::TestAllExtensions; 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) using unittest::TestEmptyMessage; 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FieldDescriptor* singular_field = 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestAllTypes::descriptor()->FindFieldByName("optional_nested_enum"); 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FieldDescriptor* repeated_field = 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestAllTypes::descriptor()->FindFieldByName("repeated_nested_enum"); 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(singular_field != NULL); 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(repeated_field != NULL); 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string data; 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestEmptyMessage empty_message; 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnknownFieldSet* unknown_fields = empty_message.mutable_unknown_fields(); 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields->AddVarint(singular_field->number(), TestAllTypes::BAR); 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields->AddVarint(singular_field->number(), 5); // not valid 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields->AddVarint(repeated_field->number(), TestAllTypes::FOO); 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields->AddVarint(repeated_field->number(), 4); // not valid 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields->AddVarint(repeated_field->number(), TestAllTypes::BAZ); 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields->AddVarint(repeated_field->number(), 6); // not valid 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) empty_message.SerializeToString(&data); 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestAllTypes message; 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(message.ParseFromString(data)); 4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(TestAllTypes::BAR, message.optional_nested_enum()); 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(2, message.repeated_nested_enum_size()); 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(TestAllTypes::FOO, message.repeated_nested_enum(0)); 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(TestAllTypes::BAZ, message.repeated_nested_enum(1)); 4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownFieldSet& unknown_fields = message.unknown_fields(); 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(3, unknown_fields.field_count()); 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(singular_field->number(), unknown_fields.field(0).number()); 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_VARINT, unknown_fields.field(0).type()); 4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(5, unknown_fields.field(0).varint()); 4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(repeated_field->number(), unknown_fields.field(1).number()); 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_VARINT, unknown_fields.field(1).type()); 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(4, unknown_fields.field(1).varint()); 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(repeated_field->number(), unknown_fields.field(2).number()); 4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_VARINT, unknown_fields.field(2).type()); 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(6, unknown_fields.field(2).varint()); 4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) using unittest::optional_nested_enum_extension; 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) using unittest::repeated_nested_enum_extension; 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestAllExtensions message; 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(message.ParseFromString(data)); 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(TestAllTypes::BAR, 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message.GetExtension(optional_nested_enum_extension)); 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(2, message.ExtensionSize(repeated_nested_enum_extension)); 4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(TestAllTypes::FOO, 4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message.GetExtension(repeated_nested_enum_extension, 0)); 4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(TestAllTypes::BAZ, 4525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message.GetExtension(repeated_nested_enum_extension, 1)); 4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const UnknownFieldSet& unknown_fields = message.unknown_fields(); 4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(3, unknown_fields.field_count()); 4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(singular_field->number(), unknown_fields.field(0).number()); 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_VARINT, unknown_fields.field(0).type()); 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(5, unknown_fields.field(0).varint()); 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(repeated_field->number(), unknown_fields.field(1).number()); 4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_VARINT, unknown_fields.field(1).type()); 4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(4, unknown_fields.field(1).varint()); 4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(repeated_field->number(), unknown_fields.field(2).number()); 4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(UnknownField::TYPE_VARINT, unknown_fields.field(2).type()); 4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(6, unknown_fields.field(2).varint()); 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, SpaceUsed) { 4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unittest::TestEmptyMessage empty_message; 4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make sure an unknown field set has zero space used until a field is 4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // actually added. 4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int base_size = empty_message.SpaceUsed(); 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnknownFieldSet* unknown_fields = empty_message.mutable_unknown_fields(); 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(base_size, empty_message.SpaceUsed()); 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make sure each thing we add to the set increases the SpaceUsed(). 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields->AddVarint(1, 0); 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_LT(base_size, empty_message.SpaceUsed()); 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base_size = empty_message.SpaceUsed(); 4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string* str = unknown_fields->AddLengthDelimited(1); 4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_LT(base_size, empty_message.SpaceUsed()); 4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base_size = empty_message.SpaceUsed(); 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) str->assign(sizeof(string) + 1, 'x'); 4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_LT(base_size, empty_message.SpaceUsed()); 4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base_size = empty_message.SpaceUsed(); 4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnknownFieldSet* group = unknown_fields->AddGroup(1); 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_LT(base_size, empty_message.SpaceUsed()); 4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base_size = empty_message.SpaceUsed(); 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) group->AddVarint(1, 0); 4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_LT(base_size, empty_message.SpaceUsed()); 4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(UnknownFieldSetTest, Empty) { 5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnknownFieldSet unknown_fields; 5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(unknown_fields.empty()); 5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields.AddVarint(6, 123); 5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(unknown_fields.empty()); 5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unknown_fields.Clear(); 5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(unknown_fields.empty()); 5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace protobuf 5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace google 513