WinCairo build fails to link.
[WebKit-https.git] / Source / WebKit / chromium / tests / 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 "PODArena.h"
30 #include <gtest/gtest.h>
31 #include <wtf/NotFound.h>
32 #include <wtf/Vector.h>
33
34 namespace WebCore {
35 namespace ArenaTestHelpers {
36
37 // An allocator for the PODArena which tracks the regions which have
38 // been allocated.
39 class TrackedAllocator : public PODArena::FastMallocAllocator {
40 public:
41     static PassRefPtr<TrackedAllocator> create()
42     {
43         return adoptRef(new TrackedAllocator);
44     }
45
46     virtual void* allocate(size_t size)
47     {
48         void* result = PODArena::FastMallocAllocator::allocate(size);
49         m_allocatedRegions.append(result);
50         return result;
51     }
52
53     virtual void free(void* ptr)
54     {
55         size_t slot = m_allocatedRegions.find(ptr);
56         ASSERT_NE(slot, notFound);
57         m_allocatedRegions.remove(slot);
58         PODArena::FastMallocAllocator::free(ptr);
59     }
60
61     bool isEmpty() const
62     {
63         return !numRegions();
64     }
65
66     int numRegions() const
67     {
68         return m_allocatedRegions.size();
69     }
70
71 private:
72     TrackedAllocator() { }
73     Vector<void*> m_allocatedRegions;
74 };
75
76 } // namespace ArenaTestHelpers
77 } // namespace WebCore
78
79 #endif // ArenaTestHelpers_h