Metacat

package edu.ucsb.nceas.metacat.index;

import static org.junit.Assert.assertEquals;

import static org.junit.Assert.assertTrue;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.io.InputStream;

import java.util.List;

import javax.xml.parsers.ParserConfigurationException;

import org.apache.commons.io.FileUtils;

import org.apache.solr.client.solrj.SolrServer;

import org.apache.solr.client.solrj.SolrServerException;

import org.apache.solr.client.solrj.embedded.EmbeddedSolrServer;

import org.apache.solr.client.solrj.response.QueryResponse;

import org.apache.solr.common.SolrDocument;

import org.apache.solr.common.SolrDocumentList;

import org.apache.solr.common.params.SolrParams;

import org.apache.solr.core.CoreContainer;

import org.apache.solr.servlet.SolrRequestParsers;

import org.dataone.configuration.Settings;

import org.dataone.service.types.v1.SystemMetadata;

import org.dataone.service.util.TypeMarshaller;

import org.junit.Test;

import org.xml.sax.SAXException;

public class SolrIndexTest  {

    private static final String SYSTEMMETAFILEPATH = "src/test/resources/eml-system-meta-example.xml";

    private static final String EMLFILEPATH = "src/test/resources/eml-example.xml";

    private static final String SYSTEMMETAUPDATEFILEPATH = "src/test/resources/eml-updating-system-meta-example.xml";

    private static final String EMLUPDATEFILEPATH = "src/test/resources/eml-updating-example.xml";

    private static final String id = "urn:uuid:606a19dd-b531-4bf4-b5a5-6d06c3d39098";

    private static final String newId = "urn:uuid:606a19dd-b531-4bf4-b5a5-6d06c3d39099";

    private static final String SOLRHOME = "solr-home";

    private static final String SOLRTESTHOMEPATH = "target/"+SOLRHOME;

    /**

     * Test building index for an insert.

     */

    @Test

    public void testInsert() throws Exception {

        SolrIndex solrIndex = generateSolrIndex();

       //InputStream systemInputStream = new FileInputStream(new File(SYSTEMMETAFILEPATH));

       SystemMetadata systemMetadata = TypeMarshaller.unmarshalTypeFromFile(SystemMetadata.class, SYSTEMMETAFILEPATH);

       InputStream emlInputStream = new FileInputStream(new File(EMLFILEPATH));    

       solrIndex.insert(id, systemMetadata, emlInputStream);

       String result = doQuery(solrIndex.getSolrServer());

       assertTrue(result.contains("version1"));

    }

    /**

     * Test building index for an insert.

     */

    @Test

    public void testUpdate() throws Exception {

        SolrIndex solrIndex = generateSolrIndex();

       //InputStream systemInputStream = new FileInputStream(new File(SYSTEMMETAFILEPATH));

       SystemMetadata systemMetadata = TypeMarshaller.unmarshalTypeFromFile(SystemMetadata.class, SYSTEMMETAUPDATEFILEPATH);

       InputStream emlInputStream = new FileInputStream(new File(EMLUPDATEFILEPATH));    

       solrIndex.update(newId, systemMetadata, emlInputStream);

       String result = doQuery(solrIndex.getSolrServer());

       assertTrue(result.contains("version2"));

    }

    /**

     * Test building index for an insert.

     */

    //@Test

    /*public void testDelete() throws Exception {

       SolrIndex solrIndex = generateSolrIndex();

       //solrIndex.remove(newId);

       //String result = doQuery(solrIndex.getSolrServer());

       //assertTrue(!result.contains("version1"));

       //assertTrue(!result.contains("version2"));

    }*/

    /*

     * Do query

     */

    private String doQuery(SolrServer server)

                    throws SolrServerException {

                StringBuffer request = new StringBuffer();

                request.append("q=" + "*:*");

                SolrParams solrParams = SolrRequestParsers.parseQueryString(request

                        .toString());

                QueryResponse reponse = server.query(solrParams);

                System.out.println("**************************************************************************");

                System.out.println("The query result:\n");

                System.out.println(reponse.toString());

                System.out.println("**************************************************************************");

                return reponse.toString();

    }

    /*

     * Print out the QueryResponse

     */

    /*private void print(QueryResponse response) {

        SolrDocumentList docs = response.getResults();

        if (docs != null) {

            System.out.println(docs.getNumFound() + " documents found, "

                    + docs.size() + " returned : ");

            for (int i = 0; i < docs.size(); i++) {

                SolrDocument doc = docs.get(i);

                System.out.println("\t" + doc.toString());

            }

        }

    }*/

    private SolrIndex generateSolrIndex() throws Exception {

        Settings.getConfiguration().setProperty("dataone.hazelcast.configFilePath", ApplicationControllerTest.pathToHazelcastFile);

        ApplicationController controller = new ApplicationController();

        List<SolrIndex> list = controller.getSolrIndexes();

        SolrIndex[] solrIndexesarray = list.toArray(new SolrIndex[list.size()]);

        SolrIndex index = solrIndexesarray[0];

        index.setSolrServer(generateSolrServer());

        return index;

    }

    /*

     * Generate a test solr server

     */

    private SolrServer generateSolrServer() throws IOException, ParserConfigurationException, SAXException {

        createSolrHome();

        System.setProperty(SolrIndex.SOLRHOME, SOLRTESTHOMEPATH);

        //System.out.println("====="+"/Users/tao/projects/metacat-index/"+SOLRTESTHOMEPATH);

        CoreContainer.Initializer init = new CoreContainer.Initializer();

        CoreContainer c = init.initialize();

        SolrServer solrServer = new EmbeddedSolrServer(c, "collection1");

        return solrServer;

    }

    /*

     * Create a solr home directory

     */

    private void createSolrHome() throws IOException {

        File solrHomeTestDir =  new File(SOLRTESTHOMEPATH);

        solrHomeTestDir.mkdir();

        File solrData = new File(SOLRTESTHOMEPATH+"/data");

        solrData.mkdir();

        File solr_home_source_dir = new File("src/main/resources/"+SOLRHOME);

        FileUtils.copyDirectory(solr_home_source_dir, solrHomeTestDir);

    }

}