/*
 * Copyright 2011 Global Biodiversity Information Facility (GBIF)
 * 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 org.gbif.checklistbank.search;

import org.gbif.api.search.SearchRequest;
import org.gbif.api.search.SearchResponse;
import org.gbif.checklistbank.api.model.search.NameUsageFacetParameter;
import org.gbif.checklistbank.api.service.NameUsageSearchService;
import org.gbif.checklistbank.search.inject.SearchModule;
import org.gbif.checklistbank.search.model.NameUsageSolrSearchResult;

import com.google.inject.Guice;
import com.google.inject.Injector;
import com.google.inject.Key;
import com.google.inject.TypeLiteral;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;

/**
 * Integration tests using an embedded solr server with the mybatis squirrels test dataset as an prebuild test index.
 * To regenerate this index please @see SolrIndexBuildIT.
 */
public class NameUsageSearchTestIT {

  private static NameUsageSearchService<NameUsageSolrSearchResult> searchService;

  private static Injector injector;

  @BeforeClass
  public static void init() {
    injector = Guice.createInjector(new SearchModule());
    searchService = injector.getInstance(Key.get(new TypeLiteral<NameUsageSearchService<NameUsageSolrSearchResult>>() {
    }));
  }

  private void assertSearch(String q, NameUsageFacetParameter facet, String facetFilter, Long expectedCount,
    String expectedFacetCounts) {
    // build request
    SearchRequest searchRequest = new SearchRequest(0L, 10);
    searchRequest.setQ(q);
    if (facetFilter != null) {
      searchRequest.addFacetedParameter(facet, facetFilter);
    }
    if (facet != null) {
      searchRequest.addFacets(facet);
    }

    // query
    SearchResponse<NameUsageSolrSearchResult> response = searchService.search(searchRequest);

    // assert
    if (expectedCount != null) {
      Assert.assertEquals(expectedCount, response.getCount());
    }

    if (facet != null && expectedFacetCounts != null) {
      Assert.assertEquals(1, response.getFacets().size());
      Assert.assertEquals(expectedFacetCounts, response.getFacets().get(0).getCounts().size());
    }
    if (facet == null) {
      Assert.assertTrue(response.getFacets().isEmpty());
    }
  }

  private void assertSearch(String q, Long expectedCount) {
    assertSearch(q, null, null, expectedCount, null);
  }

  @Test
  public void testSearchScientificNameNoFacets() {
    assertSearch("Rodentia", 42l);
    assertSearch("Rodentia Bowdich, 1821", 2l);
  }

  @Test
  public void testSearchScientificNameWithRankFacet() {
    assertSearch("vulgaris", NameUsageFacetParameter.RANK, null, 10l, null);
    assertSearch("Sciurus vulgaris", NameUsageFacetParameter.RANK, null, 14l, null);
  }
}