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 <string>
32 
33 #include <google/protobuf/unittest.pb.h>
34 #include <google/protobuf/unittest_no_field_presence.pb.h>
35 #include <google/protobuf/descriptor.pb.h>
36 #include <google/protobuf/descriptor.h>
37 #include <gtest/gtest.h>
38 
39 namespace google {
40 namespace protobuf {
41 namespace {
42 
43 // Helper: checks that all fields have default (zero/empty) values.
CheckDefaultValues(const proto2_nofieldpresence_unittest::TestAllTypes & m)44 void CheckDefaultValues(
45     const proto2_nofieldpresence_unittest::TestAllTypes& m) {
46   EXPECT_EQ(0, m.optional_int32());
47   EXPECT_EQ(0, m.optional_int64());
48   EXPECT_EQ(0, m.optional_uint32());
49   EXPECT_EQ(0, m.optional_uint64());
50   EXPECT_EQ(0, m.optional_sint32());
51   EXPECT_EQ(0, m.optional_sint64());
52   EXPECT_EQ(0, m.optional_fixed32());
53   EXPECT_EQ(0, m.optional_fixed64());
54   EXPECT_EQ(0, m.optional_sfixed32());
55   EXPECT_EQ(0, m.optional_sfixed64());
56   EXPECT_EQ(0, m.optional_float());
57   EXPECT_EQ(0, m.optional_double());
58   EXPECT_EQ(false, m.optional_bool());
59   EXPECT_EQ(0, m.optional_string().size());
60   EXPECT_EQ(0, m.optional_bytes().size());
61 
62   EXPECT_EQ(false, m.has_optional_nested_message());
63   // accessor for message fields returns default instance when not present
64   EXPECT_EQ(0, m.optional_nested_message().bb());
65   EXPECT_EQ(false, m.has_optional_proto2_message());
66   // Embedded proto2 messages still have proto2 semantics, e.g. non-zero default
67   // values. Here the submessage is not present but its accessor returns the
68   // default instance.
69   EXPECT_EQ(41, m.optional_proto2_message().default_int32());
70   EXPECT_EQ(false, m.has_optional_foreign_message());
71   EXPECT_EQ(proto2_nofieldpresence_unittest::TestAllTypes::FOO,
72             m.optional_nested_enum());
73   EXPECT_EQ(proto2_nofieldpresence_unittest::FOREIGN_FOO,
74             m.optional_foreign_enum());
75 
76 
77   EXPECT_EQ(0, m.repeated_int32_size());
78   EXPECT_EQ(0, m.repeated_int64_size());
79   EXPECT_EQ(0, m.repeated_uint32_size());
80   EXPECT_EQ(0, m.repeated_uint64_size());
81   EXPECT_EQ(0, m.repeated_sint32_size());
82   EXPECT_EQ(0, m.repeated_sint64_size());
83   EXPECT_EQ(0, m.repeated_fixed32_size());
84   EXPECT_EQ(0, m.repeated_fixed64_size());
85   EXPECT_EQ(0, m.repeated_sfixed32_size());
86   EXPECT_EQ(0, m.repeated_sfixed64_size());
87   EXPECT_EQ(0, m.repeated_float_size());
88   EXPECT_EQ(0, m.repeated_double_size());
89   EXPECT_EQ(0, m.repeated_bool_size());
90   EXPECT_EQ(0, m.repeated_string_size());
91   EXPECT_EQ(0, m.repeated_bytes_size());
92   EXPECT_EQ(0, m.repeated_nested_message_size());
93   EXPECT_EQ(0, m.repeated_foreign_message_size());
94   EXPECT_EQ(0, m.repeated_proto2_message_size());
95   EXPECT_EQ(0, m.repeated_nested_enum_size());
96   EXPECT_EQ(0, m.repeated_foreign_enum_size());
97   EXPECT_EQ(0, m.repeated_lazy_message_size());
98   EXPECT_EQ(proto2_nofieldpresence_unittest::TestAllTypes::ONEOF_FIELD_NOT_SET,
99             m.oneof_field_case());
100 }
101 
FillValues(proto2_nofieldpresence_unittest::TestAllTypes * m)102 void FillValues(proto2_nofieldpresence_unittest::TestAllTypes* m) {
103   m->set_optional_int32(100);
104   m->set_optional_int64(101);
105   m->set_optional_uint32(102);
106   m->set_optional_uint64(103);
107   m->set_optional_sint32(104);
108   m->set_optional_sint64(105);
109   m->set_optional_fixed32(106);
110   m->set_optional_fixed64(107);
111   m->set_optional_sfixed32(108);
112   m->set_optional_sfixed64(109);
113   m->set_optional_float(110.0);
114   m->set_optional_double(111.0);
115   m->set_optional_bool(true);
116   m->set_optional_string("asdf");
117   m->set_optional_bytes("jkl;");
118   m->mutable_optional_nested_message()->set_bb(42);
119   m->mutable_optional_foreign_message()->set_c(43);
120   m->mutable_optional_proto2_message()->set_optional_int32(44);
121   m->set_optional_nested_enum(
122       proto2_nofieldpresence_unittest::TestAllTypes::BAZ);
123   m->set_optional_foreign_enum(proto2_nofieldpresence_unittest::FOREIGN_BAZ);
124   m->mutable_optional_lazy_message()->set_bb(45);
125   m->add_repeated_int32(100);
126   m->add_repeated_int64(101);
127   m->add_repeated_uint32(102);
128   m->add_repeated_uint64(103);
129   m->add_repeated_sint32(104);
130   m->add_repeated_sint64(105);
131   m->add_repeated_fixed32(106);
132   m->add_repeated_fixed64(107);
133   m->add_repeated_sfixed32(108);
134   m->add_repeated_sfixed64(109);
135   m->add_repeated_float(110.0);
136   m->add_repeated_double(111.0);
137   m->add_repeated_bool(true);
138   m->add_repeated_string("asdf");
139   m->add_repeated_bytes("jkl;");
140   m->add_repeated_nested_message()->set_bb(46);
141   m->add_repeated_foreign_message()->set_c(47);
142   m->add_repeated_proto2_message()->set_optional_int32(48);
143   m->add_repeated_nested_enum(
144       proto2_nofieldpresence_unittest::TestAllTypes::BAZ);
145   m->add_repeated_foreign_enum(proto2_nofieldpresence_unittest::FOREIGN_BAZ);
146   m->add_repeated_lazy_message()->set_bb(49);
147 
148   m->set_oneof_uint32(1);
149   m->mutable_oneof_nested_message()->set_bb(50);
150   m->set_oneof_string("test");  // only this one remains set
151 }
152 
CheckNonDefaultValues(const proto2_nofieldpresence_unittest::TestAllTypes & m)153 void CheckNonDefaultValues(
154     const proto2_nofieldpresence_unittest::TestAllTypes& m) {
155   EXPECT_EQ(100, m.optional_int32());
156   EXPECT_EQ(101, m.optional_int64());
157   EXPECT_EQ(102, m.optional_uint32());
158   EXPECT_EQ(103, m.optional_uint64());
159   EXPECT_EQ(104, m.optional_sint32());
160   EXPECT_EQ(105, m.optional_sint64());
161   EXPECT_EQ(106, m.optional_fixed32());
162   EXPECT_EQ(107, m.optional_fixed64());
163   EXPECT_EQ(108, m.optional_sfixed32());
164   EXPECT_EQ(109, m.optional_sfixed64());
165   EXPECT_EQ(110.0, m.optional_float());
166   EXPECT_EQ(111.0, m.optional_double());
167   EXPECT_EQ(true, m.optional_bool());
168   EXPECT_EQ("asdf", m.optional_string());
169   EXPECT_EQ("jkl;", m.optional_bytes());
170   EXPECT_EQ(true, m.has_optional_nested_message());
171   EXPECT_EQ(42, m.optional_nested_message().bb());
172   EXPECT_EQ(true, m.has_optional_foreign_message());
173   EXPECT_EQ(43, m.optional_foreign_message().c());
174   EXPECT_EQ(true, m.has_optional_proto2_message());
175   EXPECT_EQ(44, m.optional_proto2_message().optional_int32());
176   EXPECT_EQ(proto2_nofieldpresence_unittest::TestAllTypes::BAZ,
177             m.optional_nested_enum());
178   EXPECT_EQ(proto2_nofieldpresence_unittest::FOREIGN_BAZ,
179             m.optional_foreign_enum());
180   EXPECT_EQ(true, m.has_optional_lazy_message());
181   EXPECT_EQ(45, m.optional_lazy_message().bb());
182 
183   EXPECT_EQ(1, m.repeated_int32_size());
184   EXPECT_EQ(100, m.repeated_int32(0));
185   EXPECT_EQ(1, m.repeated_int64_size());
186   EXPECT_EQ(101, m.repeated_int64(0));
187   EXPECT_EQ(1, m.repeated_uint32_size());
188   EXPECT_EQ(102, m.repeated_uint32(0));
189   EXPECT_EQ(1, m.repeated_uint64_size());
190   EXPECT_EQ(103, m.repeated_uint64(0));
191   EXPECT_EQ(1, m.repeated_sint32_size());
192   EXPECT_EQ(104, m.repeated_sint32(0));
193   EXPECT_EQ(1, m.repeated_sint64_size());
194   EXPECT_EQ(105, m.repeated_sint64(0));
195   EXPECT_EQ(1, m.repeated_fixed32_size());
196   EXPECT_EQ(106, m.repeated_fixed32(0));
197   EXPECT_EQ(1, m.repeated_fixed64_size());
198   EXPECT_EQ(107, m.repeated_fixed64(0));
199   EXPECT_EQ(1, m.repeated_sfixed32_size());
200   EXPECT_EQ(108, m.repeated_sfixed32(0));
201   EXPECT_EQ(1, m.repeated_sfixed64_size());
202   EXPECT_EQ(109, m.repeated_sfixed64(0));
203   EXPECT_EQ(1, m.repeated_float_size());
204   EXPECT_EQ(110.0, m.repeated_float(0));
205   EXPECT_EQ(1, m.repeated_double_size());
206   EXPECT_EQ(111.0, m.repeated_double(0));
207   EXPECT_EQ(1, m.repeated_bool_size());
208   EXPECT_EQ(true, m.repeated_bool(0));
209   EXPECT_EQ(1, m.repeated_string_size());
210   EXPECT_EQ("asdf", m.repeated_string(0));
211   EXPECT_EQ(1, m.repeated_bytes_size());
212   EXPECT_EQ("jkl;", m.repeated_bytes(0));
213   EXPECT_EQ(1, m.repeated_nested_message_size());
214   EXPECT_EQ(46, m.repeated_nested_message(0).bb());
215   EXPECT_EQ(1, m.repeated_foreign_message_size());
216   EXPECT_EQ(47, m.repeated_foreign_message(0).c());
217   EXPECT_EQ(1, m.repeated_proto2_message_size());
218   EXPECT_EQ(48, m.repeated_proto2_message(0).optional_int32());
219   EXPECT_EQ(1, m.repeated_nested_enum_size());
220   EXPECT_EQ(proto2_nofieldpresence_unittest::TestAllTypes::BAZ,
221             m.repeated_nested_enum(0));
222   EXPECT_EQ(1, m.repeated_foreign_enum_size());
223   EXPECT_EQ(proto2_nofieldpresence_unittest::FOREIGN_BAZ,
224             m.repeated_foreign_enum(0));
225   EXPECT_EQ(1, m.repeated_lazy_message_size());
226   EXPECT_EQ(49, m.repeated_lazy_message(0).bb());
227 
228   EXPECT_EQ(proto2_nofieldpresence_unittest::TestAllTypes::kOneofString,
229             m.oneof_field_case());
230   EXPECT_EQ("test", m.oneof_string());
231 }
232 
TEST(NoFieldPresenceTest,BasicMessageTest)233 TEST(NoFieldPresenceTest, BasicMessageTest) {
234   proto2_nofieldpresence_unittest::TestAllTypes message;
235   // Check default values, fill all fields, check values. We just want to
236   // exercise the basic getters/setter paths here to make sure no
237   // field-presence-related changes broke these.
238   CheckDefaultValues(message);
239   FillValues(&message);
240   CheckNonDefaultValues(message);
241 
242   // Clear() should be equivalent to getting a freshly-constructed message.
243   message.Clear();
244   CheckDefaultValues(message);
245 }
246 
TEST(NoFieldPresenceTest,MessageFieldPresenceTest)247 TEST(NoFieldPresenceTest, MessageFieldPresenceTest) {
248   // check that presence still works properly for message fields.
249   proto2_nofieldpresence_unittest::TestAllTypes message;
250   EXPECT_EQ(false, message.has_optional_nested_message());
251   // Getter should fetch default instance, and not cause the field to become
252   // present.
253   EXPECT_EQ(0, message.optional_nested_message().bb());
254   EXPECT_EQ(false, message.has_optional_nested_message());
255   message.mutable_optional_nested_message()->set_bb(42);
256   EXPECT_EQ(true, message.has_optional_nested_message());
257   message.clear_optional_nested_message();
258   EXPECT_EQ(false, message.has_optional_nested_message());
259 
260   // Likewise for a lazy message field.
261   EXPECT_EQ(false, message.has_optional_lazy_message());
262   // Getter should fetch default instance, and not cause the field to become
263   // present.
264   EXPECT_EQ(0, message.optional_lazy_message().bb());
265   EXPECT_EQ(false, message.has_optional_lazy_message());
266   message.mutable_optional_lazy_message()->set_bb(42);
267   EXPECT_EQ(true, message.has_optional_lazy_message());
268   message.clear_optional_lazy_message();
269   EXPECT_EQ(false, message.has_optional_lazy_message());
270 
271   // Test field presence of a message field on the default instance.
272   EXPECT_EQ(false,
273             proto2_nofieldpresence_unittest::TestAllTypes::default_instance()
274                 .has_optional_nested_message());
275 }
276 
TEST(NoFieldPresenceTest,ReflectionHasFieldTest)277 TEST(NoFieldPresenceTest, ReflectionHasFieldTest) {
278   // check that HasField reports true on all scalar fields. Check that it
279   // behaves properly for message fields.
280 
281   proto2_nofieldpresence_unittest::TestAllTypes message;
282   const Reflection* r = message.GetReflection();
283   const Descriptor* desc = message.GetDescriptor();
284 
285   // Check initial state: scalars not present (due to need to be consistent with
286   // MergeFrom()), message fields not present, oneofs not present.
287   for (int i = 0; i < desc->field_count(); i++) {
288     const FieldDescriptor* field = desc->field(i);
289     if (field->is_repeated()) continue;
290     EXPECT_EQ(false, r->HasField(message, field));
291   }
292 
293   // Test field presence of a message field on the default instance.
294   const FieldDescriptor* msg_field =
295       desc->FindFieldByName("optional_nested_message");
296   EXPECT_EQ(
297       false,
298       r->HasField(
299           proto2_nofieldpresence_unittest::TestAllTypes::default_instance(),
300           msg_field));
301 
302   // Fill all fields, expect everything to report true (check oneofs below).
303   FillValues(&message);
304   for (int i = 0; i < desc->field_count(); i++) {
305     const FieldDescriptor* field = desc->field(i);
306     if (field->is_repeated() || field->containing_oneof()) {
307       continue;
308     }
309     if (field->options().ctype() != FieldOptions::STRING) {
310       continue;
311     }
312     EXPECT_EQ(true, r->HasField(message, field));
313   }
314 
315   message.Clear();
316 
317   // Check zero/empty-means-not-present semantics.
318   const FieldDescriptor* field_int32 = desc->FindFieldByName("optional_int32");
319   const FieldDescriptor* field_double =
320       desc->FindFieldByName("optional_double");
321   const FieldDescriptor* field_string =
322       desc->FindFieldByName("optional_string");
323 
324   EXPECT_EQ(false, r->HasField(message, field_int32));
325   EXPECT_EQ(false, r->HasField(message, field_double));
326   EXPECT_EQ(false, r->HasField(message, field_string));
327 
328   message.set_optional_int32(42);
329   EXPECT_EQ(true, r->HasField(message, field_int32));
330   message.set_optional_int32(0);
331   EXPECT_EQ(false, r->HasField(message, field_int32));
332 
333   message.set_optional_double(42.0);
334   EXPECT_EQ(true, r->HasField(message, field_double));
335   message.set_optional_double(0.0);
336   EXPECT_EQ(false, r->HasField(message, field_double));
337 
338   message.set_optional_string("test");
339   EXPECT_EQ(true, r->HasField(message, field_string));
340   message.set_optional_string("");
341   EXPECT_EQ(false, r->HasField(message, field_string));
342 }
343 
TEST(NoFieldPresenceTest,ReflectionClearFieldTest)344 TEST(NoFieldPresenceTest, ReflectionClearFieldTest) {
345   proto2_nofieldpresence_unittest::TestAllTypes message;
346 
347   const Reflection* r = message.GetReflection();
348   const Descriptor* desc = message.GetDescriptor();
349 
350   const FieldDescriptor* field_int32 = desc->FindFieldByName("optional_int32");
351   const FieldDescriptor* field_double =
352       desc->FindFieldByName("optional_double");
353   const FieldDescriptor* field_string =
354       desc->FindFieldByName("optional_string");
355   const FieldDescriptor* field_message =
356       desc->FindFieldByName("optional_nested_message");
357   const FieldDescriptor* field_lazy =
358       desc->FindFieldByName("optional_lazy_message");
359 
360   message.set_optional_int32(42);
361   r->ClearField(&message, field_int32);
362   EXPECT_EQ(0, message.optional_int32());
363 
364   message.set_optional_double(42.0);
365   r->ClearField(&message, field_double);
366   EXPECT_EQ(0.0, message.optional_double());
367 
368   message.set_optional_string("test");
369   r->ClearField(&message, field_string);
370   EXPECT_EQ("", message.optional_string());
371 
372   message.mutable_optional_nested_message()->set_bb(1234);
373   r->ClearField(&message, field_message);
374   EXPECT_FALSE(message.has_optional_nested_message());
375   EXPECT_EQ(0, message.optional_nested_message().bb());
376 
377   message.mutable_optional_lazy_message()->set_bb(42);
378   r->ClearField(&message, field_lazy);
379   EXPECT_FALSE(message.has_optional_lazy_message());
380   EXPECT_EQ(0, message.optional_lazy_message().bb());
381 }
382 
TEST(NoFieldPresenceTest,HasFieldOneofsTest)383 TEST(NoFieldPresenceTest, HasFieldOneofsTest) {
384   // check that HasField behaves properly for oneofs.
385   proto2_nofieldpresence_unittest::TestAllTypes message;
386 
387   const Reflection* r = message.GetReflection();
388   const Descriptor* desc = message.GetDescriptor();
389   const FieldDescriptor* desc_oneof_uint32 =
390       desc->FindFieldByName("oneof_uint32");
391   const FieldDescriptor* desc_oneof_nested_message =
392       desc->FindFieldByName("oneof_nested_message");
393   const FieldDescriptor* desc_oneof_string =
394       desc->FindFieldByName("oneof_string");
395   GOOGLE_CHECK(desc_oneof_uint32 != nullptr);
396   GOOGLE_CHECK(desc_oneof_nested_message != nullptr);
397   GOOGLE_CHECK(desc_oneof_string != nullptr);
398 
399   EXPECT_EQ(false, r->HasField(message, desc_oneof_uint32));
400   EXPECT_EQ(false, r->HasField(message, desc_oneof_nested_message));
401   EXPECT_EQ(false, r->HasField(message, desc_oneof_string));
402 
403   message.set_oneof_string("test");
404   EXPECT_EQ(false, r->HasField(message, desc_oneof_uint32));
405   EXPECT_EQ(false, r->HasField(message, desc_oneof_nested_message));
406   EXPECT_EQ(true, r->HasField(message, desc_oneof_string));
407   message.mutable_oneof_nested_message()->set_bb(42);
408   EXPECT_EQ(false, r->HasField(message, desc_oneof_uint32));
409   EXPECT_EQ(true, r->HasField(message, desc_oneof_nested_message));
410   EXPECT_EQ(false, r->HasField(message, desc_oneof_string));
411 
412   message.Clear();
413   EXPECT_EQ(false, r->HasField(message, desc_oneof_uint32));
414   EXPECT_EQ(false, r->HasField(message, desc_oneof_nested_message));
415   EXPECT_EQ(false, r->HasField(message, desc_oneof_string));
416 }
417 
TEST(NoFieldPresenceTest,DontSerializeDefaultValuesTest)418 TEST(NoFieldPresenceTest, DontSerializeDefaultValuesTest) {
419   // check that serialized data contains only non-zero numeric fields/non-empty
420   // string/byte fields.
421   proto2_nofieldpresence_unittest::TestAllTypes message;
422   std::string output;
423 
424   // All default values -> no output.
425   message.SerializeToString(&output);
426   EXPECT_EQ(0, output.size());
427 
428   // Zero values -> still no output.
429   message.set_optional_int32(0);
430   message.set_optional_int64(0);
431   message.set_optional_uint32(0);
432   message.set_optional_uint64(0);
433   message.set_optional_sint32(0);
434   message.set_optional_sint64(0);
435   message.set_optional_fixed32(0);
436   message.set_optional_fixed64(0);
437   message.set_optional_sfixed32(0);
438   message.set_optional_sfixed64(0);
439   message.set_optional_float(0);
440   message.set_optional_double(0);
441   message.set_optional_bool(0);
442   message.set_optional_string("");
443   message.set_optional_bytes("");
444   message.set_optional_nested_enum(
445       proto2_nofieldpresence_unittest::TestAllTypes::FOO);  // first enum entry
446   message.set_optional_foreign_enum(
447       proto2_nofieldpresence_unittest::FOREIGN_FOO);  // first enum entry
448 
449   message.SerializeToString(&output);
450   EXPECT_EQ(0, output.size());
451 
452   message.set_optional_int32(1);
453   message.SerializeToString(&output);
454   EXPECT_EQ(2, output.size());
455   EXPECT_EQ("\x08\x01", output);
456 
457   message.set_optional_int32(0);
458   message.SerializeToString(&output);
459   EXPECT_EQ(0, output.size());
460 }
461 
TEST(NoFieldPresenceTest,MergeFromIfNonzeroTest)462 TEST(NoFieldPresenceTest, MergeFromIfNonzeroTest) {
463   // check that MergeFrom copies if nonzero/nondefault only.
464   proto2_nofieldpresence_unittest::TestAllTypes source;
465   proto2_nofieldpresence_unittest::TestAllTypes dest;
466 
467   dest.set_optional_int32(42);
468   dest.set_optional_string("test");
469   source.set_optional_int32(0);
470   source.set_optional_string("");
471   // MergeFrom() copies only if present in serialization, i.e., non-zero.
472   dest.MergeFrom(source);
473   EXPECT_EQ(42, dest.optional_int32());
474   EXPECT_EQ("test", dest.optional_string());
475 
476   source.set_optional_int32(84);
477   source.set_optional_string("test2");
478   dest.MergeFrom(source);
479   EXPECT_EQ(84, dest.optional_int32());
480   EXPECT_EQ("test2", dest.optional_string());
481 }
482 
TEST(NoFieldPresenceTest,IsInitializedTest)483 TEST(NoFieldPresenceTest, IsInitializedTest) {
484   // Check that IsInitialized works properly.
485   proto2_nofieldpresence_unittest::TestProto2Required message;
486 
487   EXPECT_EQ(true, message.IsInitialized());
488   message.mutable_proto2()->set_a(1);
489   EXPECT_EQ(false, message.IsInitialized());
490   message.mutable_proto2()->set_b(1);
491   EXPECT_EQ(false, message.IsInitialized());
492   message.mutable_proto2()->set_c(1);
493   EXPECT_EQ(true, message.IsInitialized());
494 }
495 
TEST(NoFieldPresenceTest,LazyMessageFieldHasBit)496 TEST(NoFieldPresenceTest, LazyMessageFieldHasBit) {
497   // Check that has-bit interaction with lazy message works (has-bit before and
498   // after lazy decode).
499   proto2_nofieldpresence_unittest::TestAllTypes message;
500   const Reflection* r = message.GetReflection();
501   const Descriptor* desc = message.GetDescriptor();
502   const FieldDescriptor* field = desc->FindFieldByName("optional_lazy_message");
503   GOOGLE_CHECK(field != nullptr);
504 
505   EXPECT_EQ(false, message.has_optional_lazy_message());
506   EXPECT_EQ(false, r->HasField(message, field));
507 
508   message.mutable_optional_lazy_message()->set_bb(42);
509   EXPECT_EQ(true, message.has_optional_lazy_message());
510   EXPECT_EQ(true, r->HasField(message, field));
511 
512   // Serialize and parse with a new message object so that lazy field on new
513   // object is in unparsed state.
514   std::string output;
515   message.SerializeToString(&output);
516   proto2_nofieldpresence_unittest::TestAllTypes message2;
517   message2.ParseFromString(output);
518 
519   EXPECT_EQ(true, message2.has_optional_lazy_message());
520   EXPECT_EQ(true, r->HasField(message2, field));
521 
522   // Access field to force lazy parse.
523   EXPECT_EQ(42, message.optional_lazy_message().bb());
524   EXPECT_EQ(true, message2.has_optional_lazy_message());
525   EXPECT_EQ(true, r->HasField(message2, field));
526 }
527 
TEST(NoFieldPresenceTest,OneofPresence)528 TEST(NoFieldPresenceTest, OneofPresence) {
529   proto2_nofieldpresence_unittest::TestAllTypes message;
530   // oneof fields still have field presence -- ensure that this goes on the wire
531   // even though its value is the empty string.
532   message.set_oneof_string("");
533   std::string serialized;
534   message.SerializeToString(&serialized);
535   // Tag: 113 --> tag is (113 << 3) | 2 (length delimited) = 906
536   // varint: 0x8a 0x07
537   // Length: 0x00
538   EXPECT_EQ(3, serialized.size());
539   EXPECT_EQ(static_cast<char>(0x8a), serialized.at(0));
540   EXPECT_EQ(static_cast<char>(0x07), serialized.at(1));
541   EXPECT_EQ(static_cast<char>(0x00), serialized.at(2));
542 
543   message.Clear();
544   EXPECT_TRUE(message.ParseFromString(serialized));
545   EXPECT_EQ(proto2_nofieldpresence_unittest::TestAllTypes::kOneofString,
546             message.oneof_field_case());
547 
548   // Also test int32 and enum fields.
549   message.Clear();
550   message.set_oneof_uint32(0);  // would not go on wire if ordinary field.
551   message.SerializeToString(&serialized);
552   EXPECT_EQ(3, serialized.size());
553   EXPECT_TRUE(message.ParseFromString(serialized));
554   EXPECT_EQ(proto2_nofieldpresence_unittest::TestAllTypes::kOneofUint32,
555             message.oneof_field_case());
556 
557   message.Clear();
558   message.set_oneof_enum(
559       proto2_nofieldpresence_unittest::TestAllTypes::FOO);  // default
560                                                             // value.
561   message.SerializeToString(&serialized);
562   EXPECT_EQ(3, serialized.size());
563   EXPECT_TRUE(message.ParseFromString(serialized));
564   EXPECT_EQ(proto2_nofieldpresence_unittest::TestAllTypes::kOneofEnum,
565             message.oneof_field_case());
566 
567   message.Clear();
568   message.set_oneof_string("test");
569   message.clear_oneof_string();
570   EXPECT_EQ(0, message.ByteSizeLong());
571 }
572 
573 }  // namespace
574 }  // namespace protobuf
575 }  // namespace google
576