third_party/WebKit/Source/platform/testing/ArenaTestHelpers.h

   1 /*
   2  * Copyright (C) 2010 Google Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   8  * 1.  Redistributions of source code must retain the above copyright
   9  *     notice, this list of conditions and the following disclaimer.
  10  * 2.  Redistributions in binary form must reproduce the above copyright
  11  *     notice, this list of conditions and the following disclaimer in the
  12  *     documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  16  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  17  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  18  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  19  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  20  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  21  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #ifndef ArenaTestHelpers_h
  27 #define ArenaTestHelpers_h
  28
  29 #include "platform/PODArena.h"
  30 #include "wtf/NotFound.h"
  31 #include "wtf/Vector.h"
  32
  33 #include <gtest/gtest.h>
  34
  35 namespace blink {
  36 namespace ArenaTestHelpers {
  37
  38 // An allocator for the PODArena which tracks the regions which have
  39 // been allocated.
  40 class TrackedAllocator final : public PODArena::FastMallocAllocator {
  41 public:
  42     static PassRefPtr<TrackedAllocator> create()
  43     {
  44         return adoptRef(new TrackedAllocator);
  45     }
  46
  47     void* allocate(size_t size) override
  48     {
  49         void* result = PODArena::FastMallocAllocator::allocate(size);
  50         m_allocatedRegions.append(result);
  51         return result;
  52     }
  53
  54     void free(void* ptr) override
  55     {
  56         size_t slot = m_allocatedRegions.find(ptr);
  57         ASSERT_NE(slot, kNotFound);
  58         m_allocatedRegions.remove(slot);
  59         PODArena::FastMallocAllocator::free(ptr);
  60     }
  61
  62     bool isEmpty() const
  63     {
  64         return !numRegions();
  65     }
  66
  67     int numRegions() const
  68     {
  69         return m_allocatedRegions.size();
  70     }
  71
  72 private:
  73     TrackedAllocator() { }
  74     Vector<void*> m_allocatedRegions;
  75 };
  76
  77 } // namespace ArenaTestHelpers
  78 } // namespace blink
  79
  80 #endif // ArenaTestHelpers_h