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");

            assertTrue(initialActive >= 2, "Should have at least 2 active regions");

            // Test buffer release and reuse
            bufferManager.releaseSendRegion(sendRegion);
            bufferManager.releaseReceiveRegion(recvRegion);

            // Allocate again to test pooling
            RdmaMemoryRegion newSendRegion = bufferManager.getSendRegion(4096);

        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();

            EnumSet.of(RdmaAccess.LOCAL_WRITE, RdmaAccess.REMOTE_WRITE));
        totalAllocated.incrementAndGet();
        return region;
    }
    
    public void releaseReceiveRegion(RdmaMemoryRegion region) {
        if (availableReceiveRegions.size() < initialReceiveBuffers * 2) {
            availableReceiveRegions.offer(region);
        } else {
            region.close();