}
    }

    @Test
    @DisplayName("Test setFileInfoClass method")
    void testSetFileInfoClass() {
        request = new Smb2QueryInfoRequest(mockConfig);
        byte testFileInfoClass = (byte) 0x10;

        request.setFileInfoClass(testFileInfoClass);

        // Verify file info class was set and info type was set to SMB2_0_INFO_FILE
        try {

    @Test
    @DisplayName("Multiple reads update offset correctly")
    void testMultipleReadsUpdateOffset() {
        byte[] buffer1 = new byte[] { 0x10, 0x00, 0x00, 0x00 }; // 16
        byte[] buffer2 = new byte[] { 0x20, 0x00, 0x00, 0x00 }; // 32

        response.readParameterWordsWireFormat(buffer1, 0);

        assertEquals((byte) 0x03, SmbComTransaction.TRANS2_QUERY_FS_INFORMATION);
        assertEquals((byte) 0x05, SmbComTransaction.TRANS2_QUERY_PATH_INFORMATION);
        assertEquals((byte) 0x10, SmbComTransaction.TRANS2_GET_DFS_REFERRAL);
        assertEquals((byte) 0x08, SmbComTransaction.TRANS2_SET_FILE_INFORMATION);

        // Test NET subcommands
        assertEquals(0x0000, SmbComTransaction.NET_SHARE_ENUM);

        return "4fc742e0-4a10-11cf-8273-00aa004ae673:3.0";
    }

    /**
     * DFS volume flavor indicating standalone DFS
     */
    public static final int DFS_VOLUME_FLAVOR_STANDALONE = 0x100;
    /**
     * DFS volume flavor indicating Active Directory blob storage
     */
    public static final int DFS_VOLUME_FLAVOR_AD_BLOB = 0x200;
    /**
     * DFS storage state indicating offline status
     */

                // OID length claims more than available
                Arguments.of("oid length exceeds buffer", new byte[] { 0x04, 0x01, 0x00, 0x10, 0x06 }, IllegalArgumentException.class),

                // Missing NAME_LEN (not enough bytes for 4-byte length)
                Arguments.of("missing NAME_LEN bytes", new byte[] { 0x04, 0x01, 0x00, (byte) der.length, der[0] },

        // Initialize test data
        for (int i = 0; i < 16; i++) {
            testMachineId[i] = (byte) (i + 1);
        }
        for (int i = 0; i < 32; i++) {
            testSalt[i] = (byte) (i + 0x10);
        }

        // Default configuration for SMB 3.1.1 with encryption
        when(mockConfig.isDfsDisabled()).thenReturn(false);
        when(mockConfig.isEncryptionEnabled()).thenReturn(true);

        assertEquals(0x02, Smb2QueryDirectoryRequest.SMB2_RETURN_SINGLE_ENTRY);
        assertEquals(0x04, Smb2QueryDirectoryRequest.SMB2_INDEX_SPECIFIED);
        assertEquals(0x10, Smb2QueryDirectoryRequest.SMB2_REOPEN);
    }

    @Test
    @DisplayName("Test with Unicode filename")
    void testUnicodeFileName() {
        request = new Smb2QueryDirectoryRequest(mockConfig);

        return Stream.of(Arguments.of(new byte[] {}, ""), Arguments.of(new byte[] { (byte) 0x00 }, "00"),
                Arguments.of(new byte[] { (byte) 0x0F }, "0f"), Arguments.of(new byte[] { (byte) 0x10 }, "10"),
                Arguments.of(new byte[] { (byte) 0xAB }, "ab"),
                Arguments.of(new byte[] { (byte) 0x7F, (byte) 0x80, (byte) 0xFF }, "7f80ff"));
    }

        for (int i = 0; i < 16; i++) {
            expectedData[i] = (byte) (0xA0 + i);
            buffer[16 + i] = expectedData[i];
        }

        // Set header values
        SMBUtil.writeInt4(0x100, buffer, 0);
        SMBUtil.writeInt4(0x200, buffer, 4);
        SMBUtil.writeInt4(0x300, buffer, 8);
        SMBUtil.writeInt4(0x400, buffer, 12);

        // Decode
        response.decode(buffer, 0, 32);

    static final byte TRANS2_QUERY_FS_INFORMATION = (byte) 0x03;
    static final byte TRANS2_QUERY_PATH_INFORMATION = (byte) 0x05;
    static final byte TRANS2_GET_DFS_REFERRAL = (byte) 0x10;
    static final byte TRANS2_SET_FILE_INFORMATION = (byte) 0x08;

    static final int NET_SHARE_ENUM = 0x0000;
    static final int NET_SERVER_ENUM2 = 0x0068;
    static final int NET_SERVER_ENUM3 = 0x00D7;