diff -r 44986c0e2b05 -r d92a59f5d261 tests/test_mempool.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/test_mempool.cpp Wed Jun 28 20:07:52 2023 +0200 @@ -0,0 +1,153 @@ +/* + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. + * + * Copyright 2021 Mike Becker, Olaf Wintermann All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + */ + +#include "cx/mempool.h" +#include "util_allocator.h" +#include + +TEST(Mempool, Create) { + auto pool = cxBasicMempoolCreate(16); + ASSERT_EQ(pool->auto_destr, nullptr); + ASSERT_NE(pool->allocator, nullptr); + ASSERT_NE(pool->allocator->cl, nullptr); + EXPECT_EQ(pool->allocator->data, pool); + EXPECT_NE(pool->allocator->cl->malloc, nullptr); + EXPECT_NE(pool->allocator->cl->calloc, nullptr); + EXPECT_NE(pool->allocator->cl->realloc, nullptr); + EXPECT_NE(pool->allocator->cl->free, nullptr); + EXPECT_EQ(pool->capacity, 16); + EXPECT_EQ(pool->size, 0); + EXPECT_NE(pool->data, nullptr); + cxMempoolDestroy(pool); +} + +TEST(Mempool, malloc) { + auto pool = cxBasicMempoolCreate(4); + EXPECT_NE(cxMalloc(pool->allocator, sizeof(int)), nullptr); + EXPECT_NE(cxMalloc(pool->allocator, sizeof(int)), nullptr); + EXPECT_EQ(pool->size, 2); + EXPECT_EQ(pool->capacity, 4); + EXPECT_NE(cxMalloc(pool->allocator, sizeof(int)), nullptr); + EXPECT_NE(cxMalloc(pool->allocator, sizeof(int)), nullptr); + EXPECT_EQ(pool->size, 4); + EXPECT_EQ(pool->capacity, 4); + EXPECT_NE(cxMalloc(pool->allocator, sizeof(int)), nullptr); + EXPECT_NE(cxMalloc(pool->allocator, sizeof(int)), nullptr); + EXPECT_EQ(pool->size, 6); + EXPECT_GE(pool->capacity, 6); + cxMempoolDestroy(pool); +} + +TEST(Mempool, calloc) { + auto pool = cxBasicMempoolCreate(4); + + auto test = (int *) cxCalloc(pool->allocator, 2, sizeof(int)); + ASSERT_NE(test, nullptr); + EXPECT_EQ(test[0], 0); + EXPECT_EQ(test[1], 0); + cxMempoolDestroy(pool); +} + +static unsigned test_destructor_called; + +static void test_destructor([[maybe_unused]] void *mem) { + test_destructor_called++; +} + +TEST(Mempool, realloc) { + auto pool = cxMempoolCreate(4, test_destructor); + ASSERT_EQ(pool->auto_destr, test_destructor); + auto data = cxMalloc(pool->allocator, sizeof(int)); + *((int *) data) = 13; + + void *rdata = data; + unsigned n = 1; + while (rdata == data) { + n <<= 1; + ASSERT_LT(n, 65536); // eventually the memory should be moved elsewhere + rdata = cxRealloc(pool->allocator, data, n * sizeof(intptr_t)); + } + + EXPECT_EQ(*((int *) rdata), 13); + // test if destructor is still intact + test_destructor_called = 0; + cxFree(pool->allocator, rdata); + EXPECT_EQ(test_destructor_called, 1); + cxMempoolDestroy(pool); +} + + +TEST(Mempool, free) { + auto pool = cxBasicMempoolCreate(4); + + void *mem1; + void *mem2; + + mem1 = cxMalloc(pool->allocator, 16); + cxFree(pool->allocator, mem1); + EXPECT_EQ(pool->size, 0); + + cxMalloc(pool->allocator, 16); + cxMalloc(pool->allocator, 16); + mem1 = cxMalloc(pool->allocator, 16); + cxMalloc(pool->allocator, 16); + mem2 = cxMalloc(pool->allocator, 16); + + EXPECT_EQ(pool->size, 5); + cxFree(pool->allocator, mem1); + EXPECT_EQ(pool->size, 4); + cxFree(pool->allocator, mem2); + EXPECT_EQ(pool->size, 3); + cxMempoolDestroy(pool); +} + +TEST(Mempool, Destroy) { + auto pool = cxBasicMempoolCreate(4); + auto data = cxMalloc(pool->allocator, sizeof(int)); + *((int *) data) = 13; + cxMempoolSetDestructor(data, test_destructor); + EXPECT_EQ(*((int *) data), 13); + test_destructor_called = 0; + cxFree(pool->allocator, data); + EXPECT_EQ(test_destructor_called, 1); + data = cxMalloc(pool->allocator, sizeof(int)); + cxMempoolSetDestructor(data, test_destructor); + cxMempoolDestroy(pool); + EXPECT_EQ(test_destructor_called, 2); +} + +TEST(Mempool, Register) { + auto pool = cxBasicMempoolCreate(4); + auto data = cxMalloc(pool->allocator, sizeof(int)); + test_destructor_called = 0; + cxMempoolSetDestructor(data, test_destructor); + int donotfree = 0; + cxMempoolRegister(pool, &donotfree, test_destructor); + cxMempoolDestroy(pool); + EXPECT_EQ(test_destructor_called, 2); +}