}

    @Test
    @DisplayName("Test setCommand and getCommand methods")
    void testSetAndGetCommand() {
        // Given
        int expectedCommand = 0x73;
        doNothing().when(messageBlock).setCommand(expectedCommand);
        when(messageBlock.getCommand()).thenReturn(expectedCommand);

        // When
        messageBlock.setCommand(expectedCommand);

    }

    @Override
    protected int writeParameterWordsWireFormat(final byte[] dst, int dstIndex) {
        final int start = dstIndex;

        if (this.getCommand() != SMB_COM_NT_TRANSACT_SECONDARY) {
            dst[dstIndex++] = this.maxSetupCount;
        } else {
            dst[dstIndex++] = (byte) 0x00; // Reserved
        }
        dst[dstIndex] = (byte) 0x00; // Reserved

            assertEquals(0x25, testBlock.getCommand());
            assertNull(testBlock.getPath());
        }

        @Test
        @DisplayName("Test constructor with config, command, and path")
        void testConstructorWithPath() {
            testBlock = new TestServerMessageBlock(mockConfig, (byte) 0x25, "\\test\\path");

            assertEquals(0x25, testBlock.getCommand());

        setupMockConfig();
        Smb2IoctlRequest request = new Smb2IoctlRequest(mockConfig, TEST_CONTROL_CODE);

        assertNotNull(request);
        assertEquals(0x0B, request.getCommand()); // SMB2_IOCTL command
        verify(mockConfig).getTransactionBufferSize();
    }

    @Test
    @DisplayName("Test constructor with config, control code and file ID")
    void testConstructorWithFileId() {

        when(mockConfig.isUseBatching()).thenReturn(true);
    }

    @Test
    @DisplayName("Test constructor with andx command")
    void testConstructorWithAndx() {
        when(mockAndxCommand.getCommand()).thenReturn(0x42);

        testBlock = new TestAndXServerMessageBlock(mockConfig, (byte) 0x25, mockAndxCommand);

        assertEquals(mockAndxCommand, testBlock.getAndx());

        SmbComTreeDisconnect instance2 = new SmbComTreeDisconnect(mockConfig);

        // Then
        assertNotSame(instance1, instance2);
        assertEquals(instance1.getCommand(), instance2.getCommand());
        assertNotEquals(instance1.toString(), instance2.toString()); // Different object references in toString
    }

    /**
     * Test thread safety of read/write operations
     */

        // Test that the constructor properly initializes the object
        assertNotNull(response);
        // The command is not set in the constructor, defaults to 0
        assertEquals(0, response.getCommand());
        // The subcommand should be TRANS2_SET_FILE_INFORMATION (0x08)
        assertEquals(SmbComTransaction.TRANS2_SET_FILE_INFORMATION, response.getSubCommand());
    }

    @Test

            this.dataCount = Math.min(this.totalDataCount, available);
        } else {
            if (this.getCommand() != SMB_COM_NT_TRANSACT) {
                this.setCommand(SMB_COM_TRANSACTION_SECONDARY);
            } else {
                this.setCommand(SMB_COM_NT_TRANSACT_SECONDARY);
            }
            // totalParameterCount and totalDataCount are set ok from primary

        // Test command methods
        int command = 0x25;
        doNothing().when(response).setCommand(command);
        when(response.getCommand()).thenReturn(command);
        response.setCommand(command);
        assertEquals(command, response.getCommand());

        // Test other setter methods
        doNothing().when(response).setUid(1000);
        response.setUid(1000);

        assertEquals(SmbComTransaction.SMB_COM_TRANSACTION, resp1.getCommand());

        // Test with NT_TRANSACT command
        TestSmbComTransactionResponse resp2 =
                new TestSmbComTransactionResponse(mockConfig, SmbComTransaction.SMB_COM_NT_TRANSACT, (byte) 0x02);
        assertEquals(SmbComTransaction.SMB_COM_NT_TRANSACT, resp2.getCommand());
    }

    @Test