org.unidal.eunit.handler.testng.AfterClassHandler Maven / Gradle / Ivy

Go to download
package org.unidal.eunit.handler.testng;

import java.lang.reflect.Method;
import java.lang.reflect.Modifier;

import org.testng.annotations.AfterClass;

import org.unidal.eunit.model.entity.EunitMethod;
import org.unidal.eunit.testfwk.spi.IAnnotationHandler;
import org.unidal.eunit.testfwk.spi.IClassContext;
import org.unidal.eunit.testfwk.spi.IClassContext.IEunitContext;

public enum AfterClassHandler implements IAnnotationHandler {
   INSTANCE;
   
   @Override
   public Class getTargetAnnotation() {
      return AfterClass.class;
   }

   @Override
   public void handle(IClassContext context, AfterClass meta, Method method) {
      int modifier = method.getModifiers();

      if (Modifier.isPublic(modifier) && Modifier.isStatic(modifier)) {
         IEunitContext ctx = context.forEunit();
         EunitMethod eunitMethod = ctx.peek();

         eunitMethod.setBeforeAfter(Boolean.FALSE);
         eunitMethod.setStatic(true);
      } else {
         throw new RuntimeException(String.format("Method %s() should be public static.", method.getName()));
      }
   }

   @Override
   public boolean isAfter() {
      return false;
   }

   @Override
   public String toString() {
      return String.format("%s.%s", getClass().getSimpleName(), name());
   }
}