blob: f9931e6e516ad2b63a69506410976497897495b5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
package org.testng.internal;
import org.testng.IResultMap;
import org.testng.ITestNGMethod;
import org.testng.ITestResult;
import org.testng.collections.Objects;
import java.util.Collection;
import java.util.HashSet;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
public class ResultMap implements IResultMap {
/**
*
*/
private static final long serialVersionUID = 80134376515999093L;
private Map<ITestResult, ITestNGMethod> m_map = new ConcurrentHashMap<>();
@Override
public void addResult(ITestResult result, ITestNGMethod method) {
m_map.put(result, method);
}
@Override
public Set<ITestResult> getResults(ITestNGMethod method) {
Set<ITestResult> result = new HashSet<>();
for (ITestResult tr : m_map.keySet()) {
if (m_map.get(tr).equals(method)) {
result.add(tr);
}
}
return result;
}
@Override
public void removeResult(ITestNGMethod m) {
for (Entry<ITestResult, ITestNGMethod> entry : m_map.entrySet()) {
if (entry.getValue().equals(m)) {
m_map.remove(entry.getKey());
return;
}
}
}
@Override
public void removeResult(ITestResult r) {
m_map.remove(r);
}
@Override
public Set<ITestResult> getAllResults() {
return m_map.keySet();
}
@Override
public int size() {
return m_map.size();
}
@Override
public Collection<ITestNGMethod> getAllMethods() {
return m_map.values();
}
@Override
public String toString() {
return Objects.toStringHelper(getClass())
.add("map", m_map)
.toString();
}
}
|