/*
 * 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.slider.funtest.commands

import groovy.transform.CompileStatic
import groovy.util.logging.Slf4j
import org.apache.slider.common.SliderExitCodes
import org.apache.slider.funtest.categories.FunctionalTests
import org.apache.slider.common.params.Arguments
import org.apache.slider.common.params.SliderActions
import org.apache.slider.funtest.framework.CommandTestBase
import org.apache.slider.funtest.framework.SliderShell
import org.junit.BeforeClass
import org.junit.Test

/**
 * Test the return code from ops against unknown clusters are what we expect
 */
@CompileStatic
@Slf4j
@org.junit.experimental.categories.Category(FunctionalTests)
public class TestUnknownClusterOperations extends CommandTestBase {

  public static final String UNKNOWN = "unknown_cluster"

  @BeforeClass
  public static void prepareCluster() {
    assumeFunctionalTestsEnabled();
  }

  @Test
  public void testFreezeUnknownCluster() throws Throwable {
    SliderShell shell = freeze(UNKNOWN)
    assertUnknownCluster(shell)
  }

  @Test
  public void testFreezeUnknownClusterWithMessage() throws Throwable {
      slider(SliderExitCodes.EXIT_UNKNOWN_INSTANCE,
         [
        SliderActions.ACTION_FREEZE, UNKNOWN,
        Arguments.ARG_WAIT, Integer.toString(FREEZE_WAIT_TIME),
        Arguments.ARG_MESSAGE, "testFreezeUnknownClusterWithMessage"
        ])
  }

  @Test
  public void testFreezeForceUnknownCluster() throws Throwable {
    SliderShell shell = freezeForce(UNKNOWN)
    assertUnknownCluster(shell)
  }

  @Test
  public void testDestroyUnknownCluster() throws Throwable {
    SliderShell shell = destroy(UNKNOWN)
    assertSuccess(shell)
  }

  @Test
  public void testListUnknownCluster() throws Throwable {
    assertUnknownCluster(list(UNKNOWN))
  }

  @Test
  public void testExistsUnknownCluster() throws Throwable {
    assertUnknownCluster(exists(UNKNOWN, false))
  }

  @Test
  public void testExistsLiveUnknownCluster() throws Throwable {
    assertUnknownCluster(exists(UNKNOWN, true))
  }

  @Test
  public void testThawUnknownCluster() throws Throwable {
    assertUnknownCluster(thaw(UNKNOWN))
  }

  @Test
  public void testStatusUnknownCluster() throws Throwable {
    assertUnknownCluster(status(UNKNOWN))
  }

  @Test
  public void testGetConfUnknownCluster() throws Throwable {
    assertUnknownCluster(getConf(UNKNOWN))
  }

}
