Java 类org.apache.hadoop.hbase.testclassification.MediumTests 实例源码

public Timeout.Builder withTimeout(Class<?> clazz) {
  Annotation annotation = clazz.getAnnotation(Category.class);
  if (annotation != null) {
    Category category = (Category)annotation;
    for (Class<?> c: category.value()) {
      if (c == SmallTests.class) {
        // See SmallTests. Supposed to run 15 seconds.
        return withTimeout(30, TimeUnit.SECONDS);
      } else if (c == MediumTests.class) {
        // See MediumTests. Supposed to run 50 seconds.
        return withTimeout(180, TimeUnit.SECONDS);
      } else if (c == LargeTests.class) {
        // Let large tests have a ten minute timeout.
        return withTimeout(10, TimeUnit.MINUTES);
      }
    }
  }
  return this;
}

private static long getTimeoutInSeconds(Class<?> clazz) {
  Category[] categories = clazz.getAnnotationsByType(Category.class);
  if (categories.length == 0) {
    throw new IllegalArgumentException(clazz.getName() + " is not annotated with @Category");
  }
  for (Class<?> c : categories[0].value()) {
    if (c == SmallTests.class) {
      // See SmallTests. Supposed to run 15 seconds.
      // Lots of these timeout on Jenkins... a stall of ten or twenty seconds mess up what looks
      // fine when run local.
      return 60;
    } else if (c == MediumTests.class) {
      // See MediumTests. Supposed to run 50 seconds.
      return 180;
    } else if (c == LargeTests.class) {
      // Let large tests have a ten minute timeout.
      return TimeUnit.MINUTES.toSeconds(10);
    }
  }
  throw new IllegalArgumentException(
      clazz.getName() + " does not have SmallTests/MediumTests/LargeTests in @Category");
}