fm.last.commons.test.file.ClassDataFolder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lastcommons-test Show documentation

Last.fm common test utilities

There is a newer version: 7.0.2

/*
 * Copyright 2012-2019 Last.fm
 *
 *  Licensed 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 fm.last.commons.test.file;

import java.io.File;
import java.lang.annotation.Annotation;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;

public final class ClassDataFolder extends DataFolder {

  private final File parent;
  private File folder;

  public ClassDataFolder() {
    parent = new File("src" + File.separator + "test" + File.separator + "data");
  }

  @SuppressWarnings("unchecked")
  @Override
  public Statement apply(final Statement base, Description description) {
    if (notAnnotatedWithAny(description, Test.class, Before.class, After.class)) {
      return base;
    }

    Class targetClass = description.getTestClass();
    folder = new File(parent, targetClass.getName().replaceAll(PACKAGE_DELIMITER_PATTERN, FILE_SEPARATOR_REPLACEMENT));
    return new Statement() {
      @Override
      public void evaluate() throws Throwable {
        base.evaluate();
      }
    };
  }

  @Override
  public File getDataFolder() {
    return folder;
  }

  private boolean notAnnotatedWithAny(Description description, Class... annotationClasses){
    for (Class annotationClass : annotationClasses) {
      Annotation annotationInstance = description.getAnnotation(annotationClass);
      if (annotationInstance != null) {
        return false;
      }
    }
    return true;
  }
}