Support @Listeners in annotation transformer

src/main/java/org/testng/IAnnotationTransformer3.java

@@ -0,0 +1,9 @@
+package org.testng;
+
+import org.testng.annotations.IListenersAnnotation;
+
+public interface IAnnotationTransformer3 extends IAnnotationTransformer2 {
+
+  void transform(IListenersAnnotation annotation, Class testClass);
+
+}

src/main/java/org/testng/TestRunner.java

@@ -15,6 +15,7 @@
 import java.util.concurrent.TimeUnit;
 
 import org.testng.annotations.Guice;
+import org.testng.annotations.IListenersAnnotation;
 import org.testng.collections.ListMultiMap;
 import org.testng.collections.Lists;
 import org.testng.collections.Maps;
@@ -265,7 +266,7 @@ private ListenerHolder findAllListeners(Class<?> cls) {
     result.listenerClasses = Lists.newArrayList();
 
     do {
-      IListeners l = m_annotationFinder.findAnnotation(cls, IListeners.class);
+      IListenersAnnotation l = m_annotationFinder.findAnnotation(cls, IListenersAnnotation.class);
       if (l != null) {
         Class<? extends ITestNGListener>[] classes = l.getValue();
         for (Class<? extends ITestNGListener> c : classes) {

src/main/java/org/testng/annotations/IListenersAnnotation.java

@@ -0,0 +1,10 @@
+package org.testng.annotations;
+
+import org.testng.ITestNGListener;
+
+public interface IListenersAnnotation extends IAnnotation {
+
+  Class<? extends ITestNGListener>[] getValue();
+
+  void setValue(Class<? extends ITestNGListener>[] value);
+}

src/main/java/org/testng/internal/annotations/IListeners.java

@@ -1,10 +1,7 @@
 package org.testng.internal.annotations;
 
-import org.testng.ITestNGListener;
-import org.testng.annotations.IAnnotation;
+import org.testng.annotations.IListenersAnnotation;
 
-public interface IListeners extends IAnnotation {
-  Class<? extends ITestNGListener>[] getValue();
+public interface IListeners extends IListenersAnnotation {
 
-  void setValue(Class<? extends ITestNGListener>[] value);
 }

src/main/java/org/testng/internal/annotations/JDK15AnnotationFinder.java

@@ -8,6 +8,7 @@
 
 import org.testng.IAnnotationTransformer;
 import org.testng.IAnnotationTransformer2;
+import org.testng.IAnnotationTransformer3;
 import org.testng.ITestNGMethod;
 import org.testng.annotations.AfterClass;
 import org.testng.annotations.AfterGroups;
@@ -28,6 +29,7 @@
 import org.testng.annotations.IDataProviderAnnotation;
 import org.testng.annotations.IExpectedExceptionsAnnotation;
 import org.testng.annotations.IFactoryAnnotation;
+import org.testng.annotations.IListenersAnnotation;
 import org.testng.annotations.IObjectFactoryAnnotation;
 import org.testng.annotations.IParametersAnnotation;
 import org.testng.annotations.ITestAnnotation;
@@ -56,13 +58,15 @@ public class JDK15AnnotationFinder implements IAnnotationFinder {
   @SuppressWarnings({"deprecation"})
   public JDK15AnnotationFinder(IAnnotationTransformer transformer) {
     m_transformer = transformer;
+    m_annotationMap.put(IListenersAnnotation.class, Listeners.class);
     m_annotationMap.put(IConfigurationAnnotation.class, Configuration.class);
     m_annotationMap.put(IDataProviderAnnotation.class, DataProvider.class);
     m_annotationMap.put(IExpectedExceptionsAnnotation.class, ExpectedExceptions.class);
     m_annotationMap.put(IFactoryAnnotation.class, Factory.class);
     m_annotationMap.put(IObjectFactoryAnnotation.class, ObjectFactory.class);
     m_annotationMap.put(IParametersAnnotation.class, Parameters.class);
     m_annotationMap.put(ITestAnnotation.class, Test.class);
+    // internal
     m_annotationMap.put(IBeforeSuite.class, BeforeSuite.class);
     m_annotationMap.put(IAfterSuite.class, AfterSuite.class);
     m_annotationMap.put(IBeforeTest.class, BeforeTest.class);
@@ -73,7 +77,6 @@ public JDK15AnnotationFinder(IAnnotationTransformer transformer) {
     m_annotationMap.put(IAfterGroups.class, AfterGroups.class);
     m_annotationMap.put(IBeforeMethod.class, BeforeMethod.class);
     m_annotationMap.put(IAfterMethod.class, AfterMethod.class);
-    m_annotationMap.put(IListeners.class, Listeners.class);
   }
 
   private <A extends Annotation> A findAnnotationInSuperClasses(Class<?> cls, Class<A> a) {
@@ -164,7 +167,18 @@ else if (a instanceof IDataProviderAnnotation) {
       else if (a instanceof IFactoryAnnotation) {
         transformer2.transform((IFactoryAnnotation) a, testMethod);
       }
-    }
+
+      else if (m_transformer instanceof IAnnotationTransformer3) {
+        IAnnotationTransformer3 transformer = (IAnnotationTransformer3) m_transformer;
+
+        //
+        // Transform @Listeners
+        //
+        if (a instanceof IListenersAnnotation) {
+          transformer.transform((IListenersAnnotation)a, testClass);
+        }
+      } // End IAnnotationTransformer3
+    } // End IAnnotationTransformer2
   }
 
   @Override

src/main/java/org/testng/internal/annotations/JDK15TagFactory.java

@@ -27,6 +27,7 @@
 import org.testng.annotations.IDataProviderAnnotation;
 import org.testng.annotations.IExpectedExceptionsAnnotation;
 import org.testng.annotations.IFactoryAnnotation;
+import org.testng.annotations.IListenersAnnotation;
 import org.testng.annotations.IObjectFactoryAnnotation;
 import org.testng.annotations.IParametersAnnotation;
 import org.testng.annotations.ITestAnnotation;
@@ -71,7 +72,7 @@ else if (annotationClass == IObjectFactoryAnnotation.class) {
       else if (annotationClass == ITestAnnotation.class) {
         result = createTestTag(cls, a, transformer);
       }
-      else if (annotationClass == IListeners.class) {
+      else if (annotationClass == IListenersAnnotation.class) {
         result = createListenersTag(cls, a, transformer);
       }
       else if (annotationClass == IBeforeSuite.class || annotationClass == IAfterSuite.class ||

src/test/java/test/annotationtransformer/AnnotationTransformerClassSampleTest.java

@@ -1,5 +1,6 @@
 package test.annotationtransformer;
 
+import org.testng.annotations.Listeners;
 import org.testng.annotations.Test;
 
 /**
@@ -10,6 +11,7 @@
  *
  */
 @Test(timeOut = 1000)
+@Listeners(MySuiteListener.class)
 public class AnnotationTransformerClassSampleTest {
 
   public void one() {

src/test/java/test/annotationtransformer/AnnotationTransformerTest.java

@@ -123,6 +123,8 @@ public void verifyAnnotationTransformerClass() {
   private void runTest(IAnnotationTransformer transformer,
       String passedName, String failedName)
   {
+    MySuiteListener.triggered = false;
+    MySuiteListener2.triggered = false;
     TestNG tng = new TestNG();
     tng.setVerbose(0);
     if (transformer != null) {
@@ -140,8 +142,23 @@ private void runTest(IAnnotationTransformer transformer,
 
     Assert.assertEquals(results.size(), 1);
     Assert.assertEquals(name, results.get(0).getMethod().getMethodName());
+    Assert.assertTrue(MySuiteListener.triggered);
+    Assert.assertFalse(MySuiteListener2.triggered);
   }
 
+  @Test
+  public void verifyListenerAnnotationTransformerClass() {
+    MySuiteListener.triggered = false;
+    MySuiteListener2.triggered = false;
+    TestNG tng = new TestNG();
+    tng.setVerbose(0);
+    tng.setAnnotationTransformer(new MyListenerTransformer());
+    tng.setTestClasses(new Class[]{AnnotationTransformerClassSampleTest.class});
+
+    tng.run();
+    Assert.assertFalse(MySuiteListener.triggered);
+    Assert.assertTrue(MySuiteListener2.triggered);
+  }
 
   @Test
   public void verifyConfigurationTransformer() {

src/test/java/test/annotationtransformer/MyListenerTransformer.java

@@ -0,0 +1,33 @@
+package test.annotationtransformer;
+
+import org.testng.IAnnotationTransformer3;
+import org.testng.annotations.IConfigurationAnnotation;
+import org.testng.annotations.IDataProviderAnnotation;
+import org.testng.annotations.IFactoryAnnotation;
+import org.testng.annotations.IListenersAnnotation;
+import org.testng.annotations.ITestAnnotation;
+
+import java.lang.reflect.Constructor;
+import java.lang.reflect.Method;
+
+public class MyListenerTransformer implements IAnnotationTransformer3 {
+
+  @Override
+  public void transform(IListenersAnnotation annotation, Class testClass) {
+    annotation.setValue(new Class[]{MySuiteListener2.class});
+  }
+
+  @Override
+  public void transform(IConfigurationAnnotation annotation, Class testClass,
+                        Constructor testConstructor, Method testMethod) {}
+
+  @Override
+  public void transform(IDataProviderAnnotation annotation, Method method) {}
+
+  @Override
+  public void transform(IFactoryAnnotation annotation, Method method) {}
+
+  @Override
+  public void transform(ITestAnnotation annotation, Class testClass, Constructor testConstructor,
+                        Method testMethod) {}
+}

src/test/java/test/annotationtransformer/MySuiteListener.java

@@ -0,0 +1,17 @@
+package test.annotationtransformer;
+
+import org.testng.ISuite;
+import org.testng.ISuiteListener;
+
+public class MySuiteListener implements ISuiteListener {
+
+  public static boolean triggered = false;
+
+  @Override
+  public void onStart(ISuite suite) {
+    triggered = true;
+  }
+
+  @Override
+  public void onFinish(ISuite suite) {}
+}

src/test/java/test/annotationtransformer/MySuiteListener2.java

@@ -0,0 +1,17 @@
+package test.annotationtransformer;
+
+import org.testng.ISuite;
+import org.testng.ISuiteListener;
+
+public class MySuiteListener2 implements ISuiteListener {
+
+  public static boolean triggered = false;
+
+  @Override
+  public void onStart(ISuite suite) {
+    triggered = true;
+  }
+
+  @Override
+  public void onFinish(ISuite suite) {}
+}