View Javadoc

1   package org.apache.maven.surefire.junitcore;
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 org.apache.maven.surefire.report.ConsoleOutputReceiver;
23  import org.apache.maven.surefire.report.ConsoleOutputReceiverForCurrentThread;
24  import org.apache.maven.surefire.report.ReportEntry;
25  import org.apache.maven.surefire.report.Reporter;
26  
27  /**
28   * Represents the test-state of a single test method that is run.
29   * <p/>
30   * Notes about thread safety: This instance is serially confined to 1-3 threads (construction, test-run, reporting),
31   * without any actual parallel access
32   */
33  class TestMethod
34      implements ConsoleOutputReceiver
35  {
36      private final ReportEntry description;
37  
38      private final long startTime;
39  
40      private long endTime;
41  
42      private volatile ReportEntry testFailure;
43  
44      private volatile ReportEntry testError;
45  
46      private volatile ReportEntry ignored;
47  
48      private static final InheritableThreadLocal<TestMethod> TEST_METHOD = new InheritableThreadLocal<TestMethod>();
49  
50      private volatile LogicalStream output;
51  
52      public TestMethod( ReportEntry description )
53      {
54          this.description = description;
55          startTime = System.currentTimeMillis();
56      }
57  
58      public void testFinished()
59      {
60          setEndTime();
61      }
62  
63      public void testIgnored( ReportEntry description )
64      {
65          ignored = description;
66          setEndTime();
67      }
68  
69      public void testFailure( ReportEntry failure )
70      {
71          this.testFailure = failure;
72      }
73  
74      public void testError( ReportEntry failure )
75      {
76          this.testError = failure;
77          setEndTime();
78      }
79  
80      private void setEndTime()
81      {
82          this.endTime = System.currentTimeMillis();
83      }
84  
85      public int getElapsed()
86      {
87          return (int) ( endTime - startTime );
88      }
89  
90  
91      public void replay( Reporter reporter )
92          throws Exception
93      {
94  
95          if ( ignored != null )
96          {
97              reporter.testSkipped( createReportEntry() );
98              return;
99          }
100 
101         reporter.testStarting( createReportEntry() );
102         if ( output != null )
103         {
104             output.writeDetails( reporter );
105         }
106 
107         if ( testFailure != null )
108         {
109             reporter.testFailed( testFailure, getStdout(), getStdErr() );
110         }
111         else if ( testError != null )
112         {
113             reporter.testError( testError, getStdout(), getStdErr() );
114         }
115         else
116         {
117             reporter.testSucceeded( createReportEntry() );
118         }
119     }
120 
121     private ReportEntry createReportEntry()
122     {
123         return this.description;
124     }
125 
126     public void attachToThread()
127     {
128         TEST_METHOD.set( this );
129         ConsoleOutputReceiverForCurrentThread.set( this );
130 
131     }
132 
133     public static void detachFromCurrentThread()
134     {
135         TEST_METHOD.remove();
136         ConsoleOutputReceiverForCurrentThread.remove();
137     }
138 
139     public static TestMethod getThreadTestMethod()
140     {
141         return TEST_METHOD.get();
142     }
143 
144     public LogicalStream getLogicalStream()
145     {
146         if ( output == null )
147         {
148             output = new LogicalStream();
149         }
150         return output;
151     }
152 
153     public void writeTestOutput( byte[] buf, int off, int len, boolean stdout )
154     {
155         getLogicalStream().write( stdout, buf, off, len );
156     }
157 
158     private String getStdout()
159     {
160         return output != null ? output.getOutput( true ) : "";
161     }
162 
163     private String getStdErr()
164     {
165         return output != null ? output.getOutput( false ) : "";
166     }
167 }