Allow Mojo structs as map keys

mojo/public/cpp/bindings/tests/wtf_map_unittest.cc

 // 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 "mojo/public/cpp/bindings/wtf_map.h"
 
 #include "mojo/public/cpp/bindings/lib/wtf_serialization.h"
 #include "mojo/public/cpp/bindings/tests/container_test_util.h"
 #include "mojo/public/cpp/bindings/tests/map_common_test.h"
+#include "mojo/public/cpp/bindings/tests/rect_blink.h"
 #include "mojo/public/cpp/bindings/wtf_array.h"
+#include "mojo/public/interfaces/bindings/tests/rect.mojom-blink.h"
+#include "mojo/public/interfaces/bindings/tests/test_structs.mojom-blink.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/Source/wtf/text/WTFString.h"
 
 namespace mojo {
 namespace test {
 
 namespace {
 
 using WTFMapTest = testing::Test;
 
@@ skipping 34 matching lines @@
   ASSERT_NE(mojo_map.end(), mojo_map.find(123));
 
   WTF::HashMap<int32_t, MoveOnlyType> map2(mojo_map.PassStorage());
   ASSERT_EQ(1u, map2.size());
   ASSERT_NE(map2.end(), map2.find(123));
 
   ASSERT_EQ(0u, mojo_map.size());
   ASSERT_TRUE(mojo_map.is_null());
 }
 
+static blink::RectPtr MakeRect(int32_t x,
+                               int32_t y,
+                               int32_t width,
+                               int32_t height) {
+  blink::RectPtr rect_ptr = blink::Rect::New();
+  rect_ptr->x = x;
+  rect_ptr->y = y;
+  rect_ptr->width = width;
+  rect_ptr->height = height;
+  return rect_ptr;
+}
+
+TEST_F(WTFMapTest, StructKey) {
+  WTF::HashMap<blink::RectPtr, int32_t> map;
+  map.add(MakeRect(1, 2, 3, 4), 123);
+
+  blink::RectPtr key = MakeRect(1, 2, 3, 4);
+  ASSERT_NE(map.end(), map.find(key));
+  ASSERT_EQ(123, map.find(key)->value);
+
+  map.remove(key);
+  ASSERT_EQ(0u, map.size());
+}
+
+static blink::ContainsHashablePtr MakeContainsHashablePtr(RectBlink rect) {
+  blink::ContainsHashablePtr ptr = blink::ContainsHashable::New();
+  ptr->rect = rect;
+  return ptr;
+}
+
+TEST_F(WTFMapTest, TypemappedStructKey) {
+  WTF::HashMap<blink::ContainsHashablePtr, int32_t> map;
+  map.add(MakeContainsHashablePtr(RectBlink(1, 2, 3, 4)), 123);
+
+  blink::ContainsHashablePtr key =
+      MakeContainsHashablePtr(RectBlink(1, 2, 3, 4));
+  ASSERT_NE(map.end(), map.find(key));
+  ASSERT_EQ(123, map.find(key)->value);
+
+  map.remove(key);
+  ASSERT_EQ(0u, map.size());
+}
+
 }  // namespace
 }  // namespace test
 }  // namespace mojo