1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2008 Google Inc. All rights reserved.
3 // https://developers.google.com/protocol-buffers/
4 //
5 // Redistribution and use in source and binary forms, with or without
6 // modification, are permitted provided that the following conditions are
7 // met:
8 //
9 // * Redistributions of source code must retain the above copyright
10 // notice, this list of conditions and the following disclaimer.
11 // * Redistributions in binary form must reproduce the above
12 // copyright notice, this list of conditions and the following disclaimer
13 // in the documentation and/or other materials provided with the
14 // distribution.
15 // * Neither the name of Google Inc. nor the names of its
16 // contributors may be used to endorse or promote products derived from
17 // this software without specific prior written permission.
18 //
19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30
31 #include <memory>
32 #ifndef _SHARED_PTR_H
33 #include <google/protobuf/stubs/shared_ptr.h>
34 #endif
35
36 #include <google/protobuf/unittest_drop_unknown_fields.pb.h>
37 #include <google/protobuf/dynamic_message.h>
38 #include <gtest/gtest.h>
39
40 namespace google {
41 using unittest_drop_unknown_fields::Foo;
42 using unittest_drop_unknown_fields::FooWithExtraFields;
43
44 namespace protobuf {
45
TEST(DropUnknownFieldsTest,GeneratedMessage)46 TEST(DropUnknownFieldsTest, GeneratedMessage) {
47 FooWithExtraFields foo_with_extra_fields;
48 foo_with_extra_fields.set_int32_value(1);
49 foo_with_extra_fields.set_enum_value(FooWithExtraFields::QUX);
50 foo_with_extra_fields.set_extra_int32_value(2);
51
52 Foo foo;
53 ASSERT_TRUE(foo.ParseFromString(foo_with_extra_fields.SerializeAsString()));
54 EXPECT_EQ(1, foo.int32_value());
55 EXPECT_EQ(static_cast<int>(FooWithExtraFields::QUX),
56 static_cast<int>(foo.enum_value()));
57 // We don't generate unknown field accessors but the UnknownFieldSet is
58 // still exposed through reflection API.
59 EXPECT_TRUE(foo.GetReflection()->GetUnknownFields(foo).empty());
60
61 ASSERT_TRUE(foo_with_extra_fields.ParseFromString(foo.SerializeAsString()));
62 EXPECT_EQ(1, foo_with_extra_fields.int32_value());
63 EXPECT_EQ(FooWithExtraFields::QUX, foo_with_extra_fields.enum_value());
64 // The "extra_int32_value" field should be lost.
65 EXPECT_EQ(0, foo_with_extra_fields.extra_int32_value());
66 }
67
TEST(DropUnknownFieldsTest,DynamicMessage)68 TEST(DropUnknownFieldsTest, DynamicMessage) {
69 FooWithExtraFields foo_with_extra_fields;
70 foo_with_extra_fields.set_int32_value(1);
71 foo_with_extra_fields.set_enum_value(FooWithExtraFields::QUX);
72 foo_with_extra_fields.set_extra_int32_value(2);
73
74 google::protobuf::DynamicMessageFactory factory;
75 google::protobuf::scoped_ptr<google::protobuf::Message> foo(
76 factory.GetPrototype(Foo::descriptor())->New());
77 ASSERT_TRUE(foo->ParseFromString(foo_with_extra_fields.SerializeAsString()));
78 EXPECT_TRUE(foo->GetReflection()->GetUnknownFields(*foo).empty());
79
80 ASSERT_TRUE(foo_with_extra_fields.ParseFromString(foo->SerializeAsString()));
81 EXPECT_EQ(1, foo_with_extra_fields.int32_value());
82 EXPECT_EQ(FooWithExtraFields::QUX, foo_with_extra_fields.enum_value());
83 // The "extra_int32_value" field should be lost.
84 EXPECT_EQ(0, foo_with_extra_fields.extra_int32_value());
85 }
86
87 } // namespace protobuf
88 } // namespace google
89