lenient().when(request.getSession()).thenReturn(session);
        lenient().when(request.getSession(false)).thenReturn(session);
        lenient().when(request.getSession(true)).thenReturn(session);

        // Setup default session attribute behavior
        lenient().when(session.getAttribute(anyString())).thenReturn(null);

        // Setup SmbFile mock
        lenient().when(smbFile.getLocator()).thenReturn(locator);

    void testGetCipherId() {
        // When
        int cipherId = encryptionContext.getCipherId();

        // Then
        assertEquals(1, cipherId, "Should return the cipher ID set in constructor");
    }

    @Test
    @DisplayName("Should return correct dialect version")
    void testGetDialect() {
        // When
        DialectVersion dialect = encryptionContext.getDialect();

        // Set up mock behaviors
        when(mockContext.getConfig()).thenReturn(mockConfig);
        when(mockContext.getCredentials()).thenReturn(mockCredentials);
        when(mockContext.getTransportPool()).thenReturn(mockTransportPool);
        when(mockConfig.getDfsTtl()).thenReturn(300L);

        // Mock transport to throw IOException - simulating connection failure

        when(readResponse.getFlags2()).thenReturn(SmbConstants.FLAGS2_SECURITY_SIGNATURES);
        when(readResponse.getSignSeq()).thenReturn(10);
        when(readResponse.getCommand()).thenReturn((int) ServerMessageBlock.SMB_COM_READ_ANDX);
        when(readResponse.getLength()).thenReturn(100);
        when(readResponse.getDataLength()).thenReturn(50);
        when(readResponse.getData()).thenReturn(readData);

        when(mockReferralData.getShare()).thenReturn(share);

        // When
        String result = mockReferralData.getShare();

        // Then
        assertEquals(share, result);
        verify(mockReferralData).getShare();
    }

    @Test
    @DisplayName("Should get path consumed")
    void testGetPathConsumed() {
        // Given
        int pathConsumed = 10;

        lenient().when(mockSmbNamedPipe.openPipe()).thenReturn(mockSmbPipeHandle);
        lenient().when(mockSmbPipeHandle.unwrap(SmbPipeHandleInternal.class)).thenReturn(mockSmbPipeHandleInternal);
        lenient().when(mockSmbNamedPipe.getContext()).thenReturn(mockContext);
        lenient().when(mockSmbNamedPipe.getLocator()).thenReturn(mockSmbResourceLocator);

        sessionBindingField.setAccessible(true);

        // When
        request.setSessionBinding(true);

        // Then
        assertTrue((boolean) sessionBindingField.get(request));

        // When
        request.setSessionBinding(false);

        // Then
        assertFalse((boolean) sessionBindingField.get(request));
    }

    @Test

        String expectedNextName = "SERVER15";
        when(mockAddress.nextCalledName(mockContext)).thenReturn(expectedNextName);

        // When
        String nextName = mockAddress.nextCalledName(mockContext);

        // Then
        assertEquals(expectedNextName, nextName, "Should return next called name");
        // Note: nextCalledName can return null when no more names available
    }

import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.net.InetAddress;
import java.net.UnknownHostException;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

        when(request.getNext()).thenReturn(nextRequest);
        when(nextRequest.getNext()).thenReturn(thirdRequest);
        when(thirdRequest.getNext()).thenReturn(null);
        when(request.allowChain(nextRequest)).thenReturn(true);
        when(nextRequest.allowChain(thirdRequest)).thenReturn(true);

        // When
        CommonServerMessageBlockRequest second = request.getNext();