| Index: third_party/protobuf/src/google/protobuf/repeated_field_reflection_unittest.cc
|
| ===================================================================
|
| --- third_party/protobuf/src/google/protobuf/repeated_field_reflection_unittest.cc (revision 0)
|
| +++ third_party/protobuf/src/google/protobuf/repeated_field_reflection_unittest.cc (revision 0)
|
| @@ -0,0 +1,193 @@
|
| +// Protocol Buffers - Google's data interchange format
|
| +// Copyright 2008 Google Inc. All rights reserved.
|
| +// http://code.google.com/p/protobuf/
|
| +//
|
| +// Redistribution and use in source and binary forms, with or without
|
| +// modification, are permitted provided that the following conditions are
|
| +// met:
|
| +//
|
| +// * Redistributions of source code must retain the above copyright
|
| +// notice, this list of conditions and the following disclaimer.
|
| +// * Redistributions in binary form must reproduce the above
|
| +// copyright notice, this list of conditions and the following disclaimer
|
| +// in the documentation and/or other materials provided with the
|
| +// distribution.
|
| +// * Neither the name of Google Inc. nor the names of its
|
| +// contributors may be used to endorse or promote products derived from
|
| +// this software without specific prior written permission.
|
| +//
|
| +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| +// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| +// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| +// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| +// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| +// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| +// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| +// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| +
|
| +// Author: tgs@google.com (Tom Szymanski)
|
| +//
|
| +// Test reflection methods for aggregate access to Repeated[Ptr]Fields.
|
| +// This test proto2 methods on a proto2 layout.
|
| +
|
| +#include <google/protobuf/stubs/common.h>
|
| +#include <google/protobuf/stubs/stringprintf.h>
|
| +#include <google/protobuf/unittest.pb.h>
|
| +#include <google/protobuf/test_util.h>
|
| +#include <gtest/gtest.h>
|
| +
|
| +namespace google {
|
| +namespace protobuf {
|
| +
|
| +using unittest::ForeignMessage;
|
| +using unittest::TestAllTypes;
|
| +using unittest::TestAllExtensions;
|
| +
|
| +namespace {
|
| +
|
| +static int Func(int i, int j) {
|
| + return i * j;
|
| +}
|
| +
|
| +static string StrFunc(int i, int j) {
|
| + string str;
|
| + SStringPrintf(&str, "%d", Func(i, 4));
|
| + return str;
|
| +}
|
| +
|
| +
|
| +TEST(RepeatedFieldReflectionTest, RegularFields) {
|
| + TestAllTypes message;
|
| + const Reflection* refl = message.GetReflection();
|
| + const Descriptor* desc = message.GetDescriptor();
|
| +
|
| + for (int i = 0; i < 10; ++i) {
|
| + message.add_repeated_int32(Func(i, 1));
|
| + message.add_repeated_double(Func(i, 2));
|
| + message.add_repeated_string(StrFunc(i, 5));
|
| + message.add_repeated_foreign_message()->set_c(Func(i, 6));
|
| + }
|
| +
|
| + // Get FieldDescriptors for all the fields of interest.
|
| + const FieldDescriptor* fd_repeated_int32 =
|
| + desc->FindFieldByName("repeated_int32");
|
| + const FieldDescriptor* fd_repeated_double =
|
| + desc->FindFieldByName("repeated_double");
|
| + const FieldDescriptor* fd_repeated_string =
|
| + desc->FindFieldByName("repeated_string");
|
| + const FieldDescriptor* fd_repeated_foreign_message =
|
| + desc->FindFieldByName("repeated_foreign_message");
|
| +
|
| + // Get RepeatedField objects for all fields of interest.
|
| + const RepeatedField<int32>& rf_int32 =
|
| + refl->GetRepeatedField<int32>(message, fd_repeated_int32);
|
| + const RepeatedField<double>& rf_double =
|
| + refl->GetRepeatedField<double>(message, fd_repeated_double);
|
| +
|
| + // Get mutable RepeatedField objects for all fields of interest.
|
| + RepeatedField<int32>* mrf_int32 =
|
| + refl->MutableRepeatedField<int32>(&message, fd_repeated_int32);
|
| + RepeatedField<double>* mrf_double =
|
| + refl->MutableRepeatedField<double>(&message, fd_repeated_double);
|
| +
|
| + // Get RepeatedPtrField objects for all fields of interest.
|
| + const RepeatedPtrField<string>& rpf_string =
|
| + refl->GetRepeatedPtrField<string>(message, fd_repeated_string);
|
| + const RepeatedPtrField<ForeignMessage>& rpf_foreign_message =
|
| + refl->GetRepeatedPtrField<ForeignMessage>(
|
| + message, fd_repeated_foreign_message);
|
| + const RepeatedPtrField<Message>& rpf_message =
|
| + refl->GetRepeatedPtrField<Message>(
|
| + message, fd_repeated_foreign_message);
|
| +
|
| + // Get mutable RepeatedPtrField objects for all fields of interest.
|
| + RepeatedPtrField<string>* mrpf_string =
|
| + refl->MutableRepeatedPtrField<string>(&message, fd_repeated_string);
|
| + RepeatedPtrField<ForeignMessage>* mrpf_foreign_message =
|
| + refl->MutableRepeatedPtrField<ForeignMessage>(
|
| + &message, fd_repeated_foreign_message);
|
| + RepeatedPtrField<Message>* mrpf_message =
|
| + refl->MutableRepeatedPtrField<Message>(
|
| + &message, fd_repeated_foreign_message);
|
| +
|
| + // Make sure we can do get and sets through the Repeated[Ptr]Field objects.
|
| + for (int i = 0; i < 10; ++i) {
|
| + // Check gets through const objects.
|
| + EXPECT_EQ(rf_int32.Get(i), Func(i, 1));
|
| + EXPECT_EQ(rf_double.Get(i), Func(i, 2));
|
| + EXPECT_EQ(rpf_string.Get(i), StrFunc(i, 5));
|
| + EXPECT_EQ(rpf_foreign_message.Get(i).c(), Func(i, 6));
|
| + EXPECT_EQ(down_cast<const ForeignMessage&>(rpf_message.Get(i)).c(),
|
| + Func(i, 6));
|
| +
|
| + // Check gets through mutable objects.
|
| + EXPECT_EQ(mrf_int32->Get(i), Func(i, 1));
|
| + EXPECT_EQ(mrf_double->Get(i), Func(i, 2));
|
| + EXPECT_EQ(mrpf_string->Get(i), StrFunc(i, 5));
|
| + EXPECT_EQ(mrpf_foreign_message->Get(i).c(), Func(i, 6));
|
| + EXPECT_EQ(down_cast<const ForeignMessage&>(mrpf_message->Get(i)).c(),
|
| + Func(i, 6));
|
| +
|
| + // Check sets through mutable objects.
|
| + mrf_int32->Set(i, Func(i, -1));
|
| + mrf_double->Set(i, Func(i, -2));
|
| + mrpf_string->Mutable(i)->assign(StrFunc(i, -5));
|
| + mrpf_foreign_message->Mutable(i)->set_c(Func(i, -6));
|
| + EXPECT_EQ(message.repeated_int32(i), Func(i, -1));
|
| + EXPECT_EQ(message.repeated_double(i), Func(i, -2));
|
| + EXPECT_EQ(message.repeated_string(i), StrFunc(i, -5));
|
| + EXPECT_EQ(message.repeated_foreign_message(i).c(), Func(i, -6));
|
| + down_cast<ForeignMessage*>(mrpf_message->Mutable(i))->set_c(Func(i, 7));
|
| + EXPECT_EQ(message.repeated_foreign_message(i).c(), Func(i, 7));
|
| + }
|
| +
|
| + // Make sure types are checked correctly at runtime.
|
| + const FieldDescriptor* fd_optional_int32 =
|
| + desc->FindFieldByName("optional_int32");
|
| + EXPECT_DEATH(refl->GetRepeatedField<int32>(
|
| + message, fd_optional_int32), "requires a repeated field");
|
| + EXPECT_DEATH(refl->GetRepeatedField<double>(
|
| + message, fd_repeated_int32), "not the right type");
|
| + EXPECT_DEATH(refl->GetRepeatedPtrField<TestAllTypes>(
|
| + message, fd_repeated_foreign_message), "wrong submessage type");
|
| +}
|
| +
|
| +
|
| +
|
| +
|
| +TEST(RepeatedFieldReflectionTest, ExtensionFields) {
|
| + TestAllExtensions extended_message;
|
| + const Reflection* refl = extended_message.GetReflection();
|
| + const Descriptor* desc = extended_message.GetDescriptor();
|
| +
|
| + for (int i = 0; i < 10; ++i) {
|
| + extended_message.AddExtension(
|
| + unittest::repeated_int64_extension, Func(i, 1));
|
| + }
|
| +
|
| + const FieldDescriptor* fd_repeated_int64_extension =
|
| + desc->file()->FindExtensionByName("repeated_int64_extension");
|
| + GOOGLE_CHECK(fd_repeated_int64_extension != NULL);
|
| +
|
| + const RepeatedField<int64>& rf_int64_extension =
|
| + refl->GetRepeatedField<int64>(extended_message,
|
| + fd_repeated_int64_extension);
|
| +
|
| + RepeatedField<int64>* mrf_int64_extension =
|
| + refl->MutableRepeatedField<int64>(&extended_message,
|
| + fd_repeated_int64_extension);
|
| +
|
| + for (int i = 0; i < 10; ++i) {
|
| + EXPECT_EQ(Func(i, 1), rf_int64_extension.Get(i));
|
| + mrf_int64_extension->Set(i, Func(i, -1));
|
| + EXPECT_EQ(Func(i, -1),
|
| + extended_message.GetExtension(unittest::repeated_int64_extension, i));
|
| + }
|
| +}
|
| +
|
| +} // namespace
|
| +} // namespace protobuf
|
| +} // namespace google
|
|
|
| Property changes on: third_party/protobuf/src/google/protobuf/repeated_field_reflection_unittest.cc
|
| ___________________________________________________________________
|
| Added: svn:eol-style
|
| + LF
|
|
|
|
|