Expanding WebSocket test

tools/testing/dart/http_server.dart

 // Copyright (c) 2013, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 library http_server;
 
 import 'dart:async';
 import 'dart:io';
 import 'dart:isolate';
 import 'dart:uri';
 import 'test_suite.dart';  // For TestUtils.
 // TODO(efortuna): Rewrite to not use the args library and simply take an
 // expected number of arguments, so test.dart doesn't rely on the args library?
 // See discussion on https://codereview.chromium.org/11931025/.
 import 'vendored_pkg/args/args.dart';
 import 'utils.dart';
 
 
 /// Interface of the HTTP server:
 ///
 /// /echo: This will stream the data received in the request stream back
 ///        to the client.
 /// /root_dart/X: This will serve the corresponding file from the dart
 ///               directory (i.e. '$DartDirectory/X').
 /// /root_build/X: This will serve the corresponding file from the build
 ///                directory (i.e. '$BuildDirectory/X').
 /// /FOO/packages/BAR: This will serve the corresponding file from the packages
 ///                    directory (i.e. '$BuildDirectory/packages/BAR')
+/// /ws: This will upgrade the connection to a WebSocket connection and echo
+///      all data back to the client.
 ///
 /// In case a path does not refer to a file but rather to a directory, a
 /// directory listing will be displayed.
 
 const PREFIX_BUILDDIR = 'root_build';
 const PREFIX_DARTDIR = 'root_dart';
 
 // TODO(kustermann,ricow): We could change this to the following scheme:
 // http://host:port/root_packages/X -> $BuildDir/packages/X
 // Issue: 8368
@@ skipping 86 matching lines @@
     for (var server in _serverList) {
       server.close();
     }
   }
 
   Future _startHttpServer(String host, {int port: 0, int allowedPort: -1}) {
     return HttpServer.bind(host, port).then((HttpServer httpServer) {
       httpServer.listen((HttpRequest request) {
         if (request.uri.path == "/echo") {
           _handleEchoRequest(request, request.response);
+        } else if (request.uri.path == '/ws') {
+          _handleWebSocketRequest(request);
         } else {
           _handleFileOrDirectoryRequest(
               request, request.response, allowedPort);
         }
       },
       onError: (e) {
         DebugLogger.error('HttpServer: an error occured: $e');
       });
       _serverList.add(httpServer);
     });
@@ skipping 34 matching lines @@
   }
 
   void _handleEchoRequest(HttpRequest request, HttpResponse response) {
     response.headers.set("Access-Control-Allow-Origin", "*");
     request.pipe(response).catchError((e) {
       DebugLogger.warning(
           'HttpServer: error while closing the response stream: $e');
     });
   }
 
+  void _handleWebSocketRequest(HttpRequest request) {
+    WebSocketTransformer.upgrade(request).then((websocket) {
+      websocket.listen((data) {
+        websocket.send(data);
+        websocket.close();
+      }, onError: (e) {
+        DebugLogger.warning(
+            'HttpServer: error while echoing to WebSocket: $e');
+      });
+    }).catchError((e) {
+      DebugLogger.warning(
+          'HttpServer: error while transforming to WebSocket: $e');
+    });
+  }
+
   Path _getFilePathFromRequestPath(String urlRequestPath) {
     // Go to the top of the file to see an explanation of the URL path scheme.
     var requestPath = new Path(urlRequestPath.substring(1)).canonicalize();
     var pathSegments = requestPath.segments();
     if (pathSegments.length > 0) {
       var basePath;
       var relativePath;
       if (pathSegments[0] == PREFIX_BUILDDIR) {
         basePath = _buildDirectory;
         relativePath = new Path(
@@ skipping 143 matching lines @@
 class _Entry {
   final String name;
   final String displayName;
 
   _Entry(this.name, this.displayName);
 
   int compareTo(_Entry other) {
     return name.compareTo(other.name);
   }
 }