Index: chrome/common/common_param_traits_unittest.cc |
diff --git a/chrome/common/common_param_traits_unittest.cc b/chrome/common/common_param_traits_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f3a7e922097b2aaf996bd7a09b098a442eb5fc8b |
--- /dev/null |
+++ b/chrome/common/common_param_traits_unittest.cc |
@@ -0,0 +1,151 @@ |
+// Copyright (c) 2006-2008 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 <string.h> |
+ |
+#include "base/scoped_ptr.h" |
+#include "base/values.h" |
+#include "chrome/common/common_param_traits.h" |
+#include "googleurl/src/gurl.h" |
+#include "ipc/ipc_message.h" |
+#include "ipc/ipc_message_utils.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+#include "third_party/skia/include/core/SkBitmap.h" |
+ |
+// Tests that serialize/deserialize correctly understand each other |
+TEST(IPCMessageTest, Serialize) { |
+ const char* serialize_cases[] = { |
+ "http://www.google.com/", |
+ "http://user:pass@host.com:888/foo;bar?baz#nop", |
+ "#inva://idurl/", |
+ }; |
+ |
+ for (size_t i = 0; i < arraysize(serialize_cases); i++) { |
+ GURL input(serialize_cases[i]); |
+ IPC::Message msg(1, 2, IPC::Message::PRIORITY_NORMAL); |
+ IPC::ParamTraits<GURL>::Write(&msg, input); |
+ |
+ GURL output; |
+ void* iter = NULL; |
+ EXPECT_TRUE(IPC::ParamTraits<GURL>::Read(&msg, &iter, &output)); |
+ |
+ // We want to test each component individually to make sure its range was |
+ // correctly serialized and deserialized, not just the spec. |
+ EXPECT_EQ(input.possibly_invalid_spec(), output.possibly_invalid_spec()); |
+ EXPECT_EQ(input.is_valid(), output.is_valid()); |
+ EXPECT_EQ(input.scheme(), output.scheme()); |
+ EXPECT_EQ(input.username(), output.username()); |
+ EXPECT_EQ(input.password(), output.password()); |
+ EXPECT_EQ(input.host(), output.host()); |
+ EXPECT_EQ(input.port(), output.port()); |
+ EXPECT_EQ(input.path(), output.path()); |
+ EXPECT_EQ(input.query(), output.query()); |
+ EXPECT_EQ(input.ref(), output.ref()); |
+ } |
+ |
+ // Also test the corrupt case. |
+ IPC::Message msg(1, 2, IPC::Message::PRIORITY_NORMAL); |
+ msg.WriteInt(99); |
+ GURL output; |
+ void* iter = NULL; |
+ EXPECT_FALSE(IPC::ParamTraits<GURL>::Read(&msg, &iter, &output)); |
+} |
+ |
+// Tests bitmap serialization. |
+TEST(IPCMessageTest, Bitmap) { |
+ SkBitmap bitmap; |
+ |
+ bitmap.setConfig(SkBitmap::kARGB_8888_Config, 10, 5); |
+ bitmap.allocPixels(); |
+ memset(bitmap.getPixels(), 'A', bitmap.getSize()); |
+ |
+ IPC::Message msg(1, 2, IPC::Message::PRIORITY_NORMAL); |
+ IPC::ParamTraits<SkBitmap>::Write(&msg, bitmap); |
+ |
+ SkBitmap output; |
+ void* iter = NULL; |
+ EXPECT_TRUE(IPC::ParamTraits<SkBitmap>::Read(&msg, &iter, &output)); |
+ |
+ EXPECT_EQ(bitmap.config(), output.config()); |
+ EXPECT_EQ(bitmap.width(), output.width()); |
+ EXPECT_EQ(bitmap.height(), output.height()); |
+ EXPECT_EQ(bitmap.rowBytes(), output.rowBytes()); |
+ EXPECT_EQ(bitmap.getSize(), output.getSize()); |
+ EXPECT_EQ(memcmp(bitmap.getPixels(), output.getPixels(), bitmap.getSize()), |
+ 0); |
+ |
+ // Also test the corrupt case. |
+ IPC::Message bad_msg(1, 2, IPC::Message::PRIORITY_NORMAL); |
+ // Copy the first message block over to |bad_msg|. |
+ const char* fixed_data; |
+ int fixed_data_size; |
+ iter = NULL; |
+ msg.ReadData(&iter, &fixed_data, &fixed_data_size); |
+ bad_msg.WriteData(fixed_data, fixed_data_size); |
+ // Add some bogus pixel data. |
+ const size_t bogus_pixels_size = bitmap.getSize() * 2; |
+ scoped_array<char> bogus_pixels(new char[bogus_pixels_size]); |
+ memset(bogus_pixels.get(), 'B', bogus_pixels_size); |
+ bad_msg.WriteData(bogus_pixels.get(), bogus_pixels_size); |
+ // Make sure we don't read out the bitmap! |
+ SkBitmap bad_output; |
+ iter = NULL; |
+ EXPECT_FALSE(IPC::ParamTraits<SkBitmap>::Read(&bad_msg, &iter, &bad_output)); |
+} |
+ |
+TEST(IPCMessageTest, ListValue) { |
+ ListValue input; |
+ input.Set(0, Value::CreateRealValue(42.42)); |
+ input.Set(1, Value::CreateStringValue("forty")); |
+ input.Set(2, Value::CreateNullValue()); |
+ |
+ IPC::Message msg(1, 2, IPC::Message::PRIORITY_NORMAL); |
+ IPC::WriteParam(&msg, input); |
+ |
+ ListValue output; |
+ void* iter = NULL; |
+ EXPECT_TRUE(IPC::ReadParam(&msg, &iter, &output)); |
+ |
+ EXPECT_TRUE(input.Equals(&output)); |
+ |
+ // Also test the corrupt case. |
+ IPC::Message bad_msg(1, 2, IPC::Message::PRIORITY_NORMAL); |
+ bad_msg.WriteInt(99); |
+ iter = NULL; |
+ EXPECT_FALSE(IPC::ReadParam(&bad_msg, &iter, &output)); |
+} |
+ |
+TEST(IPCMessageTest, DictionaryValue) { |
+ DictionaryValue input; |
+ input.Set(L"null", Value::CreateNullValue()); |
+ input.Set(L"bool", Value::CreateBooleanValue(true)); |
+ input.Set(L"int", Value::CreateIntegerValue(42)); |
+ |
+ scoped_ptr<DictionaryValue> subdict(new DictionaryValue()); |
+ subdict->Set(L"str", Value::CreateStringValue("forty two")); |
+ subdict->Set(L"bool", Value::CreateBooleanValue(false)); |
+ |
+ scoped_ptr<ListValue> sublist(new ListValue()); |
+ sublist->Set(0, Value::CreateRealValue(42.42)); |
+ sublist->Set(1, Value::CreateStringValue("forty")); |
+ sublist->Set(2, Value::CreateStringValue("two")); |
+ subdict->Set(L"list", sublist.release()); |
+ |
+ input.Set(L"dict", subdict.release()); |
+ |
+ IPC::Message msg(1, 2, IPC::Message::PRIORITY_NORMAL); |
+ IPC::WriteParam(&msg, input); |
+ |
+ DictionaryValue output; |
+ void* iter = NULL; |
+ EXPECT_TRUE(IPC::ReadParam(&msg, &iter, &output)); |
+ |
+ EXPECT_TRUE(input.Equals(&output)); |
+ |
+ // Also test the corrupt case. |
+ IPC::Message bad_msg(1, 2, IPC::Message::PRIORITY_NORMAL); |
+ bad_msg.WriteInt(99); |
+ iter = NULL; |
+ EXPECT_FALSE(IPC::ReadParam(&bad_msg, &iter, &output)); |
+} |