rpc.retval = 0; // Success
            return null;
        }).when(mockDcerpcHandle).sendrecv(any(MsrpcSamrCloseHandle.class));

        // Act
        handle.close();

        // Assert
        // Verify sendrecv was called for open and then for close
        verify(mockDcerpcHandle, times(1)).sendrecv(any(MsrpcSamrOpenDomain.class));

        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 testConstructorWithNullFilename() {
        // When & Then
        assertDoesNotThrow(() -> new Trans2GetDfsReferral(mockConfig, null));
    }

    @Test
    @DisplayName("Constructor should handle empty filename")
    void testConstructorWithEmptyFilename() {
        // When & Then
        assertDoesNotThrow(() -> new Trans2GetDfsReferral(mockConfig, ""));
    }

    @Test

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

        byte[] dst = new byte[42];

        // When
        int result = trans2SetFileInformation.writeDataWireFormat(dst, 0);

        // Then
        assertEquals(40, result); // Returns 40 bytes written
    }

    @Test
    void testReadSetupWireFormat() {
        // When
        int result = trans2SetFileInformation.readSetupWireFormat(new byte[0], 0, 0);

        // Then
        assertEquals(0, result);
    }

    @BeforeEach
    void setUp() throws Exception {
        when(ctx.getConfig()).thenReturn(cfg);
        when(cfg.isSigningEnforced()).thenReturn(false);
        when(cfg.getSessionTimeout()).thenReturn(30_000);
        when(cfg.getResponseTimeout()).thenReturn(5_000);
        when(address.getHostAddress()).thenReturn("127.0.0.1");
        when(address.getHostName()).thenReturn("localhost");

            when(mockChunk1.encode(any(byte[].class), anyInt())).thenReturn(CHUNK_SIZE);
            when(mockChunk2.encode(any(byte[].class), anyInt())).thenReturn(CHUNK_SIZE);

            SrvCopychunkCopy copy = new SrvCopychunkCopy(sourceKey, mockChunk1, mockChunk2);
            byte[] buffer = new byte[200];

            // When
            copy.encode(buffer, 0);

            // Then

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(kerberosKey.getKeyType()).thenReturn(ENCRYPTION_TYPE);

        try (MockedStatic<KerberosEncData> mockedEncData = Mockito.mockStatic(KerberosEncData.class)) {
            mockedEncData.when(() -> KerberosEncData.decrypt(ENCRYPTED_DATA, kerberosKey, ENCRYPTION_TYPE)).thenReturn(decryptedData);

            // When