Introduce copyToVector for HeapHashCountedSet

third_party/WebKit/Source/platform/heap/HeapTest.cpp

 /*
  * Copyright (C) 2013 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 27 matching lines @@
 #include "platform/heap/ThreadState.h"
 #include "platform/heap/Visitor.h"
 #include "platform/testing/UnitTestHelpers.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebTaskRunner.h"
 #include "public/platform/WebTraceLocation.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/HashTraits.h"
 #include "wtf/LinkedHashSet.h"
 #include "wtf/PtrUtil.h"
+#include <algorithm>
 #include <memory>
 
 namespace blink {
 
 static void preciselyCollectGarbage()
 {
     ThreadHeap::collectGarbage(BlinkGC::NoHeapPointersOnStack, BlinkGC::GCWithSweep, BlinkGC::ForcedGC);
 }
 
 static void conservativelyCollectGarbage()
@@ skipping 3788 matching lines @@
             preciselyCollectGarbage();
             EXPECT_EQ(0u, weakStrong->size());
             EXPECT_EQ(0u, strongWeak->size());
             EXPECT_EQ(0u, weakWeak->size());
             EXPECT_EQ(0u, weakSet->size());
             EXPECT_EQ(0u, weakOrderedSet->size());
         }
     }
 }
 
+TEST(HeapTest, HeapHashCountedSetToVector)
+{
+    HeapHashCountedSet<Member<IntWrapper>> set;
+    HeapVector<Member<IntWrapper>> vector;
+    set.add(new IntWrapper(1));
+    set.add(new IntWrapper(1));
+    set.add(new IntWrapper(2));
+
+    copyToVector(set, vector);
+    EXPECT_EQ(3u, vector.size());
+
+    Vector<int> intVector;
+    for (const auto& i : vector)
+        intVector.append(i->value());
+    std::sort(intVector.begin(), intVector.end());
+    ASSERT_EQ(3u, intVector.size());
+    EXPECT_EQ(1, intVector[0]);
+    EXPECT_EQ(1, intVector[1]);
+    EXPECT_EQ(2, intVector[2]);
+}
+
+TEST(HeapTest, WeakHeapHashCountedSetToVector)
+{
+    HeapHashCountedSet<WeakMember<IntWrapper>> set;
+    HeapVector<Member<IntWrapper>> vector;
+    set.add(new IntWrapper(1));
+    set.add(new IntWrapper(1));
+    set.add(new IntWrapper(2));
+
+    copyToVector(set, vector);
+    EXPECT_LE(3u, vector.size());
+    for (const auto& i : vector)
+        EXPECT_TRUE(i->value() == 1 || i->value() == 2);
+}
+
 TEST(HeapTest, RefCountedGarbageCollected)
 {
     RefCountedAndGarbageCollected::s_destructorCalls = 0;
     {
         RefPtr<RefCountedAndGarbageCollected> refPtr3;
         {
             Persistent<RefCountedAndGarbageCollected> persistent;
             {
                 Persistent<RefCountedAndGarbageCollected> refPtr1 = RefCountedAndGarbageCollected::create();
                 Persistent<RefCountedAndGarbageCollected> refPtr2 = RefCountedAndGarbageCollected::create();
@@ skipping 3080 matching lines @@
     static_assert(WTF::IsGarbageCollectedType<HeapLinkedHashSet<Member<IntWrapper>>>::value, "HeapLinkedHashSet");
     static_assert(WTF::IsGarbageCollectedType<HeapListHashSet<Member<IntWrapper>>>::value, "HeapListHashSet");
     static_assert(WTF::IsGarbageCollectedType<HeapHashCountedSet<Member<IntWrapper>>>::value, "HeapHashCountedSet");
     static_assert(WTF::IsGarbageCollectedType<HeapHashMap<int, Member<IntWrapper>>>::value, "HeapHashMap");
     static_assert(WTF::IsGarbageCollectedType<HeapVector<Member<IntWrapper>>>::value, "HeapVector");
     static_assert(WTF::IsGarbageCollectedType<HeapDeque<Member<IntWrapper>>>::value, "HeapDeque");
     static_assert(WTF::IsGarbageCollectedType<HeapTerminatedArray<Member<IntWrapper>>>::value, "HeapTerminatedArray");
 }
 
 } // namespace blink