Index: ipc/ipc_message_repeated_field_utils_unittest.cc |
diff --git a/ipc/ipc_message_repeated_field_utils_unittest.cc b/ipc/ipc_message_repeated_field_utils_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..c368947494211d9f3ede0fe344717a96904cb38e |
--- /dev/null |
+++ b/ipc/ipc_message_repeated_field_utils_unittest.cc |
@@ -0,0 +1,172 @@ |
+// Copyright 2017 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 "build/build_config.h" |
+ |
+#if defined(OS_NACL_NONSFI) |
+static_assert(false, |
+ "ipc_message_repeated_field_utils is not able to work with " |
+ "nacl_nonsfi configuration."); |
+#endif |
+ |
+#include "ipc/ipc_message_repeated_field_utils.h" |
+ |
+#include <initializer_list> |
+ |
+#include "ipc/test_proto.pb.h" |
+#include "ipc/ipc_message.h" |
+#include "ipc/ipc_message_utils.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace IPC { |
+ |
+template <> |
+struct ParamTraits<ipc_message_utils_test::TestMessage1> { |
+ typedef ipc_message_utils_test::TestMessage1 param_type; |
+ static void GetSize(base::PickleSizer* sizer, const param_type& p) { |
+ GetParamSize(sizer, p.number()); |
+ } |
+ static void Write(base::Pickle* m, const param_type& p) { |
+ WriteParam(m, p.number()); |
+ } |
+ static bool Read(const base::Pickle* m, |
+ base::PickleIterator* iter, |
+ param_type* r) { |
+ int number; |
+ if (!iter->ReadInt(&number)) |
+ return false; |
+ r->set_number(number); |
+ return true; |
+ } |
+}; |
+ |
+template <> |
+struct ParamTraits<ipc_message_utils_test::TestMessage2> { |
+ typedef ipc_message_utils_test::TestMessage2 param_type; |
+ static void GetSize(base::PickleSizer* sizer, const param_type& p) { |
+ GetParamSize(sizer, p.numbers()); |
+ GetParamSize(sizer, p.strings()); |
+ GetParamSize(sizer, p.messages()); |
+ } |
+ static void Write(base::Pickle* m, const param_type& p) { |
+ WriteParam(m, p.numbers()); |
+ WriteParam(m, p.strings()); |
+ WriteParam(m, p.messages()); |
+ } |
+ static bool Read(const base::Pickle* m, |
+ base::PickleIterator* iter, |
+ param_type* r) { |
+ return ReadParam(m, iter, r->mutable_numbers()) && |
+ ReadParam(m, iter, r->mutable_strings()) && |
+ ReadParam(m, iter, r->mutable_messages()); |
+ } |
+}; |
+ |
+namespace { |
+ |
+template <class P1, class P2> |
+void AssertEqual(const P1& left, const P2& right) { |
+ ASSERT_EQ(left, right); |
+} |
+ |
+template<> |
+void AssertEqual(const int& left, |
+ const ipc_message_utils_test::TestMessage1& right) { |
+ ASSERT_EQ(left, right.number()); |
+} |
+ |
+template <template<class> class RepeatedFieldLike, class P1, class P2> |
+void AssertRepeatedFieldEquals(std::initializer_list<P1> expected, |
+ const RepeatedFieldLike<P2>& fields) { |
+ ASSERT_EQ(static_cast<int>(expected.size()), fields.size()); |
+ auto it = expected.begin(); |
+ int i = 0; |
+ for (; it != expected.end(); it++, i++) { |
+ AssertEqual(*it, fields.Get(i)); |
+ } |
+} |
+ |
+TEST(IPCMessageRepeatedFieldUtilsTest, RepeatedFieldShouldBeSerialized) { |
+ ipc_message_utils_test::TestMessage2 message; |
+ message.add_numbers(1); |
+ message.add_numbers(100); |
+ message.add_strings("abc"); |
+ message.add_strings("def"); |
+ { |
+ ipc_message_utils_test::TestMessage1 message1; |
+ message1.set_number(1000); |
+ *message.add_messages() = message1; |
+ message1.set_number(10000); |
+ *message.add_messages() = message1; |
+ } |
+ |
+ base::Pickle pickle; |
+ IPC::WriteParam(&pickle, message); |
+ |
+ base::PickleSizer sizer; |
+ IPC::GetParamSize(&sizer, message); |
+ |
+ EXPECT_EQ(sizer.payload_size(), pickle.payload_size()); |
+ |
+ base::PickleIterator iter(pickle); |
+ ipc_message_utils_test::TestMessage2 output; |
+ EXPECT_TRUE(IPC::ReadParam(&pickle, &iter, &output)); |
+ |
+ AssertRepeatedFieldEquals({1, 100}, output.numbers()); |
+ AssertRepeatedFieldEquals({"abc", "def"}, output.strings()); |
+ AssertRepeatedFieldEquals({1000, 10000}, output.messages()); |
+} |
+ |
+TEST(IPCMessageRepeatedFieldUtilsTest, PartialEmptyRepeatedFieldShouldBeSerialized) { |
+ ipc_message_utils_test::TestMessage2 message; |
+ message.add_numbers(1); |
+ message.add_numbers(100); |
+ { |
+ ipc_message_utils_test::TestMessage1 message1; |
+ message1.set_number(1000); |
+ *message.add_messages() = message1; |
+ message1.set_number(10000); |
+ *message.add_messages() = message1; |
+ } |
+ |
+ base::Pickle pickle; |
+ IPC::WriteParam(&pickle, message); |
+ |
+ base::PickleSizer sizer; |
+ IPC::GetParamSize(&sizer, message); |
+ |
+ EXPECT_EQ(sizer.payload_size(), pickle.payload_size()); |
+ |
+ base::PickleIterator iter(pickle); |
+ ipc_message_utils_test::TestMessage2 output; |
+ EXPECT_TRUE(IPC::ReadParam(&pickle, &iter, &output)); |
+ |
+ AssertRepeatedFieldEquals({1, 100}, output.numbers()); |
+ ASSERT_EQ(0, output.strings_size()); |
+ AssertRepeatedFieldEquals({1000, 10000}, output.messages()); |
+} |
+ |
+TEST(IPCMessageRepeatedFieldUtilsTest, EmptyRepeatedFieldShouldBeSerialized) { |
+ ipc_message_utils_test::TestMessage2 message; |
+ |
+ base::Pickle pickle; |
+ IPC::WriteParam(&pickle, message); |
+ |
+ base::PickleSizer sizer; |
+ IPC::GetParamSize(&sizer, message); |
+ |
+ EXPECT_EQ(sizer.payload_size(), pickle.payload_size()); |
+ |
+ base::PickleIterator iter(pickle); |
+ ipc_message_utils_test::TestMessage2 output; |
+ EXPECT_TRUE(IPC::ReadParam(&pickle, &iter, &output)); |
+ |
+ ASSERT_EQ(0, output.numbers_size()); |
+ ASSERT_EQ(0, output.strings_size()); |
+ ASSERT_EQ(0, output.messages_size()); |
+} |
+ |
+} // namespace |
+ |
+} // namespace IPC |