src/jdk15/org/testng/internal/ResultMap.java

package org.testng.internal;

import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import org.testng.IResultMap;
import org.testng.ITestNGMethod;
import org.testng.ITestResult;

public class ResultMap implements IResultMap {
  private Map<ITestResult, ITestNGMethod> m_map = new ConcurrentHashMap<ITestResult, ITestNGMethod>();

  public void addResult(ITestResult result, ITestNGMethod method) {
    m_map.put(result, method);
  }

  public Set<ITestResult> getResults(ITestNGMethod method) {
    Set<ITestResult> result = new HashSet<ITestResult>();
    
    for( ITestResult tr : m_map.keySet()) {
      if (m_map.get(tr).equals(method)) {
        result.add(tr);
      }
    }
    
    return result;
  }

  public Set<ITestResult> getAllResults() {
    return m_map.keySet();
  }

  public int size() {
    return m_map.size();
  }

  public Collection<ITestNGMethod> getAllMethods() {
    return m_map.values();
  }

}