fm.last.commons.test.extensions.MethodDataFolderExtension 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

package fm.last.commons.test.extensions;

import java.io.File;

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.BeforeEachCallback;
import org.junit.jupiter.api.extension.ExtensionContext;

public class MethodDataFolderExtension extends BaseDataFolder implements BeforeEachCallback {

  private final File parent = new File("src" + File.separator + "test" + File.separator + "data");
  private File folder;

  @Override
  public void beforeEach(ExtensionContext extensionContext) {
    Test testAnnotation = extensionContext.getRequiredTestMethod().getAnnotation(Test.class);
    if(testAnnotation != null) {
      String methodName = extensionContext.getRequiredTestMethod().getName();
      Class targetClass = extensionContext.getRequiredTestClass();
      folder = new File(parent, targetClass.getName().replaceAll(PACKAGE_DELIMITER_PATTERN,
          FILE_SEPARATOR_REPLACEMENT)
          + File.separator + methodName);
    }
  }

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