/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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.apache.maven.index;
  
  import java.io.File;
  import java.util.Set;
  
  import org.apache.lucene.index.Term;
  import org.apache.lucene.search.WildcardQuery;
  import org.apache.lucene.store.ByteBuffersDirectory;
  import org.apache.lucene.store.Directory;
  import org.apache.maven.index.context.IndexingContext;
  import org.junit.Test;
  
  import static org.junit.Assert.assertEquals;
  
  public class Nexus3177HitLimitChecks extends AbstractNexusIndexerTest {
      protected File repo = new File(getBasedir(), "src/test/repo");
  
      protected Directory secondIndexDir = new ByteBuffersDirectory();
  
      protected IndexingContext secondContext;
  
      @Override
      protected void prepareNexusIndexer(NexusIndexer nexusIndexer) throws Exception {
          context = nexusIndexer.addIndexingContext(
                  "nexus-3177", "nexus-3177", repo, indexDir, null, null, DEFAULT_CREATORS);
  
          secondContext = nexusIndexer.addIndexingContext(
                  "nexus-3177b", "nexus-3177b", repo, secondIndexDir, null, null, DEFAULT_CREATORS);
  
          nexusIndexer.scan(context);
          nexusIndexer.scan(secondContext);
      }
  
      @Override
      protected void unprepareNexusIndexer(NexusIndexer nexusIndexer) throws Exception {
          super.unprepareNexusIndexer(nexusIndexer);
  
          nexusIndexer.removeIndexingContext(secondContext, false);
      }
  
      // ===================================================================
      // NOTE: This test, with testing search limits lost it's original meaning,
      // since version 4.1.0 there is no notion of hit limit.
      // See http://jira.codehaus.org/browse/MINDEXER-14
  
      // Hence, some of the tests, that can keep still original semantics were updated and left in place
      // but the two test explicitly testing LIMIT_EXCEEDED were just removed/commented out.
  
      @Test
      public void testHitLimitNotReachedSingleContext() throws Exception {
          WildcardQuery q = new WildcardQuery(new Term(ArtifactInfo.UINFO, "*testng*"));
  
          FlatSearchRequest request = new FlatSearchRequest(q);
          request.setCount(5);
          // request.setResultHitLimit( 5 );
          request.getContexts().add(context);
  
          FlatSearchResponse response = nexusIndexer.searchFlat(request);
          Set<ArtifactInfo> r = response.getResults();
          assertEquals(r.toString(), 4, r.size());
          assertEquals(r.toString(), 4, response.getTotalHitsCount());
      }
  
      @Test
      public void testHitLimitEqualSingleContext() throws Exception {
          WildcardQuery q = new WildcardQuery(new Term(ArtifactInfo.UINFO, "*testng*"));
  
          FlatSearchRequest request = new FlatSearchRequest(q);
          request.setCount(4);
          // request.setResultHitLimit( 4 );
          request.getContexts().add(context);
  
          FlatSearchResponse response = nexusIndexer.searchFlat(request);
          Set<ArtifactInfo> r = response.getResults();
          assertEquals(r.toString(), 4, r.size());
          assertEquals(r.toString(), 4, response.getTotalHitsCount());
      }
  
      // See NOTE above
      // public void testHitLimitExceededSingleContext()
      // throws Exception
     // {
     // WildcardQuery q = new WildcardQuery( new Term( ArtifactInfo.UINFO, "*testng*" ) );
     //
     // FlatSearchRequest request = new FlatSearchRequest( q );
     // request.setResultHitLimit( 3 );
     // request.getContexts().add( context );
     //
     // FlatSearchResponse response = nexusIndexer.searchFlat( request );
     // Set<ArtifactInfo> r = response.getResults();
     // assertEquals( r.toString(), 0, r.size() );
     // assertEquals( r.toString(), AbstractSearchResponse.LIMIT_EXCEEDED, response.getTotalHits() );
     // }
 
     @Test
     public void testHitLimitNotReachedMultipleContexts() throws Exception {
         WildcardQuery q = new WildcardQuery(new Term(ArtifactInfo.UINFO, "*testng*"));
 
         FlatSearchRequest request = new FlatSearchRequest(q);
         request.setCount(9);
         // request.setResultHitLimit( 9 );
         request.setArtifactInfoComparator(ArtifactInfo.REPOSITORY_VERSION_COMPARATOR);
         request.getContexts().add(context);
         request.getContexts().add(secondContext);
 
         FlatSearchResponse response = nexusIndexer.searchFlat(request);
         Set<ArtifactInfo> r = response.getResults();
         assertEquals(r.toString(), 8, r.size());
         assertEquals(r.toString(), 8, response.getTotalHitsCount());
     }
 
     @Test
     public void testHitLimitEqualMultipleContexts() throws Exception {
         WildcardQuery q = new WildcardQuery(new Term(ArtifactInfo.UINFO, "*testng*"));
 
         FlatSearchRequest request = new FlatSearchRequest(q);
         request.setCount(8);
         // request.setResultHitLimit( 8 );
         request.setArtifactInfoComparator(ArtifactInfo.REPOSITORY_VERSION_COMPARATOR);
         request.getContexts().add(context);
         request.getContexts().add(secondContext);
 
         FlatSearchResponse response = nexusIndexer.searchFlat(request);
         Set<ArtifactInfo> r = response.getResults();
         assertEquals(r.toString(), 8, r.size());
         assertEquals(r.toString(), 8, response.getTotalHitsCount());
     }
 
     // See NOTE above
     // public void testHitLimitExceededMultipleContexts()
     // throws Exception
     // {
     // WildcardQuery q = new WildcardQuery( new Term( ArtifactInfo.UINFO, "*testng*" ) );
     //
     // FlatSearchRequest request = new FlatSearchRequest( q );
     // request.setResultHitLimit( 7 );
     // request.setArtifactInfoComparator( ArtifactInfo.REPOSITORY_VERSION_COMPARATOR );
     // request.getContexts().add( context );
     // request.getContexts().add( secondContext );
     //
     // FlatSearchResponse response = nexusIndexer.searchFlat( request );
     // Set<ArtifactInfo> r = response.getResults();
     // assertEquals( r.toString(), 0, r.size() );
     // assertEquals( r.toString(), AbstractSearchResponse.LIMIT_EXCEEDED, response.getTotalHits() );
     // }
 }