| Index: third_party/protobuf/src/google/protobuf/proto3_arena_unittest.cc
|
| diff --git a/third_party/protobuf/src/google/protobuf/proto3_arena_unittest.cc b/third_party/protobuf/src/google/protobuf/proto3_arena_unittest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..2838e0fc644078c680032375cab424248f688e41
|
| --- /dev/null
|
| +++ b/third_party/protobuf/src/google/protobuf/proto3_arena_unittest.cc
|
| @@ -0,0 +1,209 @@
|
| +// Protocol Buffers - Google's data interchange format
|
| +// Copyright 2008 Google Inc. All rights reserved.
|
| +// https://developers.google.com/protocol-buffers/
|
| +//
|
| +// 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.
|
| +
|
| +#include <string>
|
| +#include <memory>
|
| +#ifndef _SHARED_PTR_H
|
| +#include <google/protobuf/stubs/shared_ptr.h>
|
| +#endif
|
| +#include <vector>
|
| +
|
| +#include <google/protobuf/test_util.h>
|
| +#include <google/protobuf/unittest.pb.h>
|
| +#include <google/protobuf/unittest_proto3_arena.pb.h>
|
| +#include <google/protobuf/arena.h>
|
| +#include <google/protobuf/testing/googletest.h>
|
| +#include <gtest/gtest.h>
|
| +
|
| +namespace google {
|
| +using proto3_arena_unittest::TestAllTypes;
|
| +
|
| +namespace protobuf {
|
| +namespace {
|
| +// We selectively set/check a few representative fields rather than all fields
|
| +// as this test is only expected to cover the basics of arena support.
|
| +void SetAllFields(TestAllTypes* m) {
|
| + m->set_optional_int32(100);
|
| + m->set_optional_string("asdf");
|
| + m->set_optional_bytes("jkl;");
|
| + m->mutable_optional_nested_message()->set_bb(42);
|
| + m->mutable_optional_foreign_message()->set_c(43);
|
| + m->set_optional_nested_enum(
|
| + proto3_arena_unittest::TestAllTypes_NestedEnum_BAZ);
|
| + m->set_optional_foreign_enum(
|
| + proto3_arena_unittest::FOREIGN_BAZ);
|
| + m->mutable_optional_lazy_message()->set_bb(45);
|
| + m->add_repeated_int32(100);
|
| + m->add_repeated_string("asdf");
|
| + m->add_repeated_bytes("jkl;");
|
| + m->add_repeated_nested_message()->set_bb(46);
|
| + m->add_repeated_foreign_message()->set_c(47);
|
| + m->add_repeated_nested_enum(
|
| + proto3_arena_unittest::TestAllTypes_NestedEnum_BAZ);
|
| + m->add_repeated_foreign_enum(
|
| + proto3_arena_unittest::FOREIGN_BAZ);
|
| + m->add_repeated_lazy_message()->set_bb(49);
|
| +
|
| + m->set_oneof_uint32(1);
|
| + m->mutable_oneof_nested_message()->set_bb(50);
|
| + m->set_oneof_string("test"); // only this one remains set
|
| +}
|
| +
|
| +void ExpectAllFieldsSet(const TestAllTypes& m) {
|
| + EXPECT_EQ(100, m.optional_int32());
|
| + EXPECT_EQ("asdf", m.optional_string());
|
| + EXPECT_EQ("jkl;", m.optional_bytes());
|
| + EXPECT_EQ(true, m.has_optional_nested_message());
|
| + EXPECT_EQ(42, m.optional_nested_message().bb());
|
| + EXPECT_EQ(true, m.has_optional_foreign_message());
|
| + EXPECT_EQ(43, m.optional_foreign_message().c());
|
| + EXPECT_EQ(proto3_arena_unittest::TestAllTypes_NestedEnum_BAZ,
|
| + m.optional_nested_enum());
|
| + EXPECT_EQ(proto3_arena_unittest::FOREIGN_BAZ,
|
| + m.optional_foreign_enum());
|
| + EXPECT_EQ(true, m.has_optional_lazy_message());
|
| + EXPECT_EQ(45, m.optional_lazy_message().bb());
|
| +
|
| + EXPECT_EQ(1, m.repeated_int32_size());
|
| + EXPECT_EQ(100, m.repeated_int32(0));
|
| + EXPECT_EQ(1, m.repeated_string_size());
|
| + EXPECT_EQ("asdf", m.repeated_string(0));
|
| + EXPECT_EQ(1, m.repeated_bytes_size());
|
| + EXPECT_EQ("jkl;", m.repeated_bytes(0));
|
| + EXPECT_EQ(1, m.repeated_nested_message_size());
|
| + EXPECT_EQ(46, m.repeated_nested_message(0).bb());
|
| + EXPECT_EQ(1, m.repeated_foreign_message_size());
|
| + EXPECT_EQ(47, m.repeated_foreign_message(0).c());
|
| + EXPECT_EQ(1, m.repeated_nested_enum_size());
|
| + EXPECT_EQ(proto3_arena_unittest::TestAllTypes_NestedEnum_BAZ,
|
| + m.repeated_nested_enum(0));
|
| + EXPECT_EQ(1, m.repeated_foreign_enum_size());
|
| + EXPECT_EQ(proto3_arena_unittest::FOREIGN_BAZ,
|
| + m.repeated_foreign_enum(0));
|
| + EXPECT_EQ(1, m.repeated_lazy_message_size());
|
| + EXPECT_EQ(49, m.repeated_lazy_message(0).bb());
|
| +
|
| + EXPECT_EQ(proto3_arena_unittest::TestAllTypes::kOneofString,
|
| + m.oneof_field_case());
|
| + EXPECT_EQ("test", m.oneof_string());
|
| +}
|
| +
|
| +// In this file we only test some basic functionalities of arena support in
|
| +// proto3 and expect the arena support to be fully tested in proto2 unittests
|
| +// because proto3 shares most code with proto2.
|
| +
|
| +TEST(Proto3ArenaTest, Parsing) {
|
| + TestAllTypes original;
|
| + SetAllFields(&original);
|
| +
|
| + Arena arena;
|
| + TestAllTypes* arena_message = Arena::CreateMessage<TestAllTypes>(&arena);
|
| + arena_message->ParseFromString(original.SerializeAsString());
|
| + ExpectAllFieldsSet(*arena_message);
|
| +}
|
| +
|
| +TEST(Proto3ArenaTest, UnknownFields) {
|
| + TestAllTypes original;
|
| + SetAllFields(&original);
|
| +
|
| + Arena arena;
|
| + TestAllTypes* arena_message = Arena::CreateMessage<TestAllTypes>(&arena);
|
| + arena_message->ParseFromString(original.SerializeAsString());
|
| + ExpectAllFieldsSet(*arena_message);
|
| +
|
| + // In proto3 we can still get a pointer to the UnknownFieldSet through
|
| + // reflection API.
|
| + UnknownFieldSet* unknown_fields =
|
| + arena_message->GetReflection()->MutableUnknownFields(arena_message);
|
| + // We can modify this UnknownFieldSet.
|
| + unknown_fields->AddVarint(1, 2);
|
| + // But the change will never will serialized back.
|
| + ASSERT_EQ(original.ByteSize(), arena_message->ByteSize());
|
| + ASSERT_TRUE(
|
| + arena_message->GetReflection()->GetUnknownFields(*arena_message).empty());
|
| +}
|
| +
|
| +TEST(Proto3ArenaTest, Swap) {
|
| + Arena arena1;
|
| + Arena arena2;
|
| +
|
| + // Test Swap().
|
| + TestAllTypes* arena1_message = Arena::CreateMessage<TestAllTypes>(&arena1);
|
| + TestAllTypes* arena2_message = Arena::CreateMessage<TestAllTypes>(&arena2);
|
| + arena1_message->Swap(arena2_message);
|
| + EXPECT_EQ(&arena1, arena1_message->GetArena());
|
| + EXPECT_EQ(&arena2, arena2_message->GetArena());
|
| +}
|
| +
|
| +TEST(Proto3ArenaTest, SetAllocatedMessage) {
|
| + Arena arena;
|
| + TestAllTypes *arena_message = Arena::CreateMessage<TestAllTypes>(&arena);
|
| + TestAllTypes::NestedMessage* nested = new TestAllTypes::NestedMessage;
|
| + nested->set_bb(118);
|
| + arena_message->set_allocated_optional_nested_message(nested);
|
| + EXPECT_EQ(118, arena_message->optional_nested_message().bb());
|
| +}
|
| +
|
| +TEST(Proto3ArenaTest, ReleaseMessage) {
|
| + Arena arena;
|
| + TestAllTypes* arena_message = Arena::CreateMessage<TestAllTypes>(&arena);
|
| + arena_message->mutable_optional_nested_message()->set_bb(118);
|
| + google::protobuf::scoped_ptr<TestAllTypes::NestedMessage> nested(
|
| + arena_message->release_optional_nested_message());
|
| + EXPECT_EQ(118, nested->bb());
|
| +}
|
| +
|
| +TEST(Proto3ArenaTest, MessageFieldClear) {
|
| + // GitHub issue #310: https://github.com/google/protobuf/issues/310
|
| + Arena arena;
|
| + TestAllTypes* arena_message = Arena::CreateMessage<TestAllTypes>(&arena);
|
| + arena_message->mutable_optional_nested_message()->set_bb(118);
|
| + // This should not crash, but prior to the bugfix, it tried to use `operator
|
| + // delete` the nested message (which is on the arena):
|
| + arena_message->Clear();
|
| +}
|
| +
|
| +TEST(Proto3ArenaTest, MessageFieldClearViaReflection) {
|
| + Arena arena;
|
| + TestAllTypes* message = Arena::CreateMessage<TestAllTypes>(&arena);
|
| + const Reflection* r = message->GetReflection();
|
| + const Descriptor* d = message->GetDescriptor();
|
| + const FieldDescriptor* msg_field = d->FindFieldByName(
|
| + "optional_nested_message");
|
| +
|
| + message->mutable_optional_nested_message()->set_bb(1);
|
| + r->ClearField(message, msg_field);
|
| + EXPECT_FALSE(message->has_optional_nested_message());
|
| + EXPECT_EQ(0, message->optional_nested_message().bb());
|
| +}
|
| +
|
| +} // namespace
|
| +} // namespace protobuf
|
| +} // namespace google
|
|
|