View Javadoc
1   package org.apache.maven.surefire.testng.utils;
2   
3   /*
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *     http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.
20   */
21  
22  import java.util.List;
23  
24  import org.apache.maven.surefire.testset.TestListResolver;
25  import org.testng.IMethodSelector;
26  import org.testng.IMethodSelectorContext;
27  import org.testng.ITestNGMethod;
28  
29  /**
30   * For internal use only
31   *
32   * @author Olivier Lamy
33   * @noinspection UnusedDeclaration
34   * @since 2.7.3
35   */
36  public class MethodSelector
37      implements IMethodSelector
38  {
39      private static volatile TestListResolver testListResolver = null;
40  
41      public void setTestMethods( List arg0 )
42      {
43      }
44  
45      public boolean includeMethod( IMethodSelectorContext context, ITestNGMethod testngMethod, boolean isTestMethod )
46      {
47          return testngMethod.isBeforeClassConfiguration() || testngMethod.isBeforeGroupsConfiguration()
48              || testngMethod.isBeforeMethodConfiguration() || testngMethod.isBeforeSuiteConfiguration()
49              || testngMethod.isBeforeTestConfiguration() || testngMethod.isAfterClassConfiguration()
50              || testngMethod.isAfterGroupsConfiguration() || testngMethod.isAfterMethodConfiguration()
51              || testngMethod.isAfterSuiteConfiguration() || testngMethod.isAfterTestConfiguration()
52              || shouldRun( testngMethod );
53      }
54  
55      public static void setTestListResolver( TestListResolver testListResolver )
56      {
57          MethodSelector.testListResolver = testListResolver;
58      }
59  
60      private static boolean shouldRun( ITestNGMethod test )
61      {
62          TestListResolver resolver = testListResolver;
63          boolean hasTestResolver = resolver != null && !resolver.isEmpty();
64          return hasTestResolver && resolver.shouldRun( test.getRealClass(), test.getMethodName() );
65      }
66  }