1   package org.apache.maven.surefire.testng.utils;
2   
3   
4   
5   
6   
7   
8   
9   
10  
11  
12  
13  
14  
15  
16  
17  
18  
19  
20  
21  
22  import java.util.List;
23  import org.apache.maven.shared.utils.io.SelectorUtils;
24  
25  import org.testng.IMethodSelector;
26  import org.testng.IMethodSelectorContext;
27  import org.testng.ITestNGMethod;
28  
29  
30  
31  
32  
33  
34  
35  
36  public class MethodSelector
37      implements IMethodSelector
38  {
39  
40      private static String METHOD_NAME = null;
41  
42      public void setTestMethods( List arg0 )
43      {
44          
45      }
46  
47      public boolean includeMethod( IMethodSelectorContext context, ITestNGMethod testngMethod, boolean isTestMethod )
48      {
49          if ( testngMethod.isBeforeClassConfiguration() || testngMethod.isBeforeGroupsConfiguration()
50              || testngMethod.isBeforeMethodConfiguration() || testngMethod.isBeforeSuiteConfiguration()
51              || testngMethod.isBeforeTestConfiguration() )
52          {
53              return true;
54          }
55          if ( testngMethod.isAfterClassConfiguration() || testngMethod.isAfterGroupsConfiguration()
56              || testngMethod.isAfterMethodConfiguration() || testngMethod.isAfterSuiteConfiguration()
57              || testngMethod.isAfterTestConfiguration() )
58          {
59              return true;
60          }
61  
62          return SelectorUtils.match( METHOD_NAME, testngMethod.getMethodName() );
63      }
64  
65      public static void setMethodName( String methodName )
66      {
67          METHOD_NAME = methodName;
68      }
69  }