OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 package org.chromium.chrome.test; | 5 package org.chromium.chrome.test; |
6 | 6 |
7 import android.app.Activity; | |
8 import android.os.Bundle; | 7 import android.os.Bundle; |
9 import android.os.Environment; | |
10 import android.util.Log; | 8 import android.util.Log; |
11 | 9 |
12 import junit.framework.TestCase; | 10 import junit.framework.TestCase; |
13 | 11 |
14 import org.apache.http.HttpEntity; | |
15 import org.apache.http.HttpException; | |
16 import org.apache.http.HttpRequest; | |
17 import org.apache.http.HttpResponse; | |
18 import org.apache.http.HttpStatus; | |
19 import org.apache.http.HttpVersion; | |
20 import org.apache.http.RequestLine; | |
21 import org.apache.http.StatusLine; | |
22 import org.apache.http.entity.FileEntity; | |
23 import org.apache.http.message.BasicHttpResponse; | |
24 import org.apache.http.message.BasicStatusLine; | |
25 import org.apache.http.params.BasicHttpParams; | |
26 import org.apache.http.params.CoreProtocolPNames; | |
27 import org.apache.http.params.HttpParams; | |
28 | |
29 import org.chromium.base.test.BaseInstrumentationTestRunner; | 12 import org.chromium.base.test.BaseInstrumentationTestRunner; |
30 import org.chromium.base.test.BaseInstrumentationTestRunner.SkipCheck; | 13 import org.chromium.base.test.BaseInstrumentationTestRunner.SkipCheck; |
31 import org.chromium.base.test.BaseInstrumentationTestRunner.SkippingTestResult; | 14 import org.chromium.base.test.BaseInstrumentationTestRunner.SkippingTestResult; |
32 import org.chromium.chrome.browser.util.FeatureUtilities; | 15 import org.chromium.chrome.browser.util.FeatureUtilities; |
33 import org.chromium.chrome.test.util.DisableInTabbedMode; | 16 import org.chromium.chrome.test.util.DisableInTabbedMode; |
34 import org.chromium.net.test.BaseHttpTestServer; | |
35 | 17 |
36 import java.io.File; | |
37 import java.io.IOException; | |
38 import java.lang.reflect.Method; | 18 import java.lang.reflect.Method; |
39 import java.net.Socket; | |
40 import java.util.Collections; | |
41 import java.util.HashMap; | |
42 import java.util.Map; | |
43 | 19 |
44 | 20 |
45 /** | 21 /** |
46 * An Instrumentation test runner that optionally spawns a test HTTP server. | 22 * An Instrumentation test runner that optionally spawns a test HTTP server. |
47 * The server's root directory is the device's external storage directory. | 23 * The server's root directory is the device's external storage directory. |
48 */ | 24 */ |
49 public class ChromeInstrumentationTestRunner extends BaseInstrumentationTestRunn
er { | 25 public class ChromeInstrumentationTestRunner extends BaseInstrumentationTestRunn
er { |
50 | 26 |
51 private static final String TAG = "ChromeInstrumentationTestRunner"; | 27 private static final String TAG = "ChromeInstrumentationTestRunner"; |
52 | 28 |
53 public static final String EXTRA_DISABLE_TEST_HTTP_SERVER = | |
54 "org.chromium.chrome.test.ChromeInstrumentationTestRunner.DisableTes
tHttpServer"; | |
55 public static final String EXTRA_ENABLE_TEST_HTTP_SERVER = | |
56 "org.chromium.chrome.test.ChromeInstrumentationTestRunner.EnableTest
HttpServer"; | |
57 | |
58 // TODO(jbudorick): Default this to true once the bots are all enabling the
server via the | |
59 // flag. | |
60 private static final boolean DEFAULT_ENABLE_TEST_HTTP_SERVER = false; | |
61 | |
62 private Thread mHttpServerThread; | |
63 private TestHttpServer mHttpServer; | |
64 private boolean mRunTestHttpServer; | |
65 | |
66 @Override | 29 @Override |
67 public void onCreate(Bundle arguments) { | 30 public void onCreate(Bundle arguments) { |
68 if (arguments.containsKey(EXTRA_DISABLE_TEST_HTTP_SERVER)) { | 31 loadLibraries(); |
69 mRunTestHttpServer = false; | |
70 } else if (arguments.containsKey(EXTRA_ENABLE_TEST_HTTP_SERVER)) { | |
71 mRunTestHttpServer = true; | |
72 } else { | |
73 mRunTestHttpServer = DEFAULT_ENABLE_TEST_HTTP_SERVER; | |
74 } | |
75 | |
76 // InstrumentationTestRunner.onCreate() calls start() at the end, so we
call it last. | |
77 super.onCreate(arguments); | 32 super.onCreate(arguments); |
78 } | 33 } |
79 | 34 |
80 @Override | 35 @Override |
81 public void onStart() { | |
82 if (mRunTestHttpServer) { | |
83 try { | |
84 startTestHttpServer(); | |
85 } catch (IOException e) { | |
86 Log.e(TAG, "Failed to start HTTP test server", e); | |
87 finish(Activity.RESULT_CANCELED, null); | |
88 } | |
89 } | |
90 super.onStart(); | |
91 } | |
92 | |
93 private void startTestHttpServer() throws IOException { | |
94 mHttpServer = new TestHttpServer(Environment.getExternalStorageDirectory
(), 8000); | |
95 mHttpServerThread = new Thread(mHttpServer); | |
96 mHttpServerThread.start(); | |
97 mHttpServer.waitForServerToStart(); | |
98 } | |
99 | |
100 private static class TestHttpServer extends BaseHttpTestServer { | |
101 private static final String TAG = "ChromeInstrumentationTestRunner.TestH
ttpServer"; | |
102 | |
103 private final File mRootDirectory; | |
104 private final String mRootPath; | |
105 | |
106 private static final int ACCEPT_TIMEOUT_MS = 5000; | |
107 private static final String DEFAULT_CONTENT_TYPE = "application/octet-st
ream"; | |
108 private static final Map<String, String> EXTENSION_CONTENT_TYPE_MAP; | |
109 static { | |
110 Map<String, String> m = new HashMap<String, String>(); | |
111 m.put(".conf", "text/plain"); | |
112 m.put(".css", "text/css"); | |
113 m.put(".dtd", "text/xml"); | |
114 m.put(".gif", "image/gif"); | |
115 m.put(".htm", "text/html"); | |
116 m.put(".html", "text/html"); | |
117 m.put(".jpeg", "image/jpeg"); | |
118 m.put(".jpg", "image/jpeg"); | |
119 m.put(".js", "application/x-javascript"); | |
120 m.put(".log", "text/plain"); | |
121 m.put(".manifest", "text/cache-manifest"); | |
122 m.put(".mp4", "video/mp4"); | |
123 m.put(".png", "image/png"); | |
124 m.put(".svg", "image/svg+xml"); | |
125 m.put(".text", "text/plain"); | |
126 m.put(".txt", "text/plain"); | |
127 m.put(".xhtml", "application/xhtml+xml"); | |
128 m.put(".xhtmlmp", "application/vnd.wap.xhtml+xml"); | |
129 m.put(".xml", "text/xml"); | |
130 EXTENSION_CONTENT_TYPE_MAP = Collections.unmodifiableMap(m); | |
131 } | |
132 | |
133 public TestHttpServer(File rootDirectory, int port) throws IOException { | |
134 super(port, ACCEPT_TIMEOUT_MS); | |
135 mRootDirectory = rootDirectory; | |
136 mRootPath = mRootDirectory.getAbsolutePath(); | |
137 } | |
138 | |
139 @Override | |
140 protected boolean validateSocket(Socket sock) { | |
141 return sock.getInetAddress().isLoopbackAddress(); | |
142 } | |
143 | |
144 @Override | |
145 protected HttpParams getConnectionParams() { | |
146 HttpParams httpParams = new BasicHttpParams(); | |
147 httpParams.setParameter(CoreProtocolPNames.PROTOCOL_VERSION, HttpVer
sion.HTTP_1_0); | |
148 return httpParams; | |
149 } | |
150 | |
151 @Override | |
152 protected void handleGet(HttpRequest request, HttpResponseCallback callb
ack) | |
153 throws HttpException { | |
154 RequestLine requestLine = request.getRequestLine(); | |
155 | |
156 String requestPath = requestLine.getUri(); | |
157 if (requestPath.startsWith(File.separator)) { | |
158 requestPath = requestPath.substring(File.separator.length()); | |
159 } | |
160 File requestedFile = new File(mRootDirectory, requestPath); | |
161 String requestedPath = requestedFile.getAbsolutePath(); | |
162 | |
163 int status = HttpStatus.SC_INTERNAL_SERVER_ERROR; | |
164 String reason = ""; | |
165 HttpEntity entity = null; | |
166 if (!requestedPath.startsWith(mRootPath)) { | |
167 Log.w(TAG, "Client tried to request something outside of " + mRo
otPath + ": " | |
168 + requestedPath); | |
169 status = HttpStatus.SC_FORBIDDEN; | |
170 } else if (!requestedFile.exists()) { | |
171 Log.w(TAG, "Client requested non-existent file: " + requestedPat
h); | |
172 status = HttpStatus.SC_NOT_FOUND; | |
173 } else if (!requestedFile.isFile()) { | |
174 Log.w(TAG, "Client requested something that isn't a file: " + re
questedPath); | |
175 status = HttpStatus.SC_BAD_REQUEST; | |
176 reason = requestLine.getUri() + " is not a file."; | |
177 } else { | |
178 status = HttpStatus.SC_OK; | |
179 String contentType = null; | |
180 int extensionIndex = requestedPath.lastIndexOf('.'); | |
181 if (extensionIndex == -1) { | |
182 contentType = DEFAULT_CONTENT_TYPE; | |
183 } else { | |
184 String extension = requestedPath.substring(extensionIndex); | |
185 contentType = EXTENSION_CONTENT_TYPE_MAP.get(extension); | |
186 if (contentType == null) { | |
187 Log.w(TAG, "Unrecognized extension: " + extension); | |
188 contentType = DEFAULT_CONTENT_TYPE; | |
189 } | |
190 } | |
191 entity = new FileEntity(requestedFile, contentType); | |
192 } | |
193 | |
194 StatusLine statusLine = new BasicStatusLine(HttpVersion.HTTP_1_0, st
atus, reason); | |
195 HttpResponse response = new BasicHttpResponse(statusLine); | |
196 if (entity != null) { | |
197 response.setEntity(entity); | |
198 } | |
199 callback.onResponse(response); | |
200 } | |
201 } | |
202 | |
203 @Override | |
204 public void onDestroy() { | |
205 super.onDestroy(); | |
206 if (mRunTestHttpServer) stopTestHttpServer(); | |
207 } | |
208 | |
209 private void stopTestHttpServer() { | |
210 mHttpServer.stop(); | |
211 try { | |
212 mHttpServerThread.join(); | |
213 } catch (InterruptedException e) { | |
214 Log.w(TAG, "Interrupted while joining for TestHttpServer thread: " +
e.toString()); | |
215 } | |
216 } | |
217 | |
218 @Override | |
219 protected void addSkipChecks(SkippingTestResult result) { | 36 protected void addSkipChecks(SkippingTestResult result) { |
220 super.addSkipChecks(result); | 37 super.addSkipChecks(result); |
221 result.addSkipCheck(new DisableInTabbedModeSkipCheck()); | 38 result.addSkipCheck(new DisableInTabbedModeSkipCheck()); |
222 } | 39 } |
223 | 40 |
224 /** | 41 /** |
225 * Checks for tests that should only run in document mode. | 42 * Checks for tests that should only run in document mode. |
226 */ | 43 */ |
227 private class DisableInTabbedModeSkipCheck implements SkipCheck { | 44 private class DisableInTabbedModeSkipCheck implements SkipCheck { |
228 | 45 |
(...skipping 19 matching lines...) Expand all Loading... |
248 } | 65 } |
249 } | 66 } |
250 } catch (NoSuchMethodException e) { | 67 } catch (NoSuchMethodException e) { |
251 Log.e(TAG, "Couldn't find test method: " + e.toString()); | 68 Log.e(TAG, "Couldn't find test method: " + e.toString()); |
252 } | 69 } |
253 | 70 |
254 return false; | 71 return false; |
255 } | 72 } |
256 } | 73 } |
257 } | 74 } |
OLD | NEW |