pikapython/port/linux/test/pool-test.cpp

#include "gtest/gtest.h"
extern "C" {
#include "dataArg.h"
#include "dataMemory.h"
#include "dataString.h"
}

TEST(pool, init) {
    Pool p = pool_init(125, 4);
    EXPECT_EQ(p.aline, 4);
    EXPECT_EQ(p.size, 128);
    pool_printBlocks(&p, 0, 128);
    pool_deinit(&p);
    EXPECT_EQ((uint64_t)p.mem, NULL);
}

TEST(pool, m1) {
    Pool p = pool_init(125, 4);
    EXPECT_EQ(p.aline, 4);
    EXPECT_EQ(p.size, 128);
    uint64_t mem1_offset = (uint64_t)pool_malloc(&p, 4) - (uint64_t)p.mem;
    pool_printBlocks(&p, 0, 128);
    EXPECT_EQ(1, bitmap_get(p.bitmap, 0));
    EXPECT_EQ(0, bitmap_get(p.bitmap, 1));
    EXPECT_EQ(0, mem1_offset);
    pool_deinit(&p);
}

TEST(pool, m2) {
    Pool p = pool_init(125, 4);
    EXPECT_EQ(p.aline, 4);
    EXPECT_EQ(p.size, 128);
    uint64_t mem1_offset = (uint64_t)pool_malloc(&p, 4) - (uint64_t)p.mem;
    uint64_t mem2_offset = (uint64_t)pool_malloc(&p, 4) - (uint64_t)p.mem;
    pool_printBlocks(&p, 0, 128);
    EXPECT_EQ(1, bitmap_get(p.bitmap, 0));
    EXPECT_EQ(1, bitmap_get(p.bitmap, 1));
    EXPECT_EQ(0, mem1_offset);
    EXPECT_EQ(4, mem2_offset);
    pool_deinit(&p);
}

TEST(pool, mf1) {
    Pool p = pool_init(125, 4);
    EXPECT_EQ(p.aline, 4);
    EXPECT_EQ(p.size, 128);
    void* mem1 = pool_malloc(&p, 4);
    uint64_t mem1_offset = (uint64_t)mem1 - (uint64_t)p.mem;
    EXPECT_EQ(1, bitmap_get(p.bitmap, 0));
    pool_printBlocks(&p, 0, 128);
    pool_free(&p, mem1, 4);
    pool_printBlocks(&p, 0, 128);
    EXPECT_EQ(0, bitmap_get(p.bitmap, 0));
    EXPECT_EQ(0, mem1_offset);
    pool_deinit(&p);
}

TEST(pool, mf2) {
    Pool p = pool_init(125, 16);
    EXPECT_EQ(p.aline, 16);
    EXPECT_EQ(p.size, 128);
    void* mem1 = pool_malloc(&p, 4);
    uint64_t mem1_offset = (uint64_t)mem1 - (uint64_t)p.mem;
    EXPECT_EQ(1, bitmap_get(p.bitmap, 0));
    pool_printBlocks(&p, 0, 128);
    pool_free(&p, mem1, 4);
    pool_printBlocks(&p, 0, 128);
    EXPECT_EQ(0, bitmap_get(p.bitmap, 0));
    EXPECT_EQ(0, mem1_offset);
    pool_deinit(&p);
}

TEST(pool, mfm1) {
    Pool p = pool_init(125, 4);
    EXPECT_EQ(p.size, 128);
    void* mem1 = pool_malloc(&p, 4);
    pool_malloc(&p, 16);
    pool_free(&p, mem1, 4);
    pool_malloc(&p, 16);
    pool_printBlocks(&p, 0, 128);

    pool_deinit(&p);
}