[Exist-commits] eXist-1.0/src/org/exist/xquery/test FtQueryTest.java,NONE,1.1

[Wolfgang M. M. <wol...@us...>]

Update of /cvsroot/exist/eXist-1.0/src/org/exist/xquery/test
In directory sc8-pr-cvs1.sourceforge.net:/tmp/cvs-serv20546/src/org/exist/xquery/test

Added Files:
	FtQueryTest.java 
Log Message:
Full-text index now stores the exact position of a term within the indexed text. Post-processing query results becomes simpler as the query engine knows the coordinates of a match.

--- NEW FILE: FtQueryTest.java ---
/*
 *  eXist Open Source Native XML Database
 *  Copyright (C) 2001-04 The eXist Project
 *  http://exist-db.org
 *  
 *  This program is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public License
 *  as published by the Free Software Foundation; either version 2
 *  of the License, or (at your option) any later version.
 *  
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 *  
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *  
 *  $Id$
 */
package org.exist.xquery.test;

import java.io.File;

import org.custommonkey.xmlunit.XMLTestCase;
import org.exist.xmldb.DatabaseInstanceManager;
import org.xmldb.api.DatabaseManager;
import org.xmldb.api.base.Collection;
import org.xmldb.api.base.Database;
import org.xmldb.api.base.ResourceSet;
import org.xmldb.api.base.XMLDBException;
import org.xmldb.api.modules.CollectionManagementService;
import org.xmldb.api.modules.XMLResource;
import org.xmldb.api.modules.XQueryService;

public class FtQueryTest extends XMLTestCase {

    private final static String FILES[] = { "hamlet.xml", "macbeth.xml", "r_and_j.xml" };
    
    private final static File SHAKES_DIR = new File("samples" + File.separator + "shakespeare");
    private final static File MODS_DIR = new File("samples" + File.separator + "mods");
    
    private Database database;
    private Collection testCollection;
    
    public void testFtOperators() throws Exception {
        XQueryService service = (XQueryService)
            testCollection.getService("XQueryService", "1.0");
        ResourceSet result = service.query("//SPEECH[LINE &= 'love']");
        assertEquals(result.getSize(), 160);
        result = service.query("//SPEECH[LINE &= 'thou']");
        assertEquals(result.getSize(), 290);
        result = service.query("//SPEECH[LINE &= 'thou']");
        assertEquals(result.getSize(), 290);
        result = service.query("//SPEECH[LINE &= 'fenny snake']/LINE[1]");
        assertEquals(result.getSize(), 1);
        assertXMLEqual(result.getResource(0).getContent().toString(), "<LINE>Fillet of a fenny snake,</LINE>");
        result = service.query("//SPEECH[LINE &= 'god*']");
        assertEquals(result.getSize(), 79);
        result = service.query("//SPEECH[LINE &= 'god in heaven']");
        assertEquals(result.getSize(), 2);
        result = service.query("//SPEECH[SPEAKER &= 'Nurse']");
        assertEquals(result.getSize(), 90);
        result = service.query("declare namespace mods='http://www.loc.gov/mods/v3'; //mods:titleInfo[mods:title &= 'self*']");
        assertEquals(result.getSize(), 2);
        result = service.query("declare namespace mods='http://www.loc.gov/mods/v3'; //mods:titleInfo[mods:title &= 'self employed']");
        assertEquals(result.getSize(), 1);
        result = service.query("declare namespace mods='http://www.loc.gov/mods/v3'; //mods:titleInfo[match-all(mods:title, '.*ploy.*')]");
        assertEquals(result.getSize(), 3);
    }
    
    public void testFtScan() throws Exception {
        String queryBody =
            "declare namespace f=\'http://exist-db.org/xquery/test\';\n" + 
            "declare namespace mods='http://www.loc.gov/mods/v3';\n" + 
            "import module namespace t=\'http://exist-db.org/xquery/text\';\n" + 
            "\n" + 
            "declare function f:term-callback($term as xs:string, $data as xs:int+)\n" + 
            "as element()+ {\n" + 
            "    <item>\n" + 
            "        <term>{$term}</term>\n" + 
            "        <frequency>{$data[1]}</frequency>\n" + 
            "    </item>\n" + 
            "};\n" + 
            "\n";
        
        XQueryService service = (XQueryService)
            testCollection.getService("XQueryService", "1.0");
        String query = queryBody + "t:index-terms(collection(\'/db\'), \'is\', util:function(\'f:term-callback\', 2), 1000)";
        ResourceSet result = service.query(query);
        assertEquals(result.getSize(), 7);
        
        query = queryBody + "t:index-terms(collection(\'/db\')//LINE, \'is\', util:function(\'f:term-callback\', 2), 1000)";
        result = service.query(query);
        assertEquals(result.getSize(), 6);
        
        query = queryBody + "t:index-terms(collection(\'/db\')//mods:title, \'s\', util:function(\'f:term-callback\', 2), 1000)";
        result = service.query(query);
        assertEquals(result.getSize(), 20);
    }
    
    public void testFtUpdate() throws Exception {
        XQueryService service = (XQueryService)
            testCollection.getService("XQueryService", "1.0");
        service.query(
                "update insert <SPEAKER>First Witch</SPEAKER> preceding //SPEECH[LINE &= 'fenny snake']/SPEAKER"
        );
        ResourceSet result = service.query("//SPEECH[LINE &= 'fenny snake']/SPEAKER");
        assertEquals(2, result.getSize());
        result = service.query("//SPEECH[LINE &= 'fenny snake' and SPEAKER &= 'first']");
        assertEquals(1, result.getSize());
        
        service.query(
                "update delete //SPEECH[LINE &= 'fenny snake']/SPEAKER[2]"
        );
        result = service.query("//SPEECH[LINE &= 'fenny snake' and SPEAKER &= 'first']");
        assertEquals(1, result.getSize());
    }
    
    protected void setUp() {
        try {
            // initialize driver
            Class cl = Class.forName("org.exist.xmldb.DatabaseImpl");
            database = (Database) cl.newInstance();
            database.setProperty("create-database", "true");
            DatabaseManager.registerDatabase(database);
            
            Collection root =
                DatabaseManager.getCollection(
                    "xmldb:exist:///db",
                    "admin",
                    null);
            CollectionManagementService service =
                (CollectionManagementService) root.getService(
                    "CollectionManagementService",
                    "1.0");
            testCollection = service.createCollection("test");
            assertNotNull(testCollection);

            for (int i = 0; i < FILES.length; i++) {
                XMLResource doc =
                    (XMLResource) testCollection.createResource(
                            FILES[i], "XMLResource" );
                doc.setContent(new File(SHAKES_DIR, FILES[i]));
                testCollection.storeResource(doc);
            }
            
            File modsFiles[] = MODS_DIR.listFiles();
            for (int i = 0; i < modsFiles.length; i++) {
                if (modsFiles[i].isFile()) {
                    XMLResource doc =
                        (XMLResource) testCollection.createResource(
                                modsFiles[i].getName(), "XMLResource" );
                    doc.setContent(modsFiles[i]);
                    testCollection.storeResource(doc);
                }
            }
        } catch (ClassNotFoundException e) {
        } catch (InstantiationException e) {
        } catch (IllegalAccessException e) {
        } catch (XMLDBException e) {
            e.printStackTrace();
        }
    }

    /*
     * @see TestCase#tearDown()
     */
    protected void tearDown() throws Exception {
        Collection root =
            DatabaseManager.getCollection(
                "xmldb:exist:///db",
                "admin",
                null);
        CollectionManagementService service =
            (CollectionManagementService) root.getService(
                "CollectionManagementService",
                "1.0");
        service.removeCollection("test");
        
        DatabaseManager.deregisterDatabase(database);
        DatabaseInstanceManager dim =
            (DatabaseInstanceManager) testCollection.getService(
                "DatabaseInstanceManager", "1.0");
        dim.shutdown();
        System.out.println("tearDown PASSED");
    }
}