/*
 * 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.providers.accumulo.live

import groovy.transform.CompileStatic
import groovy.util.logging.Slf4j
import org.apache.hadoop.yarn.api.records.FinalApplicationStatus
import org.apache.slider.client.SliderClient
import org.apache.slider.core.main.ServiceLauncher
import org.apache.slider.providers.accumulo.AccumuloConfigFileOptions
import org.apache.slider.providers.accumulo.AccumuloKeys
import org.apache.slider.providers.accumulo.AccumuloTestBase
import org.junit.Test

@CompileStatic
@Slf4j
class TestAccFreezeThaw extends AccumuloTestBase {

  @Test
  public void testAccFreezeThaw() throws Throwable {
    int tablets = 1
    int monitor = 1
    int gc = 1
    String clustername = createMiniCluster("",
        configuration, 1, 1, 1, true, false)
    describe(" Create an accumulo cluster");

    //now launch the cluster
    Map<String, Integer> roles = [
        (AccumuloKeys.ROLE_MASTER): 1,
        (AccumuloKeys.ROLE_TABLET): tablets,
        (AccumuloKeys.ROLE_MONITOR): monitor,
        (AccumuloKeys.ROLE_GARBAGE_COLLECTOR): gc
    ];
    ServiceLauncher launcher = createAccCluster(clustername, roles, [], true, true)
    SliderClient sliderClient = launcher.service
    addToTeardown(sliderClient);

    
    waitForRoleCount(sliderClient, roles, accumulo_cluster_startup_to_live_time)
    //now give the cluster a bit of time to actually start work

    log.info("Sleeping for a while")
    sleepForAccumuloClusterLive();
    //verify that all is still there
    waitForRoleCount(sliderClient, roles, 0, "extended cluster operation")

    String page = fetchLocalPage(AccumuloConfigFileOptions.MONITOR_PORT_CLIENT_INT,
                                 AccumuloKeys.MONITOR_PAGE_JSON)
    log.info(page);

    log.info("Stopping")
    clusterActionFreeze(sliderClient, clustername, "stop");
    def finishedAppReport = waitForAppToFinish(sliderClient)
    assert finishedAppReport.finalApplicationStatus ==
           FinalApplicationStatus.SUCCEEDED

    //make sure the fetch fails
    try {
      page = fetchLocalPage(AccumuloConfigFileOptions.MONITOR_PORT_CLIENT_INT,
                     AccumuloKeys.MONITOR_PAGE_JSON)
      //this should have failed
      log.error("The accumulo monitor is still live")
      log.error(page)
    } catch (ConnectException expected) {
      //
      log.debug("expected exception", expected)
    }
    //force kill any accumulo processes
    if (kill_supported) {
      killAllAccumuloProcesses()
    }

    sleepForAccumuloClusterLive();

    log.info("Thawing")
    
    ServiceLauncher launcher2 = thawCluster(clustername, [], true);
    SliderClient sliderClient2 = (SliderClient) launcher2.service
    addToTeardown(sliderClient2)
    waitForRoleCount(sliderClient, roles, accumulo_cluster_startup_to_live_time, "thawing")


    sleepForAccumuloClusterLive();
    //verify that all is still there
    waitForRoleCount(sliderClient, roles, 0, "extended cluster operation after thawing")
    page = fetchLocalPage(
        AccumuloConfigFileOptions.MONITOR_PORT_CLIENT_INT,
        AccumuloKeys.MONITOR_PAGE_JSON)
    log.info(page);
  }

  public void sleepForAccumuloClusterLive() {
    sleep(ACCUMULO_GO_LIVE_TIME)
  }

}