1 package junit.runner; 2 3 import java.io.BufferedReader; 4 import java.io.File; 5 import java.io.FileInputStream; 6 import java.io.FileOutputStream; 7 import java.io.IOException; 8 import java.io.InputStream; 9 import java.io.PrintWriter; 10 import java.io.StringReader; 11 import java.io.StringWriter; 12 import java.lang.reflect.InvocationTargetException; 13 import java.lang.reflect.Method; 14 import java.lang.reflect.Modifier; 15 import java.text.NumberFormat; 16 import java.util.Properties; 17 18 import junit.framework.AssertionFailedError; 19 import junit.framework.Test; 20 import junit.framework.TestListener; 21 import junit.framework.TestSuite; 22 23 /** 24 * Base class for all test runners. 25 * This class was born live on stage in Sardinia during XP2000. 26 */ 27 public abstract class BaseTestRunner implements TestListener { 28 public static final String SUITE_METHODNAME= "suite"; 29 30 private static Properties fPreferences; 31 static int fgMaxMessageLength= 500; 32 static boolean fgFilterStack= true; 33 boolean fLoading= true; 34 35 /* 36 * Implementation of TestListener 37 */ startTest(Test test)38 public synchronized void startTest(Test test) { 39 testStarted(test.toString()); 40 } 41 setPreferences(Properties preferences)42 protected static void setPreferences(Properties preferences) { 43 fPreferences= preferences; 44 } 45 getPreferences()46 protected static Properties getPreferences() { 47 if (fPreferences == null) { 48 fPreferences= new Properties(); 49 fPreferences.put("loading", "true"); 50 fPreferences.put("filterstack", "true"); 51 readPreferences(); 52 } 53 return fPreferences; 54 } 55 savePreferences()56 public static void savePreferences() throws IOException { 57 FileOutputStream fos= new FileOutputStream(getPreferencesFile()); 58 try { 59 getPreferences().store(fos, ""); 60 } finally { 61 fos.close(); 62 } 63 } 64 65 // android-changed remove 'static' qualifier for API compatibility setPreference(String key, String value)66 public void setPreference(String key, String value) { 67 getPreferences().put(key, value); 68 } 69 endTest(Test test)70 public synchronized void endTest(Test test) { 71 testEnded(test.toString()); 72 } 73 addError(final Test test, final Throwable t)74 public synchronized void addError(final Test test, final Throwable t) { 75 testFailed(TestRunListener.STATUS_ERROR, test, t); 76 } 77 addFailure(final Test test, final AssertionFailedError t)78 public synchronized void addFailure(final Test test, final AssertionFailedError t) { 79 testFailed(TestRunListener.STATUS_FAILURE, test, t); 80 } 81 82 // TestRunListener implementation 83 testStarted(String testName)84 public abstract void testStarted(String testName); 85 testEnded(String testName)86 public abstract void testEnded(String testName); 87 testFailed(int status, Test test, Throwable t)88 public abstract void testFailed(int status, Test test, Throwable t); 89 90 /** 91 * Returns the Test corresponding to the given suite. This is 92 * a template method, subclasses override runFailed(), clearStatus(). 93 */ getTest(String suiteClassName)94 public Test getTest(String suiteClassName) { 95 if (suiteClassName.length() <= 0) { 96 clearStatus(); 97 return null; 98 } 99 Class<?> testClass= null; 100 try { 101 testClass= loadSuiteClass(suiteClassName); 102 } catch (ClassNotFoundException e) { 103 String clazz= e.getMessage(); 104 if (clazz == null) 105 clazz= suiteClassName; 106 runFailed("Class not found \""+clazz+"\""); 107 return null; 108 } catch(Exception e) { 109 runFailed("Error: "+e.toString()); 110 return null; 111 } 112 Method suiteMethod= null; 113 try { 114 suiteMethod= testClass.getMethod(SUITE_METHODNAME, new Class[0]); 115 } catch(Exception e) { 116 // try to extract a test suite automatically 117 clearStatus(); 118 return new TestSuite(testClass); 119 } 120 if (! Modifier.isStatic(suiteMethod.getModifiers())) { 121 runFailed("Suite() method must be static"); 122 return null; 123 } 124 Test test= null; 125 try { 126 test= (Test)suiteMethod.invoke(null, (Object[])new Class[0]); // static method 127 if (test == null) 128 return test; 129 } 130 catch (InvocationTargetException e) { 131 runFailed("Failed to invoke suite():" + e.getTargetException().toString()); 132 return null; 133 } 134 catch (IllegalAccessException e) { 135 runFailed("Failed to invoke suite():" + e.toString()); 136 return null; 137 } 138 139 clearStatus(); 140 return test; 141 } 142 143 /** 144 * Returns the formatted string of the elapsed time. 145 */ elapsedTimeAsString(long runTime)146 public String elapsedTimeAsString(long runTime) { 147 return NumberFormat.getInstance().format((double)runTime/1000); 148 } 149 150 /** 151 * Processes the command line arguments and 152 * returns the name of the suite class to run or null 153 */ processArguments(String[] args)154 protected String processArguments(String[] args) { 155 String suiteName= null; 156 for (int i= 0; i < args.length; i++) { 157 if (args[i].equals("-noloading")) { 158 setLoading(false); 159 } else if (args[i].equals("-nofilterstack")) { 160 fgFilterStack= false; 161 } else if (args[i].equals("-c")) { 162 if (args.length > i+1) 163 suiteName= extractClassName(args[i+1]); 164 else 165 System.out.println("Missing Test class name"); 166 i++; 167 } else { 168 suiteName= args[i]; 169 } 170 } 171 return suiteName; 172 } 173 174 /** 175 * Sets the loading behaviour of the test runner 176 */ setLoading(boolean enable)177 public void setLoading(boolean enable) { 178 fLoading= enable; 179 } 180 /** 181 * Extract the class name from a String in VA/Java style 182 */ extractClassName(String className)183 public String extractClassName(String className) { 184 if(className.startsWith("Default package for")) 185 return className.substring(className.lastIndexOf(".")+1); 186 return className; 187 } 188 189 /** 190 * Truncates a String to the maximum length. 191 */ truncate(String s)192 public static String truncate(String s) { 193 if (fgMaxMessageLength != -1 && s.length() > fgMaxMessageLength) 194 s= s.substring(0, fgMaxMessageLength)+"..."; 195 return s; 196 } 197 198 /** 199 * Override to define how to handle a failed loading of 200 * a test suite. 201 */ runFailed(String message)202 protected abstract void runFailed(String message); 203 204 // BEGIN android-changed - add back getLoader() for API compatibility 205 /** 206 * Returns the loader to be used. 207 * 208 * @deprecated not present in JUnit4.10 209 */ getLoader()210 public TestSuiteLoader getLoader() { 211 return new StandardTestSuiteLoader(); 212 } 213 // END android-changed 214 215 /** 216 * Returns the loaded Class for a suite name. 217 */ loadSuiteClass(String suiteClassName)218 protected Class<?> loadSuiteClass(String suiteClassName) throws ClassNotFoundException { 219 return Class.forName(suiteClassName); 220 } 221 222 /** 223 * Clears the status message. 224 */ clearStatus()225 protected void clearStatus() { // Belongs in the GUI TestRunner class 226 } 227 useReloadingTestSuiteLoader()228 protected boolean useReloadingTestSuiteLoader() { 229 return getPreference("loading").equals("true") && fLoading; 230 } 231 getPreferencesFile()232 private static File getPreferencesFile() { 233 String home= System.getProperty("user.home"); 234 return new File(home, "junit.properties"); 235 } 236 readPreferences()237 private static void readPreferences() { 238 InputStream is= null; 239 try { 240 is= new FileInputStream(getPreferencesFile()); 241 setPreferences(new Properties(getPreferences())); 242 getPreferences().load(is); 243 } catch (IOException e) { 244 try { 245 if (is != null) 246 is.close(); 247 } catch (IOException e1) { 248 } 249 } 250 } 251 getPreference(String key)252 public static String getPreference(String key) { 253 return getPreferences().getProperty(key); 254 } 255 getPreference(String key, int dflt)256 public static int getPreference(String key, int dflt) { 257 String value= getPreference(key); 258 int intValue= dflt; 259 if (value == null) 260 return intValue; 261 try { 262 intValue= Integer.parseInt(value); 263 } catch (NumberFormatException ne) { 264 } 265 return intValue; 266 } 267 268 /** 269 * Returns a filtered stack trace 270 */ getFilteredTrace(Throwable t)271 public static String getFilteredTrace(Throwable t) { 272 StringWriter stringWriter= new StringWriter(); 273 PrintWriter writer= new PrintWriter(stringWriter); 274 t.printStackTrace(writer); 275 StringBuffer buffer= stringWriter.getBuffer(); 276 String trace= buffer.toString(); 277 return BaseTestRunner.getFilteredTrace(trace); 278 } 279 280 // BEGIN android-changed - add back this method for API compatibility 281 /** @deprecated not present in JUnit4.10 */ inVAJava()282 public static boolean inVAJava() { 283 return false; 284 } 285 // END android-changed 286 287 /** 288 * Filters stack frames from internal JUnit classes 289 */ getFilteredTrace(String stack)290 public static String getFilteredTrace(String stack) { 291 if (showStackRaw()) 292 return stack; 293 294 StringWriter sw= new StringWriter(); 295 PrintWriter pw= new PrintWriter(sw); 296 StringReader sr= new StringReader(stack); 297 // BEGIN android-changed 298 // Use a sensible default buffer size 299 BufferedReader br= new BufferedReader(sr, 1000); 300 // END android-changed 301 302 String line; 303 try { 304 while ((line= br.readLine()) != null) { 305 if (!filterLine(line)) 306 pw.println(line); 307 } 308 } catch (Exception IOException) { 309 return stack; // return the stack unfiltered 310 } 311 return sw.toString(); 312 } 313 showStackRaw()314 protected static boolean showStackRaw() { 315 return !getPreference("filterstack").equals("true") || fgFilterStack == false; 316 } 317 filterLine(String line)318 static boolean filterLine(String line) { 319 String[] patterns= new String[] { 320 "junit.framework.TestCase", 321 "junit.framework.TestResult", 322 "junit.framework.TestSuite", 323 "junit.framework.Assert.", // don't filter AssertionFailure 324 "junit.swingui.TestRunner", 325 "junit.awtui.TestRunner", 326 "junit.textui.TestRunner", 327 "java.lang.reflect.Method.invoke(" 328 }; 329 for (int i= 0; i < patterns.length; i++) { 330 if (line.indexOf(patterns[i]) > 0) 331 return true; 332 } 333 return false; 334 } 335 336 static { 337 fgMaxMessageLength= getPreference("maxmessage", fgMaxMessageLength); 338 } 339 340 } 341