1 package baih;
  2
 3 import java.io.File;  4 import java.io.FileWriter;  5 import java.io.IOException;  6 import java.io.Writer;  7  8 import org.xmlpull.v1.XmlPullParserFactory;  9 import org.xmlpull.v1.XmlSerializer;  10  11 import android.content.Context;  12 import android.os.Bundle;  13 import android.os.Environment;  14  15 /**  16  * This test runner creates a TEST-all.xml in the files directory of the application under test. The output is compatible with that of the junitreport ant task, the format  17  * that is understood by Hudson. Currently this implementation does not implement the all aspects of the junitreport format, but enough for Hudson to parse the test results.  18 */  19 public class InstrumentationTestRunner extends android.test.InstrumentationTestRunner {  20 private Writer mWriter;  21 private XmlSerializer mTestSuiteSerializer;  22 private long mTestStarted;  23 private static final String JUNIT_XML_FILE = "TEST-all.xml";  24  25  26  @Override  27 public void onStart() {  28 try{  29 File fileRobo = new File(getTestResultDir(getTargetContext()));  30 if(!fileRobo.exists()){  31  fileRobo.mkdir();  32  }  33 if(isSDCardAvaliable()){  34 File resultFile = new File(getTestResultDir(getTargetContext()),JUNIT_XML_FILE);  35 startJUnitOutput(new FileWriter(resultFile));  36 }else{  37 startJUnitOutput(new FileWriter(new File(getTargetContext().getFilesDir(), JUNIT_XML_FILE)));  38  }  39  }  40 catch(IOException e){  41 throw new RuntimeException(e);  42  }  43 super.onStart();  44  }  45  46 void startJUnitOutput(Writer writer) {  47 try {  48 mWriter = writer;  49 mTestSuiteSerializer = newSerializer(mWriter);  50 mTestSuiteSerializer.startDocument(null, null);  51 mTestSuiteSerializer.startTag(null, "testsuites");  52 mTestSuiteSerializer.startTag(null, "testsuite");  53 } catch (Exception e) {  54 throw new RuntimeException(e);  55  }  56  }  57  58 /**  59  * 判断SD卡是否存在  60  * @return  61 */  62 private boolean isSDCardAvaliable(){  63 return Environment.getExternalStorageState()  64  .equals(Environment.MEDIA_MOUNTED);  65  }  66  67 /**  68  * 获取测试结果报告文件所在的路径  69  * @param context 被测工程的context  70  * @return 返回测试结果报告文件所在的路径  71 */  72 private String getTestResultDir(Context context){  73 String packageName = "/" + "robotium";  74 String filepath = context.getCacheDir().getPath() + packageName;  75  76 if(android.os.Build.VERSION.SDK_INT < 8){  77 if(isSDCardAvaliable()){ 78 filepath = Environment.getExternalStorageDirectory().getAbsolutePath()+ packageName; 79 } 80 }else{ 81 if(isSDCardAvaliable()){ 82 filepath = Environment.getExternalStorageDirectory().getAbsolutePath()+ packageName; 83 } 84 } 85 return filepath; 86 } 87 88 private XmlSerializer newSerializer(Writer writer) { 89 try { 90 XmlPullParserFactory pf = XmlPullParserFactory.newInstance(); 91 XmlSerializer serializer = pf.newSerializer(); 92 serializer.setOutput(writer); 93 return serializer; 94 } catch (Exception e) { 95 throw new RuntimeException(e); 96 } 97 } 98 99 @Override 100 public void sendStatus(int resultCode, Bundle results) { 101 super.sendStatus(resultCode, results); 102 switch (resultCode) { 103 case REPORT_VALUE_RESULT_ERROR: 104 case REPORT_VALUE_RESULT_FAILURE: 105 case REPORT_VALUE_RESULT_OK: 106 try { 107 recordTestResult(resultCode, results); 108 } catch (IOException e) { 109 throw new RuntimeException(e); 110 } 111 break; 112 case REPORT_VALUE_RESULT_START: 113 recordTestStart(results); 114 default: 115 break; 116 } 117 } 118 119 void recordTestStart(Bundle results) { 120 mTestStarted = System.currentTimeMillis(); 121 } 122 123 void recordTestResult(int resultCode, Bundle results) throws IOException { 124 float time = (System.currentTimeMillis() - mTestStarted) / 1000.0f; 125 String className = results.getString(REPORT_KEY_NAME_CLASS); 126 String testMethod = results.getString(REPORT_KEY_NAME_TEST); 127 String stack = results.getString(REPORT_KEY_STACK); 128 int current = results.getInt(REPORT_KEY_NUM_CURRENT); 129 int total = results.getInt(REPORT_KEY_NUM_TOTAL); 130 131 mTestSuiteSerializer.startTag(null, "testcase"); 132 mTestSuiteSerializer.attribute(null, "classname", className); 133 mTestSuiteSerializer.attribute(null, "name", testMethod); 134 135 if (resultCode != REPORT_VALUE_RESULT_OK) { 136 mTestSuiteSerializer.startTag(null, "failure"); 137 if (stack != null) { 138 String reason = stack.substring(0, stack.indexOf('\n')); 139 String message = ""; 140 int index = reason.indexOf(':'); 141 if (index > -1) { 142 message = reason.substring(index+1); 143 reason = reason.substring(0, index); 144 } 145 mTestSuiteSerializer.attribute(null, "message", message); 146 mTestSuiteSerializer.attribute(null, "type", reason); 147 mTestSuiteSerializer.text(stack); 148 } 149 mTestSuiteSerializer.endTag(null, "failure"); 150 } else { 151 mTestSuiteSerializer.attribute(null, "time", String.format("%.3f", time)); 152 } 153 mTestSuiteSerializer.endTag(null, "testcase"); 154 if (current == total) { 155 mTestSuiteSerializer.startTag(null, "system-out"); 156 mTestSuiteSerializer.endTag(null, "system-out"); 157 mTestSuiteSerializer.startTag(null, "system-err"); 158 mTestSuiteSerializer.endTag(null, "system-err"); 159 mTestSuiteSerializer.endTag(null, "testsuite"); 160 mTestSuiteSerializer.flush(); 161 } 162 } 163 164 @Override 165 public void finish(int resultCode, Bundle results) { 166 endTestSuites(); 167 super.finish(resultCode, results); 168 } 169 170 void endTestSuites() { 171 try { 172 mTestSuiteSerializer.endTag(null, "testsuites"); 173 mTestSuiteSerializer.endDocument(); 174 mTestSuiteSerializer.flush(); 175 mWriter.flush(); 176 mWriter.close(); 177 } catch (IOException e) { 178 throw new RuntimeException(e); 179 } 180 } 181 }