edu.unc.lib.dl.data.ingest.solr.action.BaseEmbeddedSolrTest.java Source code

Introduction

Here is the source code for edu.unc.lib.dl.data.ingest.solr.action.BaseEmbeddedSolrTest.java

Source

/**
 * Copyright 2008 The University of North Carolina at Chapel Hill
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package edu.unc.lib.dl.data.ingest.solr.action;

import java.io.File;

import org.apache.commons.io.FileUtils;
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.SolrDocumentList;
import org.apache.solr.common.params.ModifiableSolrParams;
import org.apache.solr.core.CoreContainer;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import edu.unc.lib.dl.data.ingest.solr.indexing.SolrUpdateDriver;

public class BaseEmbeddedSolrTest extends Assert {
    private static final Logger log = LoggerFactory.getLogger(BaseEmbeddedSolrTest.class);

    protected EmbeddedSolrServer server;

    protected CoreContainer container;

    protected SolrUpdateDriver driver;

    @Before
    public void setUp() throws Exception {

        File home = new File("src/test/resources/config");
        File configFile = new File(home, "solr.xml");

        System.setProperty("solr.data.dir", "src/test/resources/config/data/");
        container = new CoreContainer("src/test/resources/config", configFile);

        server = new EmbeddedSolrServer(container, "access-master");

        driver = new SolrUpdateDriver();
        driver.setSolrServer(server);
        driver.setUpdateSolrServer(server);
    }

    protected SolrDocumentList getDocumentList(String query, String fieldList) throws SolrServerException {
        ModifiableSolrParams params = new ModifiableSolrParams();
        params.set("q", query);
        params.set("fl", fieldList);
        QueryResponse qResp = server.query(params);
        return qResp.getResults();
    }

    protected SolrDocumentList getDocumentList() throws SolrServerException {
        return getDocumentList("*:*", "id,resourceType,_version_");
    }

    @After
    public void tearDown() throws Exception {
        server.shutdown();
        log.debug("Cleaning up data directory");
        File dataDir = new File("src/test/resources/config/data");
        FileUtils.deleteDirectory(dataDir);
    }
}