proto_zero_plugin [NOT FOR REVIEW].

components/tracing/proto_zero_test/mock_messages_unittest.cc

Index: components/tracing/proto_zero_test/mock_messages_unittest.cc
diff --git a/components/tracing/proto_zero_test/mock_messages_unittest.cc b/components/tracing/proto_zero_test/mock_messages_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..3f21e98d6bd180c1f04b462e97cafe59f5804d6a
--- /dev/null
+++ b/components/tracing/proto_zero_test/mock_messages_unittest.cc
@@ -0,0 +1,64 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "components/tracing/proto_zero_test/fake_runtime.h"
+#include "components/tracing/proto_zero_test/mock_messages.zeropb.h"
+
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace tracing {
+namespace proto {
+namespace {
+
+#define EXPECT_EVAL(expected, object, appender) \
+  object.appender; EXPECT_EQ(expected, object.GetLastCall())
+
+using ::foo::bar::EveryField;
+using ::foo::bar::NestedA;
+using ::foo::bar::SmallEnum;
+using ::foo::bar::SignedEnum;
+using ::foo::bar::BigEnum;
+
+TEST(ProtoZeroTest, EveryField) {
+  EveryField msg;
+  EXPECT_EVAL("int32 f1 = -500", msg, set_field_int32(-500));
+  EXPECT_EVAL("int64 f2 = -5000000000", msg, set_field_int64(-5000000000));
+  EXPECT_EVAL("uint32 f3 = 600", msg, set_field_uint32(600));
+  EXPECT_EVAL("uint64 f4 = 6000000000", msg, set_field_uint64(6000000000));
+  EXPECT_EVAL("sint32 f5 = -700", msg, set_field_sint32(-700));
+  EXPECT_EVAL("sint64 f6 = -7000000000", msg, set_field_sint64(-7000000000));
+  EXPECT_EVAL("fixed32 f7 = 800", msg, set_field_fixed32(800));
+  EXPECT_EVAL("fixed64 f8 = 8000000000", msg, set_field_fixed64(8000000000));
+  EXPECT_EVAL("sfixed32 f9 = -900", msg, set_field_sfixed32(-900));
+  EXPECT_EVAL("sfixed64 f10 = -9000000000", msg, set_field_sfixed64(-9000000000));
+  EXPECT_EVAL("float f11 = 1", msg, set_field_float(1));
+  EXPECT_EVAL("double f12 = 2", msg, set_field_double(2));
+  EXPECT_EVAL("bool f13 = true", msg, set_field_bool(true));
+  EXPECT_EVAL("string f500 = \"foo\"", msg, set_field_string("foo"));
+  EXPECT_EVAL("bytes f505 = [3]", msg, set_field_bytes(nullptr, 3));
+  EXPECT_EVAL("int32 f999 = -8", msg, add_repeated_int32(-8));
+  EXPECT_EVAL("int32 f999 = -9", msg, add_repeated_int32(-9));
+}
+
+TEST(ProtoZeroTest, Enums) {
+  EveryField msg;
+  EXPECT_EVAL("tiny f51 = 1", msg, set_small_enum(SmallEnum::TO_BE));
+  EXPECT_EVAL("int32 f52 = 0", msg, set_signed_enum(SignedEnum::NEUTRAL));
+  EXPECT_EVAL("int32 f53 = 10", msg, set_big_enum(BigEnum::BEGIN));
+  EXPECT_EVAL("tiny f600 = 1", msg, set_nested_enum(EveryField::NestedEnum::PING));
+}
+
+TEST(ProtoZeroTest, NestedMessages) {
+  NestedA msg_a;
+  EXPECT_EVAL("message 2", msg_a, add_repeated_a());
+  EXPECT_TRUE(msg_a.add_repeated_a() != nullptr);
+  NestedA::NestedB msg_b;
+  EXPECT_EVAL("message 1", msg_b, set_value_b());
+  NestedA::NestedB::NestedC msg_c;
+  EXPECT_EVAL("int32 f1 = 100", msg_c, set_value_c(100));
+}
+
+}  // namespace
+}  // namespace proto
+}  // namespace tracing