test_object_pool.cpp
Go to the documentation of this file.00001 /********************************************************************* 00002 * Software License Agreement (BSD License) 00003 * 00004 * Copyright (c) 2010, Willow Garage, Inc. 00005 * All rights reserved. 00006 * 00007 * Redistribution and use in source and binary forms, with or without 00008 * modification, are permitted provided that the following conditions 00009 * are met: 00010 * 00011 * * Redistributions of source code must retain the above copyright 00012 * notice, this list of conditions and the following disclaimer. 00013 * * Redistributions in binary form must reproduce the above 00014 * copyright notice, this list of conditions and the following 00015 * disclaimer in the documentation and/or other materials provided 00016 * with the distribution. 00017 * * Neither the name of the Willow Garage nor the names of its 00018 * contributors may be used to endorse or promote products derived 00019 * from this software without specific prior written permission. 00020 * 00021 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 00022 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 00023 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 00024 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 00025 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 00026 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 00027 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 00028 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 00029 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 00030 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 00031 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 00032 * POSSIBILITY OF SUCH DAMAGE. 00033 *********************************************************************/ 00034 00035 #include <gtest/gtest.h> 00036 00037 #include "lockfree/object_pool.h" 00038 00039 #include <set> 00040 00041 using namespace lockfree; 00042 00043 TEST(ObjectPool, oneElement) 00044 { 00045 ObjectPool<uint32_t> pool(1, 5); 00046 00047 boost::shared_ptr<uint32_t> item = pool.allocateShared(); 00048 ASSERT_TRUE(item); 00049 EXPECT_EQ(*item, 5UL); 00050 *item = 6; 00051 ASSERT_FALSE(pool.allocateShared()); 00052 item.reset(); 00053 item = pool.allocateShared(); 00054 ASSERT_TRUE(item); 00055 EXPECT_EQ(*item, 6UL); 00056 } 00057 00058 TEST(ObjectPool, multipleElements) 00059 { 00060 const uint32_t count = 5; 00061 ObjectPool<uint32_t> pool(count, 5); 00062 00063 std::vector<boost::shared_ptr<uint32_t> > items; 00064 items.reserve(count); 00065 00066 for (uint32_t i = 0; i < count; ++i) 00067 { 00068 items.push_back(pool.allocateShared()); 00069 ASSERT_TRUE(items[i]); 00070 EXPECT_EQ(*items[i], 5UL); 00071 *items[i] = i; 00072 } 00073 00074 ASSERT_FALSE(pool.allocateShared()); 00075 items.pop_back(); 00076 items.push_back(pool.allocateShared()); 00077 ASSERT_TRUE(items.back()); 00078 ASSERT_FALSE(pool.allocateShared()); 00079 00080 std::set<boost::shared_ptr<uint32_t> > set; 00081 set.insert(items.begin(), items.end()); 00082 EXPECT_EQ(set.size(), count); 00083 } 00084 00085 int main(int argc, char** argv) 00086 { 00087 testing::InitGoogleTest(&argc, argv); 00088 return RUN_ALL_TESTS(); 00089 }