org.ambraproject.service.search.EmbeddedSolrServerFactoryTest.java Source code

Introduction

Here is the source code for org.ambraproject.service.search.EmbeddedSolrServerFactoryTest.java

Source

/*
 * $HeadURL$
 * $Id$
 * Copyright (c) 2006-2012 by Public Library of Science http://plos.org http://ambraproject.org
 * 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.0Unless 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 org.ambraproject.service.search;

import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocumentList;
import org.apache.solr.common.params.SolrParams;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
import org.ambraproject.testutils.EmbeddedSolrServerFactory;

import java.util.HashMap;
import java.util.Map;

import static org.testng.Assert.*;

/**
 * Basic test for {@link EmbeddedSolrServerFactory} . The working directory for this test needs to be set to the ambra
 * webapp home, so that the solr server factory can find the config files
 *
 * @author Alex Kudlick Date: 5/13/11
 *         <p/>
 *         org.ambraproject.ambra
 */
public class EmbeddedSolrServerFactoryTest {

    private EmbeddedSolrServerFactory solrServerFactory;

    @Test(priority = -1)
    public void testConstructor() throws Exception {
        solrServerFactory = new EmbeddedSolrServerFactory();
    }

    @Test
    public void testGetServer() throws SolrServerException {
        SolrServer server = solrServerFactory.getServer();
        assertNotNull(server, "created null server");
    }

    @Test(dependsOnMethods = { "testGetServer" })
    public void testBadQuery() throws SolrServerException {
        SolrServer server = solrServerFactory.getServer();
        SolrParams params = new SolrQuery("text that is not found");
        QueryResponse response = server.query(params);
        assertEquals(0L, response.getResults().getNumFound());
    }

    @DataProvider(name = "documentMap")
    public Object[][] getDocument() throws Exception {
        Map<String, String[]> document = new HashMap<String, String[]>();
        document.put("id", new String[] { "test" });
        document.put("title", new String[] { "test_title" });
        document.put("alternate_title", new String[] { "alt_title_1", "alt_title_2" });

        return new Object[][] { { document } };
    }

    @Test(dataProvider = "documentMap", dependsOnMethods = { "testGetServer" })
    public void testAddDocument(Map<String, String[]> document) throws Exception {
        solrServerFactory.addDocument(document);
    }

    @Test(dependsOnMethods = "testAddDocument")
    public void testBasicQuery() throws SolrServerException {
        SolrServer server = solrServerFactory.getServer();
        SolrParams params = new SolrQuery("*:*");
        SolrDocumentList results = server.query(params).getResults();
        assertTrue(results.getNumFound() > 0, "didn't return any results");
    }

    @Test(dependsOnMethods = "testBasicQuery", dataProvider = "documentMap")
    public void testIdQuery(Map<String, String[]> document) throws SolrServerException {
        String id = document.get("id")[0];
        SolrServer server = solrServerFactory.getServer();
        SolrParams params = new SolrQuery("id:" + id);
        SolrDocumentList results = server.query(params).getResults();
        assertEquals(results.getNumFound(), 1, "didn't find article by id");
    }

    @Test(dependsOnMethods = "testAddDocument", dataProvider = "documentMap")
    public void testMultiValuedField(Map<String, String[]> document) throws Exception {
        String value1 = document.get("alternate_title")[0];
        String value2 = document.get("alternate_title")[1];

        Long numFound = solrServerFactory.getServer().query(new SolrQuery("alternate_title:" + value1)).getResults()
                .getNumFound();
        assertTrue(numFound > 0l, "query didn't work on multivalued field");

        numFound = solrServerFactory.getServer().query(new SolrQuery("alternate_title:" + value2)).getResults()
                .getNumFound();
        assertTrue(numFound > 0l, "query didn't work on multivalued field");
    }

    @AfterClass
    public void tearDown() throws Exception {
        solrServerFactory.tearDown();
    }
}