solr/core/src/test/org/apache/solr/cloud/DeleteInactiveReplicaTest.java

/*
 * 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.solr.cloud;

import org.apache.solr.SolrTestUtil;
import org.apache.solr.client.solrj.SolrClient;
import org.apache.solr.client.solrj.embedded.JettySolrRunner;
import org.apache.solr.client.solrj.request.CollectionAdminRequest;
import org.apache.solr.common.cloud.DocCollection;
import org.apache.solr.common.cloud.Replica;
import org.apache.solr.common.cloud.Slice;
import org.junit.BeforeClass;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.invoke.MethodHandles;
import java.util.ArrayList;
import java.util.List;

public class DeleteInactiveReplicaTest extends SolrCloudTestCase {

  private static final Logger log = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());

  @BeforeClass
  public static void setupCluster() throws Exception {
    useFactory(null);
    configureCluster(4)
        .addConfig("conf", SolrTestUtil.configset("cloud-minimal"))
        .configure();
  }

  @Test
  public void deleteInactiveReplicaTest() throws Exception, NoOpenOverseerFoundException {

    String collectionName = "delDeadColl";
    int replicationFactor = 2;
    int numShards = 2;
    int maxShardsPerNode = ((((numShards + 1) * replicationFactor) / cluster.getJettySolrRunners().size())) + 1;

    CollectionAdminRequest.createCollection(collectionName, "conf", numShards, replicationFactor)
        .setMaxShardsPerNode(maxShardsPerNode)
        .process(cluster.getSolrClient());
    waitForState("Expected a cluster of 2 shards and 2 replicas", collectionName, (n, c) -> {
      return DocCollection.isFullyActive(n, c, numShards, replicationFactor);
    });

    DocCollection collectionState = getCollectionState(collectionName);

    Slice shard = getRandomShard(collectionState);
    Replica replica = getRandomReplica(shard);

    List<JettySolrRunner> jetties = new ArrayList<>(cluster.getJettySolrRunners());
    JettySolrRunner overseerJetty = cluster.getCurrentOverseerJetty();
    jetties.remove(overseerJetty);
    JettySolrRunner jetty = jetties.iterator().next();

    cluster.stopJettySolrRunner(jetty);

    jetties = new ArrayList<>(cluster.getJettySolrRunners());
    jetties.remove(jetty);

    JettySolrRunner ojetty = jetties.iterator().next();

    if (log.isInfoEnabled()) {
      log.info("Removing replica {}/{} ", shard.getName(), replica.getName());
    }
    try (SolrClient client = ojetty.newClient()) {
      CollectionAdminRequest.deleteReplica(collectionName, shard.getName(), replica.getName()).process(client);
    }

// TODO: this could come back in, but we should do simple, then more complicated, perhaps nightly

//    cluster.startJettySolrRunner(jetty);
//    log.info("restarted jetty");
//
//    // Check that we can't create a core with no coreNodeName
//    try (SolrClient queryClient = getHttpSolrClient(jetty.getBaseUrl().toString())) {
//      Exception e = expectThrows(Exception.class, () -> {
//        CoreAdminRequest.Create createRequest = new CoreAdminRequest.Create();
//        createRequest.setCoreName("testcore");
//        createRequest.setCollection(collectionName);
//        createRequest.setShardId("s2");
//        queryClient.request(createRequest);
//      });
//      assertTrue("Unexpected error message: " + e.getMessage(), e.getMessage().contains("No coreNodeName for"));
//
//    }
  }
}