Home | History | Annotate | Download | only in tests 
      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 #include "config.h"
     27 
     28 #include "PODArena.h"
     29 
     30 #include "ArenaTestHelpers.h"
     31 #include <algorithm>
     32 #include <gtest/gtest.h>
     33 #include <wtf/FastMalloc.h>
     34 #include <wtf/RefPtr.h>
     35 #include <wtf/Vector.h>
     36 
     37 namespace WebCore {
     38 
     39 using ArenaTestHelpers::TrackedAllocator;
     40 
     41 namespace {
     42 
     43 // A couple of simple structs to allocate.
     44 struct TestClass1 {
     45     TestClass1()
     46         : x(0), y(0), z(0), w(1) { }
     47 
     48     float x, y, z, w;
     49 };
     50 
     51 struct TestClass2 {
     52     TestClass2()
     53         : a(1), b(2), c(3), d(4) { }
     54 
     55     float a, b, c, d;
     56 };
     57 
     58 } // anonymous namespace
     59 
     60 class PODArenaTest : public testing::Test {
     61 };
     62 
     63 // Make sure the arena can successfully allocate from more than one
     64 // region.
     65 TEST_F(PODArenaTest, CanAllocateFromMoreThanOneRegion)
     66 {
     67     RefPtr<TrackedAllocator> allocator = TrackedAllocator::create();
     68     RefPtr<PODArena> arena = PODArena::create(allocator);
     69     int numIterations = 10 * PODArena::DefaultChunkSize / sizeof(TestClass1);
     70     for (int i = 0; i < numIterations; ++i)
     71         arena->allocateObject<TestClass1>();
     72     EXPECT_GT(allocator->numRegions(), 1);
     73 }
     74 
     75 // Make sure the arena frees all allocated regions during destruction.
     76 TEST_F(PODArenaTest, FreesAllAllocatedRegions)
     77 {
     78     RefPtr<TrackedAllocator> allocator = TrackedAllocator::create();
     79     {
     80         RefPtr<PODArena> arena = PODArena::create(allocator);
     81         for (int i = 0; i < 3; i++)
     82             arena->allocateObject<TestClass1>();
     83         EXPECT_GT(allocator->numRegions(), 0);
     84     }
     85     EXPECT_TRUE(allocator->isEmpty());
     86 }
     87 
     88 // Make sure the arena runs constructors of the objects allocated within.
     89 TEST_F(PODArenaTest, RunsConstructors)
     90 {
     91     RefPtr<PODArena> arena = PODArena::create();
     92     for (int i = 0; i < 10000; i++) {
     93         TestClass1* tc1 = arena->allocateObject<TestClass1>();
     94         EXPECT_EQ(0, tc1->x);
     95         EXPECT_EQ(0, tc1->y);
     96         EXPECT_EQ(0, tc1->z);
     97         EXPECT_EQ(1, tc1->w);
     98         TestClass2* tc2 = arena->allocateObject<TestClass2>();
     99         EXPECT_EQ(1, tc2->a);
    100         EXPECT_EQ(2, tc2->b);
    101         EXPECT_EQ(3, tc2->c);
    102         EXPECT_EQ(4, tc2->d);
    103     }
    104 }
    105 
    106 } // namespace WebCore
    107