package test.annotationtransformer;

import org.testng.IAnnotationTransformer2;
import org.testng.annotations.IConfigurationAnnotation;
import org.testng.annotations.IDataProviderAnnotation;
import org.testng.annotations.IFactoryAnnotation;
import org.testng.annotations.ITestAnnotation;

import java.lang.reflect.Constructor;
import java.lang.reflect.Method;

public class ConfigurationTransformer implements IAnnotationTransformer2 {

  @Override
  public void transform(ITestAnnotation annotation, Class testClass,
      Constructor testConstructor, Method testMethod)
  {
  }

  @Override
  public void transform(IConfigurationAnnotation annotation, Class testClass,
      Constructor testConstructor, Method testMethod)
  {
    if (annotation.getBeforeTestMethod()) {
      annotation.setEnabled(false);
    }
  }

  @Override
  public void transform(IDataProviderAnnotation annotation, Method testMethod) {
  }

  @Override
  public void transform(IFactoryAnnotation annotation, Method testMethod) {
  }

}