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.RunListener;
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( RunListener reporter )
92      {
93  
94          if ( ignored != null )
95          {
96              reporter.testSkipped( createReportEntry() );
97              return;
98          }
99  
100         reporter.testStarting( createReportEntry() );
101         if ( output != null )
102         {
103             output.writeDetails( ( (ConsoleOutputReceiver) reporter ) );
104         }
105 
106         if ( testFailure != null )
107         {
108             reporter.testFailed( testFailure );
109         }
110         else if ( testError != null )
111         {
112             reporter.testError( testError );
113         }
114         else
115         {
116             reporter.testSucceeded( createReportEntry() );
117         }
118     }
119 
120     private ReportEntry createReportEntry()
121     {
122         return this.description;
123     }
124 
125     public void attachToThread()
126     {
127         TEST_METHOD.set( this );
128         ConsoleOutputReceiverForCurrentThread.set( this );
129 
130     }
131 
132     public static void detachFromCurrentThread()
133     {
134         TEST_METHOD.remove();
135         ConsoleOutputReceiverForCurrentThread.remove();
136     }
137 
138     public static TestMethod getThreadTestMethod()
139     {
140         return TEST_METHOD.get();
141     }
142 
143     public LogicalStream getLogicalStream()
144     {
145         if ( output == null )
146         {
147             output = new LogicalStream();
148         }
149         return output;
150     }
151 
152     public void writeTestOutput( byte[] buf, int off, int len, boolean stdout )
153     {
154         getLogicalStream().write( stdout, buf, off, len );
155     }
156 
157 }