testng/src/main/org/testng/internal/RunInfo.java

package org.testng.internal;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.testng.IMethodSelector;
import org.testng.ITestNGMethod;

/**
 * This class contains all the information needed to determine
 * what methods should be run.  It gets invoked by the TestRunner
 * and then goes through its list of method selectors to decide what methods
 * need to be run.
 * 
 * @author <a href="mailto:cedric@beust.com">Cedric Beust</a>
 */
public class RunInfo implements Serializable {
  transient private List<MethodSelectorDescriptor>
    m_methodSelectors = new ArrayList<MethodSelectorDescriptor>();
  
  public void addMethodSelector(IMethodSelector selector, int priority) {
    Utils.log("RunInfo", 3, "Adding method selector: " + selector + " priority: " + priority);
    MethodSelectorDescriptor md = new MethodSelectorDescriptor(selector, priority);
    m_methodSelectors.add(md);
  }
  
  public boolean includeMethod(ITestNGMethod tm, boolean isTestMethod) {
    Collections.sort(m_methodSelectors);
    boolean foundNegative = false;
    
    for (MethodSelectorDescriptor mds : m_methodSelectors) {
      // If we found any negative priority, we break as soon as we encounter
      // a positive
      if (! foundNegative) foundNegative = mds.getPriority() < 0;
      if (foundNegative && mds.getPriority() >= 0) break;
      
      // Proceeed normally
      IMethodSelector md = mds.getMethodSelector();
      boolean result = md.includeMethod(tm, isTestMethod);
      if (result) {
        return true;
      }
    }
    
    return false;
  }
  
  public static void ppp(String s) {
    System.out.println("[RunInfo] " + s);
  }

  public void setTestMethods(List<ITestNGMethod> testMethods) {
    for (MethodSelectorDescriptor mds : m_methodSelectors) {
      mds.setTestMethods(testMethods);
    }
  }
}