ScopedPtrMap: Added Compare template parameter.

base/containers/scoped_ptr_map_unittest.cc

 // Copyright 2015 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 "base/containers/scoped_ptr_map.h"
 
+#include <functional>
 #include <map>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/memory/scoped_ptr.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
@@ skipping 119 matching lines @@
 TEST(ScopedPtrMapTest, Clear) {
   bool destroyed = false;
   ScopedPtrMap<int, scoped_ptr<ScopedDestroyer>> scoped_map;
   scoped_map.insert(0, make_scoped_ptr(new ScopedDestroyer(&destroyed)));
   EXPECT_FALSE(destroyed);
   scoped_map.clear();
   EXPECT_TRUE(destroyed);
   EXPECT_TRUE(scoped_map.empty());
 }
 
+TEST(ScopedPtrMapTest, Compare) {
+  // Construct a ScopedPtrMap with a custom comparison function.
+  bool destroyed = false;
+  ScopedPtrMap<int, scoped_ptr<ScopedDestroyer>, std::greater<int>> scoped_map;
+  scoped_map.insert(0, make_scoped_ptr(new ScopedDestroyer(&destroyed)));
+  scoped_map.insert(1, make_scoped_ptr(new ScopedDestroyer(&destroyed)));
+
+  auto it = scoped_map.begin();
+  EXPECT_EQ(1, it->first);
+  ++it;
+  EXPECT_EQ(0, it->first);
+}
+
 TEST(ScopedPtrMapTest, Scope) {
   bool destroyed = false;
   {
     ScopedPtrMap<int, scoped_ptr<ScopedDestroyer>> scoped_map;
     scoped_map.insert(0, make_scoped_ptr(new ScopedDestroyer(&destroyed)));
     EXPECT_FALSE(destroyed);
   }
   EXPECT_TRUE(destroyed);
 }
 
@@ skipping 59 matching lines @@
   ScopedPtrMap<int, scoped_ptr<ScopedDestroyer>> result = callback.Run();
   EXPECT_TRUE(scoped_map.empty());
   EXPECT_EQ(elem, result.find(0)->second);
   EXPECT_FALSE(destroyed);
 
   result.clear();
   EXPECT_TRUE(destroyed);
 };
 
 }  // namespace