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
        Field queryFlagsField;
        try {

    }

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

        request.setQueryFlags(flags);

        // Verify by writing to buffer and checking the queryFlags position

    }

    /**
     * Sets the query flags for the information request
     *
     * @param queryFlags
     *            the queryFlags to set
     */
    public final void setQueryFlags(final int queryFlags) {
        this.queryFlags = queryFlags;
    }

    /**
     * {@inheritDoc}
     *
     * @see jcifs.internal.smb2.ServerMessageBlock2Request#createResponse(jcifs.CIFSContext,

        query.setFileName(this.getWildcard());
        query.setFileIndex(results[results.length - 1].getFileIndex());
        query.setQueryFlags(Smb2QueryDirectoryRequest.SMB2_INDEX_SPECIFIED);
        try {
            final Smb2QueryDirectoryResponse r = th.send(query);
            if (r.getStatus() == NtStatus.NT_STATUS_NO_MORE_FILES) {

    }

    /**
     * Sets the query flags for the directory enumeration
     *
     * @param queryFlags
     *            the queryFlags to set
     */
    public void setQueryFlags(final byte queryFlags) {
        this.queryFlags = queryFlags;
    }

    /**
     * Sets the file index from which to start the directory enumeration
     *
     * @param fileIndex