}

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

        assertEquals(mid, response.getMid());

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

            testBlock = new TestServerMessageBlock(mockConfig);
        }

        @Test
        @DisplayName("Test command property")
        void testCommandProperty() {
            testBlock.setCommand(0x42);
            assertEquals(0x42, testBlock.getCommand());
        }

        @Test
        @DisplayName("Test flags property")
        void testFlagsProperty() {

        void testToStringUnknownCommand() {
            testMessage.setCommand(0xFF);
            String result = testMessage.toString();

            assertTrue(result.contains("command=UNKNOWN"));
        }

        @Test
        @DisplayName("Should include all fields in toString")
        void testToStringFields() {
            testMessage.setCommand(ServerMessageBlock2.SMB2_CREATE);
            testMessage.setMid(12345L);

     * Gets the SMB command.
     *
     * @return the command
     */
    int getCommand();

    /**
     * Sets the SMB command.
     *
     * @param command the command to set
     */
    void setCommand(int command);

    /**
     * Sets the user ID.
     *
     * @param uid the user ID to set
     */
    void setUid(int uid);

    /**
     * Sets whether extended security is enabled.

            protected int readHeaderWireFormat(byte[] buffer, int bufferIndex) {
                return 33;
            }
        };
        testBlock.configureForTest(true, 2);
        testBlock.setCommand((byte) ServerMessageBlock.SMB_COM_NT_CREATE_ANDX);

        byte[] buffer = new byte[1024];
        buffer[33] = 34; // wordCount for extended response
        buffer[34] = (byte) 0xFF; // andxCommand

     */
    public Trans2QueryFSInformationResponse(final Configuration config, final int informationLevel) {
        super(config);
        this.informationLevel = informationLevel;
        this.setCommand(SMB_COM_TRANSACTION2);
        this.setSubCommand(SmbComTransaction.TRANS2_QUERY_FS_INFORMATION);
    }

    /**
     * Gets the information level that was queried.
     *
     * @return the informationLevel

            this.mid = mid;
        }

        @Override
        public int getCommand() {
            return command;
        }

        @Override
        public void setCommand(int command) {
            this.command = command;
        }

        @Override
        public void setUid(int uid) {
            // No-op for test
        }

        @Override

    void testWriteParameterWordsWireFormatSecondary() {
        byte[] dst = new byte[256];

        // Change command to secondary
        transaction.setCommand(ServerMessageBlock.SMB_COM_NT_TRANSACT_SECONDARY);

        // Set up transaction parameters
        transaction.setTotalParameterCount(100);
        transaction.setTotalDataCount(200);

            this.mid = mid;
        }

        @Override
        public int getCommand() {
            return this.command;
        }

        @Override
        public void setCommand(int command) {
            this.command = command;
        }

        @Override
        public void setUid(int uid) {
            this.uid = uid;
        }

        @Override