alluxio-master-2.x/minicluster/src/main/java/alluxio/master/TestUtils.java
/*
 * The Alluxio Open Foundation licenses this work under the Apache License, version 2.0
 * (the "License"). You may not use this work except in compliance with the License, which is
 * available at www.apache.org/licenses/LICENSE-2.0
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied, as more fully set forth in the License.
 *
 * See the NOTICE file distributed with this work for information regarding copyright ownership.
 */

package alluxio.master;

import alluxio.Constants;
import alluxio.master.file.FileSystemMaster;
import alluxio.master.file.meta.InodeLockManager;
import alluxio.util.ThreadUtils;

import org.powermock.reflect.Whitebox;

/**
 * Test utilities.
 */
public class TestUtils {
  private static final int SERVER_START_TIMEOUT_MS = Constants.MINUTE_MS;

  /**
   * @param process the process to wait for
   */
  public static void waitForReady(alluxio.Process process) {
    if (!process.waitForReady(SERVER_START_TIMEOUT_MS)) {
      ThreadUtils.logAllThreads();
      throw new RuntimeException(
          String.format("Timed out waiting for process %s to start", process));
    }
  }

  /**
   * Verifies that all master inode tree locks have been released for the given cluster.
   *
   * @param cluster the cluster
   */
  public static void assertAllLocksReleased(LocalAlluxioCluster cluster) {
    InodeLockManager lockManager =
        (InodeLockManager) Whitebox.getInternalState(cluster.getLocalAlluxioMaster()
            .getMasterProcess().getMaster(FileSystemMaster.class), "mInodeLockManager");
    lockManager.assertAllLocksReleased();
  }
}
