// When
            int result = response.writeBytesWireFormat(dst, 0);

            // Then
            assertEquals(0, result);
        }

        @Test
        @DisplayName("Should return 0 with null array")
        void testWriteBytesWireFormatWithNullArray() {
            // Given
            byte[] dst = null;

            // When

        long asyncId = 67890L;

        // When
        Smb2CancelRequest request = new Smb2CancelRequest(mockConfig, mid, asyncId);

        // Then
        assertEquals(SMB2_CANCEL, request.getCommand());
        assertEquals(mid, request.getMid());
        assertEquals(asyncId, request.getAsyncId());
        assertTrue((request.getFlags() & SMB2_FLAGS_ASYNC_COMMAND) != 0, "Async flag should be set when asyncId is non-zero");
    }

    @Test

            when(config.getPid()).thenReturn(12345);

            // When
            Constructor<SmbComNtCancel> constructor = SmbComNtCancel.class.getDeclaredConstructor(Configuration.class, int.class);
            constructor.setAccessible(true);
            SmbComNtCancel cancel = constructor.newInstance(config, mid);

            // Then
            assertNotNull(cancel);

    void testConstructorInitialization() {
        // Given
        int offset = 0;
        int length = TEST_DATA.length;

        // When
        transCallNamedPipe = new TransCallNamedPipe(mockConfig, TEST_PIPE_NAME, TEST_DATA, offset, length);

        // Then
        assertNotNull(transCallNamedPipe);
        assertEquals(SmbComTransaction.SMB_COM_TRANSACTION, transCallNamedPipe.getCommand());

    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

            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

        when(mockConfig.isDfsDisabled()).thenReturn(false);
        when(mockConfig.isEncryptionEnabled()).thenReturn(true);
        when(mockConfig.getMinimumVersion()).thenReturn(DialectVersion.SMB202);
        when(mockConfig.getMaximumVersion()).thenReturn(DialectVersion.SMB311);
        when(mockConfig.getMachineId()).thenReturn(testMachineId);
        when(mockConfig.getRandom()).thenReturn(mockRandom);

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

            // Given
            when(mockResource.children()).thenReturn(mockIterator);
            when(mockResource.children("*.txt")).thenReturn(mockIterator);
            when(mockResource.children(mockNameFilter)).thenReturn(mockIterator);
            when(mockResource.children(mockResourceFilter)).thenReturn(mockIterator);

            // When

        // Setup default mock behaviors
        when(ctx.getConfig()).thenReturn(config);
        when(ctx.getNameServiceClient()).thenReturn(nameSvc);
        when(ctx.getCredentials()).thenReturn(creds);
        when(ctx.getTransportPool()).thenReturn(pool);

        // Default config values
        when(config.getLocalAddr()).thenReturn(null);
        when(config.getLocalPort()).thenReturn(0);
        when(config.getSessionLimit()).thenReturn(10);