solr/solrj/src/test/org/apache/solr/client/solrj/impl/CloudSolrServerTest.java

package org.apache.solr.client.solrj.impl;

/*
 * 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.util.HashSet;
import java.util.List;
import java.util.Set;

import org.apache.lucene.util.LuceneTestCase.Slow;
import org.apache.solr.cloud.AbstractFullDistribZkTestBase;
import org.apache.solr.cloud.AbstractZkTestCase;
import org.apache.solr.common.SolrInputDocument;
import org.apache.solr.util.ExternalPaths;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;

/**
 * This test would be faster if we simulated the zk state instead.
 */
@Slow
public class CloudSolrServerTest extends AbstractFullDistribZkTestBase {
  
  private static final String SOLR_HOME = ExternalPaths.SOURCE_HOME + File.separator + "solrj"
      + File.separator + "src" + File.separator + "test-files"
      + File.separator + "solrj" + File.separator + "solr";

  @BeforeClass
  public static void beforeSuperClass() {
      AbstractZkTestCase.SOLRHOME = new File(SOLR_HOME());
  }
  
  @AfterClass
  public static void afterSuperClass() {
    
  }
  
  protected String getCloudSolrConfig() {
    return "solrconfig.xml";
  }
  
  @Override
  public String getSolrHome() {
    return SOLR_HOME;
  }
  
  public static String SOLR_HOME() {
    return SOLR_HOME;
  }
  
  @Before
  @Override
  public void setUp() throws Exception {
    super.setUp();
    // we expect this time of exception as shards go up and down...
    //ignoreException(".*");
    
    System.setProperty("numShards", Integer.toString(sliceCount));
  }
  
  @Override
  @After
  public void tearDown() throws Exception {
    super.tearDown();
    resetExceptionIgnores();
  }
  
  public CloudSolrServerTest() {
    super();
    sliceCount = 2;
    shardCount = 4;
  }
  
  @Override
  public void doTest() throws Exception {
    assertNotNull(cloudClient);
    
    handle.clear();
    handle.put("QTime", SKIPVAL);
    handle.put("timestamp", SKIPVAL);
    
    waitForThingsToLevelOut(30);

    del("*:*");

    indexr(id, 0, "a_t", "to come to the aid of their country.");
    
    CloudJettyRunner shard1Leader = shardToLeaderJetty.get("shard1");
    CloudJettyRunner shard2Leader = shardToLeaderJetty.get("shard2");
    
    if (cloudClient.isUpdatesToLeaders()) {
      // compare leaders list
      assertEquals(2, cloudClient.getLeaderUrlLists().get("collection1").size());
      HashSet<String> leaderUrlSet = new HashSet<String>();
      leaderUrlSet.addAll(cloudClient.getLeaderUrlLists().get("collection1"));
      assertTrue("fail check for leader:" + shard1Leader.url + " in "
          + leaderUrlSet, leaderUrlSet.contains(shard1Leader.url + "/"));
      assertTrue("fail check for leader:" + shard2Leader.url + " in "
          + leaderUrlSet, leaderUrlSet.contains(shard2Leader.url + "/"));
      
      // compare replicas list
      Set<String> replicas = new HashSet<String>();
      List<CloudJettyRunner> jetties = shardToJetty.get("shard1");
      for (CloudJettyRunner cjetty : jetties) {
        replicas.add(cjetty.url);
      }
      jetties = shardToJetty.get("shard2");
      for (CloudJettyRunner cjetty : jetties) {
        replicas.add(cjetty.url);
      }
      replicas.remove(shard1Leader.url);
      replicas.remove(shard2Leader.url);
      
      assertEquals(replicas.size(),
          cloudClient.getReplicasLists().get("collection1").size());
      
      for (String url : cloudClient.getReplicasLists().get("collection1")) {
        assertTrue("fail check for replica:" + url + " in " + replicas,
            replicas.contains(stripTrailingSlash(url)));
      }
    }
    
  }

  private String stripTrailingSlash(String url) {
    if (url.endsWith("/")) {
      return url.substring(0, url.length() - 1);
    }
    return url;
  }
  
  
  @Override
  protected void indexr(Object... fields) throws Exception {
    SolrInputDocument doc = getDoc(fields);
    indexDoc(doc);
  }

}