mirror of
https://gitee.com/mirrors_PX4/PX4-Autopilot.git
synced 2026-05-17 05:27:34 +08:00
Pavel Kirienko
51 lines
1.3 KiB
C++
51 lines
1.3 KiB
C++
/*
|
|
* Copyright (C) 2015 Pavel Kirienko <pavel.kirienko@gmail.com>
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <uavcan/helpers/heap_based_pool_allocator.hpp>
|
|
|
|
inline bool atomicCompareAndSwap(void** address, void* expected, void* replacement)
|
|
{
|
|
return __sync_bool_compare_and_swap(address, expected, replacement);
|
|
}
|
|
|
|
TEST(ThreadSafeHeapBasedPoolAllocator, Basic)
|
|
{
|
|
uavcan::HeapBasedPoolAllocator<uavcan::MemPoolBlockSize, atomicCompareAndSwap> allocator;
|
|
|
|
ASSERT_EQ(0, allocator.getNumCachedBlocks());
|
|
|
|
void* a = allocator.allocate(10);
|
|
void* b = allocator.allocate(10);
|
|
void* c = allocator.allocate(10);
|
|
void* d = allocator.allocate(10);
|
|
|
|
ASSERT_EQ(0, allocator.getNumCachedBlocks());
|
|
|
|
allocator.deallocate(a);
|
|
ASSERT_EQ(1, allocator.getNumCachedBlocks());
|
|
|
|
allocator.deallocate(b);
|
|
ASSERT_EQ(2, allocator.getNumCachedBlocks());
|
|
|
|
allocator.deallocate(c);
|
|
ASSERT_EQ(3, allocator.getNumCachedBlocks());
|
|
|
|
a = allocator.allocate(10);
|
|
ASSERT_EQ(2, allocator.getNumCachedBlocks());
|
|
ASSERT_EQ(c, a);
|
|
|
|
allocator.deallocate(a);
|
|
ASSERT_EQ(3, allocator.getNumCachedBlocks());
|
|
|
|
allocator.shrink();
|
|
ASSERT_EQ(0, allocator.getNumCachedBlocks());
|
|
|
|
allocator.deallocate(d);
|
|
ASSERT_EQ(1, allocator.getNumCachedBlocks());
|
|
|
|
allocator.shrink();
|
|
ASSERT_EQ(0, allocator.getNumCachedBlocks());
|
|
}
|