solr/core/src/test/org/apache/solr/util/AuthToolTest.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.util;

import java.nio.file.Files;
import java.nio.file.Path;

import org.apache.commons.cli.CommandLine;
import org.apache.solr.cloud.SolrCloudTestCase;
import org.junit.After;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;

import static org.apache.solr.util.SolrCLI.findTool;
import static org.apache.solr.util.SolrCLI.parseCmdLine;

/**
 * Unit test for SolrCLI's AuthTool
 */
public class AuthToolTest extends SolrCloudTestCase {
  private Path dir;

  @BeforeClass
  public static void setupCluster() throws Exception {
    configureCluster(1)
        .addConfig("config", TEST_PATH().resolve("configsets").resolve("cloud-minimal").resolve("conf"))
        .configure();
  }

  @Before
  public void setUp() throws Exception {
    super.setUp();
    dir = createTempDir("AuthToolTest").toAbsolutePath();
  }

  @After
  public void tearDown() throws Exception {
    super.tearDown();
    if (null != dir) {
      org.apache.commons.io.FileUtils.deleteDirectory(dir.toFile());
      dir = null;
    }
  }

  @Test
  public void testEnableAuth() throws Exception {
    Path solrIncludeFile = Files.createFile(dir.resolve("solrIncludeFile.txt"));
    String[] args = {"auth", "enable",
        "-zkHost", cluster.getZkClient().getZkServerAddress(),
        "-authConfDir", dir.toAbsolutePath().toString(),
        "-solrIncludeFile", solrIncludeFile.toAbsolutePath().toString(),
        "-credentials", "solr:solr",
        "-blockUnknown", "true"};
    assertEquals(0, runTool(args));
  }

  private int runTool(String[] args) throws Exception {
    SolrCLI.Tool tool = findTool(args);
    assertTrue(tool instanceof SolrCLI.AuthTool);
    CommandLine cli = parseCmdLine(args, tool.getOptions());
    return tool.runTool(cli);
  }
}