WebSocket client for VM service

tests/standalone/vmservice/test_helper.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 vmservice_test_helper;
 
 import 'dart:async';
 import 'dart:convert';
 import 'dart:io';
 import 'package:expect/expect.dart';
 
+abstract class ServiceWebSocketRequestHelper {
+  final String url;
+  final Completer _completer = new Completer();
+  WebSocket _socket;
+
+  ServiceWebSocketRequestHelper(this.url);
+
+  // Returns [this] when connected.
+  Future connect() {
+    return WebSocket.connect(url).then((ws) {
+      _socket = ws;
+      _socket.listen((message) {
+        var map = JSON.decode(message);
+        var response = JSON.decode(map['response']);
+        onResponse(map['seq'], response);
+      });
+      return this;
+    });
+  }
+
+  void complete() {
+    _completer.complete(this);
+  }
+
+  Future get completed => _completer.future;
+
+  // Must call complete.
+  void onResponse(var seq, Map response);
+  void runTest();
+
+  Future sendMessage(var seq, List<String> path) {
+    var map = {
+      'seq': seq,
+      'path': path
+    };
+    var message = JSON.encode(map);
+    _socket.add(message);
+    return _completer.future;
+  }
+}
+
 abstract class VmServiceRequestHelper {
   final Uri uri;
   final HttpClient client;
 
   VmServiceRequestHelper(String url) :
       uri = Uri.parse(url),
       client = new HttpClient();
 
   Future makeRequest() {
     print('** GET: $uri');
@@ skipping 70 matching lines @@
     return Process.start(dartExecutable,
                          ['--enable-vm-service:0', scriptPath]).then((p) {
 
       Completer completer = new Completer();
       process = p;
       var portNumber;
       var blank;
       var first = true;
       process.stdout.transform(UTF8.decoder)
                     .transform(new LineSplitter()).listen((line) {
-        if (line.startsWith('VmService listening on port ')) {
+        if (line.startsWith('VMService listening on port ')) {
           RegExp portExp = new RegExp(r"\d+");
           var port = portExp.stringMatch(line);
           portNumber = int.parse(port);
         }
         if (line == '') {
           // Received blank line.
           blank = true;
         }
         if (portNumber != null && blank == true && first == true) {
           completer.complete(portNumber);
@@ skipping 134 matching lines @@
       });
     });
     return Future.wait(requests).then((a) {
       a.forEach((FieldRequestHelper field) {
         fields[field.field['user_name']] = field.field;
       });
       return this;
     });
   }
 }