1 package org.apache.maven.surefire.junitcore;
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 import org.apache.maven.surefire.report.ReportEntry;
23 import org.apache.maven.surefire.report.RunListener;
24 import org.apache.maven.surefire.report.SimpleReportEntry;
25 import org.apache.maven.surefire.report.TestSetReportEntry;
26
27 import java.util.Collection;
28 import java.util.Collections;
29 import java.util.Map;
30 import java.util.concurrent.ConcurrentLinkedQueue;
31 import java.util.concurrent.atomic.AtomicBoolean;
32 import java.util.concurrent.atomic.AtomicInteger;
33
34 import static org.apache.maven.surefire.util.internal.ObjectUtils.systemProps;
35
36
37
38
39 public class TestSet
40 {
41 private static final InheritableThreadLocal<TestSet> TEST_SET = new InheritableThreadLocal<TestSet>();
42
43 private final String testClassName;
44
45 private final Collection<TestMethod> testMethods = new ConcurrentLinkedQueue<TestMethod>();
46
47 private final AtomicBoolean played = new AtomicBoolean();
48
49 private final AtomicInteger numberOfCompletedChildren = new AtomicInteger();
50
51
52
53
54 private final AtomicInteger numberOfTests = new AtomicInteger();
55
56 private volatile boolean allScheduled;
57
58 public TestSet( String testClassName )
59 {
60 this.testClassName = testClassName;
61 }
62
63 public void replay( RunListener target )
64 {
65 if ( played.compareAndSet( false, true ) )
66 {
67 try
68 {
69 TestSetReportEntry report = createReportEntryStarted();
70
71 target.testSetStarting( report );
72
73 long startTile = 0;
74 long endTime = 0;
75 for ( TestMethod testMethod : testMethods )
76 {
77 if ( startTile == 0 || testMethod.getStartTime() < startTile )
78 {
79 startTile = testMethod.getStartTime();
80 }
81
82 if ( endTime == 0 || testMethod.getEndTime() > endTime )
83 {
84 endTime = testMethod.getEndTime();
85 }
86
87 testMethod.replay( target );
88 }
89
90 int elapsed = (int) ( endTime - startTile );
91
92 report = createReportEntryCompleted( elapsed );
93
94 target.testSetCompleted( report );
95 }
96 catch ( Exception e )
97 {
98 throw new RuntimeException( e );
99 }
100 }
101 }
102
103 public TestMethod createThreadAttachedTestMethod( ReportEntry description )
104 {
105 TestMethod testMethod = new TestMethod( description, this );
106 addTestMethod( testMethod );
107 testMethod.attachToThread();
108 return testMethod;
109 }
110
111 private TestSetReportEntry createReportEntryStarted()
112 {
113 return createReportEntry( null, Collections.<String, String>emptyMap() );
114 }
115
116 private TestSetReportEntry createReportEntryCompleted( int elapsed )
117 {
118 return createReportEntry( elapsed, systemProps() );
119 }
120
121 private TestSetReportEntry createReportEntry( Integer elapsed, Map<String, String> systemProps )
122 {
123 return new SimpleReportEntry( testClassName, testClassName, null, elapsed, systemProps );
124 }
125
126 public void incrementTestMethodCount()
127 {
128 numberOfTests.incrementAndGet();
129 }
130
131 private void addTestMethod( TestMethod testMethod )
132 {
133 testMethods.add( testMethod );
134 }
135
136 public void incrementFinishedTests( RunListener reporterManager, boolean reportImmediately )
137 {
138 numberOfCompletedChildren.incrementAndGet();
139 if ( allScheduled && isAllTestsDone() && reportImmediately )
140 {
141 replay( reporterManager );
142 }
143 }
144
145 public void setAllScheduled( RunListener reporterManager )
146 {
147 allScheduled = true;
148 if ( isAllTestsDone() )
149 {
150 replay( reporterManager );
151 }
152 }
153
154 private boolean isAllTestsDone()
155 {
156 return numberOfTests.get() == numberOfCompletedChildren.get();
157 }
158
159 public void attachToThread()
160 {
161 TEST_SET.set( this );
162 }
163
164 public static TestSet getThreadTestSet()
165 {
166 return TEST_SET.get();
167 }
168 }