com.rbmhtechnology.vind.test.TestSearchServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vind-testsuite Show documentation
There is a newer version: 3.2.0
package com.rbmhtechnology.vind.test;

import com.rbmhtechnology.vind.SearchServerException;
import com.rbmhtechnology.vind.api.SearchServer;
import com.rbmhtechnology.vind.configure.SearchConfiguration;
import org.junit.rules.ExternalResource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 *
 */
public class TestSearchServer extends ExternalResource {

    private Logger log = LoggerFactory.getLogger(getClass());

    private ServerType serverType = ServerType.Embedded;

    private SearchServer searchServer;

    @Override
    protected void before() throws Throwable {
        super.before();

        serverType.prepareConfig();

        searchServer = SearchServer.getInstance();
    }

    @Override
    protected void after() {
        try {
            searchServer.close();
        } catch (SearchServerException e) {
            log.error("Error closing SearchServer: {}", e.getMessage(), e);
        } finally {
            System.getProperties().remove("runtimeLib");
            super.after();
        }

    }

    public SearchServer getSearchServer() {
        return searchServer;
    }

    private enum ServerType {

        Embedded("com.rbmhtechnology.vind.solr.backend.EmbeddedSolrServerProvider", null, null, false),

        //setup:
        // * docker run --name vind-solr-2.1.0 -p 8983:8983 redlinkgmbh/vind-solr-server:2.1.0
        RemoteStandalone("com.rbmhtechnology.vind.solr.backend.RemoteSolrServerProvider", "http://localhost:8983/solr", "vind", false),

        //setup:
        // * ./bin/solr start c
        // * download exec jar for collection mgtm tool, e.g. http://central.maven.org/maven2/com/rbmhtechnology/vind/collection-managment-tool/2.1.0/
        // * java -jar collection-managment-tool-2.1.0-exec.jar -cc vind -from com.rbmhtechnology.vind:backend-solr:2.1.0 --in localhost:9983
        RemoteCloud("com.rbmhtechnology.vind.solr.backend.RemoteSolrServerProvider", "localhost:9983", "vind", true);

        private String provider;
        private String host;
        private String collection;
        private boolean cloud;

        ServerType(String provider, String host, String collection, boolean cloud) {
            this.provider = provider;
            this.host = host;
            this.collection = collection;
            this.cloud = cloud;
        }

        void prepareConfig() {
            SearchConfiguration.set(SearchConfiguration.SERVER_PROVIDER, provider);
            if(host != null) SearchConfiguration.set(SearchConfiguration.SERVER_HOST, host);
            if(collection != null) SearchConfiguration.set(SearchConfiguration.SERVER_COLLECTION, collection);
            SearchConfiguration.set(SearchConfiguration.SERVER_SOLR_CLOUD, cloud);
            System.setProperty("runtimeLib", String.valueOf(cloud));
        }
    }

}