package org.apache.solr.handler.component;

/**
 * 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.
 */

import java.io.File;
import java.io.IOException;

import org.apache.solr.BaseDistributedSearchTestCase;
import org.apache.solr.common.SolrInputDocument;
import org.apache.solr.common.params.CommonParams;
import org.apache.solr.common.util.FileUtils;
import org.junit.BeforeClass;

/**
 * 
 */
public class DistributedQueryElevationComponentTest extends BaseDistributedSearchTestCase {

  public DistributedQueryElevationComponentTest() {
    fixShardCount = true;
    shardCount = 3;
    stress = 0;

    // TODO: a better way to do this?
    configString = "solrconfig-elevate.xml";
    schemaString = "schema11.xml";
  }
  
  @BeforeClass
  public static void beforeClass() throws IOException {
    File parent = new File(TEST_HOME(), "conf");
    File elevateFile = new File(parent, "elevate.xml");
    File elevateDataFile = new File(parent, "elevate-data.xml");
    FileUtils.copyFile(elevateFile, elevateDataFile);
  }
  
  @Override
  public void doTest() throws Exception {
    
    
    del("*:*");
    indexr(id,"1", "int_i", "1", "text", "XXXX XXXX", "field_t", "anything");
    indexr(id,"2", "int_i", "2", "text", "YYYY YYYY", "plow_t", "rake");
    indexr(id,"3", "int_i", "3", "text", "ZZZZ ZZZZ");
    indexr(id,"4", "int_i", "4", "text", "XXXX XXXX");
    indexr(id,"5", "int_i", "5", "text", "ZZZZ ZZZZ ZZZZ");
    indexr(id,"6", "int_i", "6", "text", "ZZZZ");
    
    index_specific(2, id, "7", "int_i", "7", "text", "solr");
    commit();
    
    handle.put("explain", SKIPVAL);
    handle.put("debug", SKIPVAL);
    handle.put("QTime", SKIPVAL);
    handle.put("maxScore", SKIPVAL);
    handle.put("timestamp", SKIPVAL);
    handle.put("score", SKIPVAL);
    handle.put("wt", SKIP);
    handle.put("distrib", SKIP);
    handle.put("shards.qt", SKIP);
    handle.put("shards", SKIP);
    handle.put("q", SKIP);
    handle.put("qt", SKIP);
    query("q", "*:*", "qt", "/elevate", "shards.qt", "/elevate", "rows", "500", "sort", "id desc", CommonParams.FL, "id, score, [elevated]");

    query("q", "ZZZZ", "qt", "/elevate", "shards.qt", "/elevate", "rows", "500", CommonParams.FL, "*, [elevated]", "forceElevation", "true", "sort", "int_i desc");
    
    query("q", "solr", "qt", "/elevate", "shards.qt", "/elevate", "rows", "500", CommonParams.FL, "*, [elevated]", "forceElevation", "true", "sort", "int_i asc");
    
    query("q", "ZZZZ", "qt", "/elevate", "shards.qt", "/elevate", "rows", "500", CommonParams.FL, "*, [elevated]", "forceElevation", "true", "sort", "id desc");
  }
  
  protected void indexr(Object... fields) throws Exception {
    SolrInputDocument doc = new SolrInputDocument();
    addFields(doc, fields);
    indexDoc(doc);
  }
  
}

@@ -1091,16 +1091,69 @@
 eUtils;%0A
+import org.junit.After;%0Aimport org.junit.AfterClass;%0A
 import o
@@ -1520,16 +1520,89 @@
 %0A  %7D%0A  %0A
+  private static String elevateFilename = %22elevate-data-distrib.xml%22;%0A  %0A
   @Befor
@@ -1660,24 +1660,86 @@
 Exception %7B%0A
+    System.setProperty(%22elevate.data.file%22, elevateFilename);%0A
     File par
@@ -1879,26 +1879,23 @@
 nt, 
-%22
 elevate
--data.xml%22
+Filename
 );%0A 
@@ -1954,16 +1954,137 @@
 %0A  %7D%0A  %0A
+  @AfterClass%0A  public static void afterClass() throws IOException %7B%0A    System.clearProperty(%22elevate.data.file%22);%0A  %7D%0A%0A
   @Overr

use a diff elevate file for distrib QEC test