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.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() {
    
  }
  
  @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 = 6;
  }
  
  @Override
  public void doTest() throws Exception {
    
    handle.clear();
    handle.put("QTime", SKIPVAL);
    handle.put("timestamp", SKIPVAL);
    
    waitForThingsToLevelOut(15);

    del("*:*");

    indexr(id, 0, "a_t", "to come to the aid of their country.");
    
    // compare leaders list
    CloudJettyRunner shard1Leader = shardToLeaderJetty.get("shard1");
    CloudJettyRunner shard2Leader = shardToLeaderJetty.get("shard2");
    assertEquals(2, cloudClient.getLeaderUrlList().size());
    HashSet<String> leaderUrlSet = new HashSet<String>();
    leaderUrlSet.addAll(cloudClient.getLeaderUrlList());
    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.getReplicasList().size());
    
    for (String url : cloudClient.getReplicasList()) {
      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;
  }

}