return newRegion;
    }

    /**
     * Release a receive region back to the pool
     *
     * @param region memory region to release
     */
    public void releaseReceiveRegion(RdmaMemoryRegion region) {
        if (availableReceiveRegions.size() < initialReceiveBuffers * 2) {
            availableReceiveRegions.offer(region);
        } else {
            region.close();

        RdmaMemoryRegion region = bufferManager.getReceiveRegion();
        assertNotNull(region);

        // Release should not throw exception
        assertDoesNotThrow(() -> bufferManager.releaseReceiveRegion(region));
    }

    @Test
    public void testAllocateBuffer() {
        ByteBuffer buffer = bufferManager.allocateBuffer(2048);
        assertNotNull(buffer, "Buffer should not be null");