Method TestMemoryPoolBookmark

util/memory/pool_ut.cpp:285–320

    }

    void TestMemoryPoolBookmark() {
        TCheckedAllocator alloc;

        {
            TMemoryPool pool(200U, TMemoryPool::TExpGrow::Instance(), &alloc);
            ui64* someData = pool.Allocate<ui64>();
            static const ui64 TESTING{0x123456789ABCDEF};
            *someData = TESTING;

            const auto ma = pool.MemoryAllocated();
            const auto chunkOverhead = ma - sizeof(ui64);
            const auto firstChunkTotal = pool.MemoryWaste() + ma;

            // Allocate some memory in pool but not enough to need new chunks:
            {
                TMemoryPool::TBookmark bookmarkA(pool);
                for (size_t i = 0U; i != 10; ++i) {
                    UNIT_ASSERT(pool.Allocate<ui64>());
                }
            }
            UNIT_ASSERT_VALUES_EQUAL(pool.MemoryAllocated(), ma);
            UNIT_ASSERT_VALUES_EQUAL(*someData, TESTING);

            // Allocate some memory in pool enough to need a new single chunk:
            {
                TMemoryPool::TBookmark bookmarkB(pool);
                for (size_t i = 0U; i != 50; ++i) {
                    UNIT_ASSERT(pool.Allocate<ui64>());
                }
            }
            UNIT_ASSERT_VALUES_EQUAL(pool.MemoryAllocated(), firstChunkTotal + chunkOverhead); // The last (second) chunk is completely free
            UNIT_ASSERT_VALUES_EQUAL(*someData, TESTING);
        }

        alloc.CheckAtEnd();
    }
};

UNIT_TEST_SUITE_REGISTRATION(TMemPoolTest);