solr/core/src/test/org/apache/solr/cloud/DeleteStatusTest.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 java.io.IOException;
import java.util.concurrent.TimeUnit;

import org.apache.solr.SolrTestUtil;
import org.apache.solr.client.solrj.SolrClient;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.impl.CloudHttp2SolrClient;
import org.apache.solr.client.solrj.request.CollectionAdminRequest;
import org.apache.solr.client.solrj.response.CollectionAdminResponse;
import org.apache.solr.client.solrj.response.RequestStatusState;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;

public class DeleteStatusTest extends SolrCloudTestCase {

  public static final int MAX_WAIT_TIMEOUT = 15;

  @BeforeClass
  public static void createCluster() throws Exception {
    configureCluster(2)
        .addConfig("conf1", SolrTestUtil.TEST_PATH().resolve("configsets").resolve("cloud-minimal").resolve("conf"))
        .formatZk(true).configure();
  }

  @AfterClass
  public static void afterDeleteStatusTest() throws Exception {
    shutdownCluster();
  }

  // Basically equivalent to RequestStatus.waitFor(), but doesn't delete the id from the queue
  private static RequestStatusState waitForRequestState(String id, SolrClient client, int timeout)
      throws IOException, SolrServerException, InterruptedException {
    RequestStatusState state = RequestStatusState.SUBMITTED;
    long endTime = System.nanoTime() + TimeUnit.SECONDS.toNanos(MAX_WAIT_TIMEOUT);
    while (System.nanoTime() < endTime) {
      state = CollectionAdminRequest.requestStatus(id).process(client).getRequestStatus();
      if (state == RequestStatusState.COMPLETED)
        break;
    //  assumeTrue("Error creating collection - skipping test", state != RequestStatusState.FAILED);
      TimeUnit.MILLISECONDS.sleep(100);
    }
    //assumeTrue("Timed out creating collection - skipping test", state == RequestStatusState.COMPLETED);
    return state;
  }

  @Test
  public void testAsyncIdsMayBeDeleted() throws Exception {

    final CloudHttp2SolrClient client = cluster.getSolrClient();

    final String collection = "deletestatus";
    final String asyncId = CollectionAdminRequest.createCollection(collection, "conf1", 1, 1).processAsync(client);

    waitForRequestState(asyncId, client, MAX_WAIT_TIMEOUT);

    assertEquals(RequestStatusState.COMPLETED,
        CollectionAdminRequest.requestStatus(asyncId).process(client).getRequestStatus());

    CollectionAdminResponse rsp = CollectionAdminRequest.deleteAsyncId(asyncId).process(client);
    assertEquals("successfully removed stored response for [" + asyncId + "]", rsp.getResponse().get("status"));

    assertEquals(RequestStatusState.NOT_FOUND,
        CollectionAdminRequest.requestStatus(asyncId).process(client).getRequestStatus());

  }

  @Test
  public void testDeletingNonExistentRequests() throws Exception {

    final CloudHttp2SolrClient client = cluster.getSolrClient();

    CollectionAdminResponse rsp = CollectionAdminRequest.deleteAsyncId("foo").process(client);
    assertEquals("[foo] not found in stored responses", rsp.getResponse().get("status"));

  }

  @Test
  public void testProcessAndWaitDeletesAsyncIds() throws IOException, SolrServerException, InterruptedException {

    final CloudHttp2SolrClient client = cluster.getSolrClient();

    RequestStatusState state = CollectionAdminRequest.createCollection("requeststatus", "conf1", 1, 1)
                                  .processAndWait("request1", client, MAX_WAIT_TIMEOUT);

    assertSame(RequestStatusState.COMPLETED, state);

    // using processAndWait deletes the requestid
    state = CollectionAdminRequest.requestStatus("request1").process(client).getRequestStatus();
    assertSame("Request id was not deleted by processAndWait call", RequestStatusState.NOT_FOUND, state);

  }

  @Test
  public void testDeleteStatusFlush() throws Exception {

    final CloudHttp2SolrClient client = cluster.getSolrClient();

    String id1 = CollectionAdminRequest.createCollection("flush1", "conf1", 1, 1).processAsync(client);
    String id2 = CollectionAdminRequest.createCollection("flush2", "conf1", 1, 1).processAsync(client);

    assertEquals(RequestStatusState.COMPLETED, waitForRequestState(id1, client, MAX_WAIT_TIMEOUT));
    assertEquals(RequestStatusState.COMPLETED, waitForRequestState(id2, client, MAX_WAIT_TIMEOUT));

    CollectionAdminRequest.deleteAllAsyncIds().process(client);

    assertEquals(RequestStatusState.NOT_FOUND,
        CollectionAdminRequest.requestStatus(id1).process(client).getRequestStatus());
    assertEquals(RequestStatusState.NOT_FOUND,
        CollectionAdminRequest.requestStatus(id2).process(client).getRequestStatus());

  }
}