Metacat

package edu.ucsb.nceas.metacat.index;

import static org.junit.Assert.assertTrue;

import java.io.File;

import java.io.FileInputStream;

import java.io.Writer;

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

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

import org.apache.solr.core.SolrCore;

import org.apache.solr.request.LocalSolrQueryRequest;

import org.apache.solr.response.SolrQueryResponse;

import org.apache.solr.response.XMLResponseWriter;

import org.dataone.service.util.TypeMarshaller;

    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 EMLUPDATEFILEPATH = "src/test/resources/eml-updating-example.xml";

            solrIndex  = generateSolrIndex();

        ApplicationController controller = new ApplicationController();

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

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

        SolrIndex index = solrIndexesarray[0];

        SolrServer solrServer = SolrServerFactory.createSolrServer();

        index.setSolrServer(solrServer);

        return index;

    }

     * Test building index for an insert.

     */

    @Test

    public void testInsert() throws Exception {

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

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

       solrIndex.insert(id, systemMetadata, emlInputStream);

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

     * Test building index for an insert.

     */

    @Test

       //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.

     */

    public void testDelete() throws Exception {

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

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

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

    }

    /*

     * Do query

     */

                StringBuffer request = new StringBuffer();

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

                SolrParams solrParams = SolrRequestParsers.parseQueryString(request

                        .toString());

                QueryResponse reponse = server.query(solrParams);

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

    /*

        XMLResponseWriter xmlWriter = new XMLResponseWriter();

        Writer w = new StringWriter();

        SolrQueryResponse sResponse = new SolrQueryResponse();

        sResponse.setAllValues(response.getResponse());

        try {

            SolrCore core = null;

            CoreContainer container = SolrServerFactory.getCoreContainer();

            core = container.getCore("collection1");

            xmlWriter.write(w, new LocalSolrQueryRequest(core, request), sResponse);

        } catch (IOException e) {

            throw new RuntimeException("Unable to convert Solr response into XML", e);

    }