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

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

import java.lang.reflect.AnnotatedElement;

import org.testng.annotations.Test;

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 TestHandler implements IAnnotationHandler {
   INSTANCE;

   @Override
   public Class getTargetAnnotation() {
      return Test.class;
   }

   @Override
   public void handle(IClassContext context, Test meta, AnnotatedElement annotated) {
      IEunitContext ctx = context.forEunit();
      EunitMethod eunitMethod = ctx.peek();

      eunitMethod.setTest(true);

      if (!meta.enabled()) {
         eunitMethod.setIgnored(true);
      }

      if (meta.timeOut() > 0) {
         eunitMethod.setTimeout(meta.timeOut());
      }
   }

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

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