assertEquals(Smb2QueryDirectoryRequest.FILE_FULL_DIRECTORY_INFO, buffer[Smb2Constants.SMB2_HEADER_LENGTH + 2]);
    }

    @Test
    @DisplayName("Test setQueryFlags method")
    void testSetQueryFlags() {
        request = new Smb2QueryDirectoryRequest(mockConfig);
        byte flags = Smb2QueryDirectoryRequest.SMB2_RESTART_SCANS | Smb2QueryDirectoryRequest.SMB2_RETURN_SINGLE_ENTRY;

        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    @Test
    @DisplayName("Test setQueryFlags method")
    void testSetQueryFlags() {
        request = new Smb2QueryInfoRequest(mockConfig);
        int testQueryFlags = 0xABCDEF00;

        request.setQueryFlags(testQueryFlags);

        // Verify query flags were set