verify(mockReferralData, atLeastOnce()).getPath();
            verify(mockReferralData, atLeastOnce()).next();
            verify(mockReferralData, atLeastOnce()).getKey();
            verify(mockReferralData, atLeastOnce()).isResolveHashes();
            verify(mockReferralData, atLeastOnce()).isIntermediate();
            verify(mockReferralData, atLeastOnce()).combine(any());

        // Unwrap then close both
        CustomSmbTree unwrapped = smbTree.unwrap(CustomSmbTree.class);
        smbTree.close();
        unwrapped.close();

        verify(smbTree).unwrap(CustomSmbTree.class);
        verify(smbTree).close();
        verify(customTree).close();
    }

    /**
     * Custom SmbTree interface for testing unwrap functionality.
     */
    interface CustomSmbTree extends SmbTree {

        // Verify bytes written
        assertEquals(24, written);

        // Verify structure size (should be 24)
        assertEquals(24, SMBUtil.readInt2(buffer, offset));

        // Verify Reserved1 (2 bytes at offset+2, should be 0)
        assertEquals(0, SMBUtil.readInt2(buffer, offset + 2));

        // Verify Reserved2 (4 bytes at offset+4, should be 0)

        ServletRuntimeException runtimeException = new ServletRuntimeException(servletException);

        // Verify that the cause is correctly set
        assertNotNull(runtimeException);
        assertEquals(servletException, runtimeException.getCause());

        // Verify that the message is preserved from the cause
        assertTrue(runtimeException.getMessage().contains(ServletException.class.getName()));

        // Decode
        int bytesDecoded = smbInfoAllocation.decode(buffer, 0, buffer.length);

        // Verify - actual implementation reads 20 bytes (4 + 4 + 4 + 4 + 4)
        // bytesPerSect is read as Int2 but advances by 4 bytes (padding)
        assertEquals(20, bytesDecoded);

        // Verify capacity calculation: alloc * sectPerAlloc * bytesPerSect
        long expectedCapacity = alloc * sectPerAlloc * bytesPerSect;

        // Then
        assertEquals(mockPipe, pipe);
        verify(mockContext).getPipe(url, pipeType);
    }

    @Test
    @DisplayName("Should close context")
    void testClose() throws CIFSException {
        // When
        mockContext.close();

        // Then
        verify(mockContext).close();
    }

    @Test
    @DisplayName("Should get Configuration")

import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.Mockito.atLeastOnce;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.Arrays;
import java.util.concurrent.TimeUnit;

import org.junit.jupiter.api.BeforeEach;

split -n 10 /tmp/data/mpartobj
CREATE_MPART_OUT=$(aws s3api --endpoint-url=https://localhost:9001 create-multipart-upload --bucket test-bucket --key mpartobj --checksum-algorithm SHA256 --no-verify-ssl --profile enterprise)
UPLOAD_ID=$(echo "${CREATE_MPART_OUT}" | jq '.UploadId' | sed 's/"//g')

PARTS=""

        boolean inUse = pool.close();

        // Then: Should report in-use and call disconnect on all
        assertTrue(inUse, "Should report connections in use");
        verify(pooledSpy).disconnect(false, false);
        verify(nonPooledSpy).disconnect(false, false);
    }

    @Test
    @DisplayName("Should get challenge from server")
    void testGetChallenge() throws Exception {

        BaseConfiguration config = new BaseConfiguration(true);

        // Verify minimum SMB version is at least SMB 2.0.2 (SMB1 disabled)
        assertTrue("Minimum version should be at least SMB2.0.2", config.getMinimumVersion().atLeast(DialectVersion.SMB202));

        // Verify LM compatibility level is 3 or higher (NTLMv2 only)