Chromium Code Reviews| Index: tools/testing/dart/http_server.dart |
| diff --git a/tools/testing/dart/http_server.dart b/tools/testing/dart/http_server.dart |
| index 8d233a83a0000ed525aa73635bb8b3b26d4d3194..c92b0c161c2eb8885174f11078f8ed80254f950e 100644 |
| --- a/tools/testing/dart/http_server.dart |
| +++ b/tools/testing/dart/http_server.dart |
| @@ -7,51 +7,137 @@ library http_server; |
| import 'dart:io'; |
| import 'dart:isolate'; |
| import 'test_suite.dart'; // For TestUtils. |
| +import '../../../pkg/args/lib/args.dart'; |
|
ahe
2013/01/16 13:30:37
This breaks test.dart in a subtle, but severe way.
|
| -HttpServer _httpServer; |
| - |
| -void startHttpServer(String host, int port) { |
| - var basePath = TestUtils.dartDir(); |
| - _httpServer = new HttpServer(); |
| - _httpServer.onError = (e) { |
| - // Consider errors in the builtin http server fatal. |
| - // Intead of just throwing the exception we print |
| - // a message that makes it clearer what happened. |
| - print('Test http server error: $e'); |
| - exit(1); |
| - }; |
| - _httpServer.defaultRequestHandler = (request, resp) { |
| - var requestPath = new Path(request.path).canonicalize(); |
| - if (!requestPath.isAbsolute) { |
| - resp.statusCode = HttpStatus.NOT_FOUND; |
| - resp.outputStream.close(); |
| - } else { |
| - var path = basePath; |
| - requestPath.segments().forEach((s) => path = path.append(s)); |
| +main() { |
| + /** Convenience method for local testing. */ |
| + var parser = new ArgParser(); |
| + parser.addOption('port', abbr: 'p', |
| + help: 'The main server port we wish to respond to requests.', |
| + defaultsTo: '0'); |
| + parser.addOption('crossOriginPort', abbr: 'c', |
| + help: 'A different port that accepts request from the main server port.', |
| + defaultsTo: '0'); |
| + parser.addOption('mode', abbr: 'm', help: 'Testing mode.', |
| + defaultsTo: 'release'); |
| + parser.addOption('arch', abbr: 'a', help: 'Testing architecture.', |
| + defaultsTo: 'ia32'); |
| + parser.addFlag('help', abbr: 'h', negatable: false, |
| + help: 'Print this usage information.'); |
| + var args = parser.parse(new Options().arguments); |
| + if (args['help']) { |
| + print(parser.getUsage()); |
| + } else { |
| + // Pretend we're running test.dart so that TestUtils doesn't get confused |
| + // about the "current directory." This is only used if we're trying to run |
| + // this file independently for local testing. |
| + TestUtils.testScriptPath = new Path.fromNative( |
| + new Options().script).directoryPath.join( |
| + new Path('../../test.dart')).canonicalize().toNativePath(); |
| + TestingServerRunner.setPackageRootDir({'mode': args['mode'], |
| + 'arch': args['arch'], 'system': Platform.operatingSystem, |
| + 'build_directory': ''}); |
| + |
| + TestingServerRunner.startHttpServer('127.0.0.1', |
| + port: int.parse(args['port'])); |
| + print('Server listening on port ' |
| + '${TestingServerRunner.serverList[0].port}.'); |
| + TestingServerRunner.startHttpServer('127.0.0.1', |
| + allowedPort: TestingServerRunner.serverList[0].port, port: |
| + int.parse(args['crossOriginPort'])); |
| + print( |
| + 'Server listening on port ${TestingServerRunner.serverList[1].port}.'); |
| + } |
| +} |
| +/** |
| + * Runs a set of servers that are initialized specifically for the needs of our |
| + * test framework, such as dealing with package-root. |
| + */ |
| +class TestingServerRunner { |
| + static List serverList = []; |
| + static Path _packageRootDir = null; |
| + |
| + // Added as a getter so that the function will be called again each time the |
| + // default request handler closure is executed. |
| + static Path get packageRootDir => _packageRootDir; |
| + |
| + static setPackageRootDir(Map configuration) { |
| + _packageRootDir = TestUtils.currentWorkingDirectory.join( |
| + new Path(TestUtils.buildDir(configuration))); |
| + } |
| + |
| + static startHttpServer(String host, {int allowedPort:-1, int port: 0}) { |
| + var basePath = TestUtils.dartDir(); |
| + var httpServer = new HttpServer(); |
| + var packagesDirName = 'packages'; |
| + httpServer.onError = (e) { |
| + // Consider errors in the builtin http server fatal. |
| + // Intead of just throwing the exception we print |
| + // a message that makes it clearer what happened. |
| + print('Test http server error: $e'); |
| + exit(1); |
| + }; |
| + httpServer.defaultRequestHandler = (request, resp) { |
| + var requestPath = new Path(request.path.substring(1)).canonicalize(); |
| + var path = basePath.join(requestPath); |
| var file = new File(path.toNativePath()); |
| + |
| + if (requestPath.segments().contains(packagesDirName)) { |
| + // Essentially implement the packages path rewriting, so we don't have |
| + // to pass environment variables to the browsers. |
| + var requestPathStr = requestPath.toNativePath().substring( |
| + requestPath.toNativePath().indexOf(packagesDirName)); |
| + path = packageRootDir.append(requestPathStr); |
| + file = new File(path.toNativePath()); |
| + } |
| file.exists().then((exists) { |
| if (exists) { |
| - // Allow loading from localhost in browsers. |
| - resp.headers.set("Access-Control-Allow-Origin", "*"); |
| + if (allowedPort != -1) { |
| + // Allow loading from localhost:$allowedPort in browsers. |
| + resp.headers.set("Access-Control-Allow-Origin", |
| + "http://127.0.0.1:${allowedPort+1}"); |
| + resp.headers.set('Access-Control-Allow-Credentials', 'true'); |
| + } else { |
| + // No allowedPort specified. Allow from anywhere (but cross-origin |
| + // requests *with credentials* will fail because you can't use "*"). |
| + resp.headers.set("Access-Control-Allow-Origin", "*"); |
| + } |
| + if (path.toNativePath().endsWith('.html')) { |
| + resp.headers.set('Content-Type', 'text/html'); |
| + } else if (path.toNativePath().endsWith('.js')) { |
| + resp.headers.set('Content-Type', 'application/javascript'); |
| + } else if (path.toNativePath().endsWith('.dart')) { |
| + resp.headers.set('Content-Type', 'application/dart'); |
| + } |
| file.openInputStream().pipe(resp.outputStream); |
| } else { |
| resp.statusCode = HttpStatus.NOT_FOUND; |
| - resp.outputStream.close(); |
| + try { |
| + resp.outputStream.close(); |
| + } catch (e) { |
| + if (e is StreamException) { |
| + print('Test http_server error closing the response stream: $e'); |
| + } else { |
| + throw e; |
| + } |
| + } |
| } |
| }); |
| - } |
| - }; |
| - // Echos back the contents of the request as the response data. |
| - _httpServer.addRequestHandler((req) => req.path == "/echo", (request, resp) { |
| - resp.headers.set("Access-Control-Allow-Origin", "*"); |
| + }; |
| - request.inputStream.pipe(resp.outputStream); |
| - }); |
| + // Echos back the contents of the request as the response data. |
| + httpServer.addRequestHandler((req) => req.path == "/echo", (request, resp) { |
| + resp.headers.set("Access-Control-Allow-Origin", "*"); |
| - _httpServer.listen(host, port); |
| -} |
| + request.inputStream.pipe(resp.outputStream); |
| + }); |
| + |
| + httpServer.listen(host, port); |
| + serverList.add(httpServer); |
| + } |
| -terminateHttpServer() { |
| - if (_httpServer != null) _httpServer.close(); |
| + static terminateHttpServers() { |
| + for (var server in serverList) server.close(); |
| + } |
| } |