Make sure to process WebSocket frames when closing a WebSocket

tests/standalone/io/web_socket_test.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.
 //
 // VMOptions=
 // VMOptions=--short_socket_read
 // VMOptions=--short_socket_write
 // VMOptions=--short_socket_read --short_socket_write
 
 import "dart:async";
@@ skipping 155 matching lines @@
               Expect.listEquals(originalMessage, message);
               webSocket.close();
             },
             onDone: server.close);
         webSocket.add(originalMessage);
       });
     });
   }
 
 
+  void testCloseNoListen() {
+    createServer().then((server) {
+      server.transform(new WebSocketTransformer()).listen((webSocket) {
+        server.close();
+        webSocket.close();
+      });
+
+      createClient(server.port).then((webSocket) {
+        webSocket.close();
+      });
+    });
+  }
+
+
+  void testListenAfterClose() {
+    createServer().then((server) {
+      server.transform(new WebSocketTransformer()).listen((webSocket) {
+        server.close();
+        webSocket.close();
+        Expect.throws(() => webSocket.drain());
+      });
+
+      createClient(server.port).then((webSocket) {
+        webSocket.close();
+        Expect.throws(() => webSocket.drain());
+      });
+    });
+  }
+
+
   void testDoubleCloseClient() {
     createServer().then((server) {
       server.transform(new WebSocketTransformer()).listen((webSocket) {
         server.close();
         webSocket.listen((_) { }, onDone: webSocket.close);
       });
 
       createClient(server.port).then((webSocket) {
           webSocket.listen((_) { }, onDone: webSocket.close);
           webSocket.close();
@@ skipping 15 matching lines @@
         });
     });
   }
 
 
   void testImmediateCloseServer() {
     createServer().then((server) {
       server.listen((request) {
         WebSocketTransformer.upgrade(request)
             .then((webSocket) {
-              webSocket.close();
               webSocket.listen(
                   (_) { Expect.fail("Unexpected message"); },
                   onDone: server.close);
+              webSocket.close();
             });
       });
 
       createClient(server.port).then((webSocket) {
           webSocket.listen(
               (_) { Expect.fail("Unexpected message"); },
               onDone: webSocket.close);
         });
     });
   }
 
 
   void testImmediateCloseClient() {
     createServer().then((server) {
       server.listen((request) {
         WebSocketTransformer.upgrade(request)
             .then((webSocket) {
               webSocket.listen(
                   (_) { Expect.fail("Unexpected message"); },
                   onDone: () {
                 server.close();
                 webSocket.close();
               });
             });
       });
 
       createClient(server.port).then((webSocket) {
-          webSocket.close();
           webSocket.listen(
               (_) { Expect.fail("Unexpected message"); },
               onDone: webSocket.close);
+          webSocket.close();
         });
     });
   }
 
 
   void testNoUpgrade() {
     createServer().then((server) {
       // Create a server which always responds with NOT_FOUND.
       server.listen((request) {
         request.response.statusCode = HttpStatus.NOT_FOUND;
@@ skipping 249 matching lines @@
     testRequestResponseClientCloses(2, 3001, null, 2);
     testRequestResponseClientCloses(2, 3002, "Got tired", 3);
     testRequestResponseServerCloses(2, null, null);
     testRequestResponseServerCloses(2, 3001, null);
     testRequestResponseServerCloses(2, 3002, "Got tired");
     testMessageLength(125);
     testMessageLength(126);
     testMessageLength(127);
     testMessageLength(65535);
     testMessageLength(65536);
+    testCloseNoListen();
+    testListenAfterClose();
     testDoubleCloseClient();
     testDoubleCloseServer();
     testImmediateCloseServer();
     testImmediateCloseClient();
     testNoUpgrade();
     testUsePOST();
     testConnections(10, 3002, "Got tired");
     testIndividualUpgrade(5);
     testFromUpgradedSocket();
     testAdditionalHeaders();
     testBasicAuthentication();
   }
 }
 
 
 void initializeSSL() {
   var testPkcertDatabase = Platform.script.resolve('pkcert').toFilePath();
   SecureSocket.initialize(database: testPkcertDatabase,
                           password: "dartdart");
 }
 
 
 main() {
   new SecurityConfiguration(secure: false).runTests();
   initializeSSL();
   new SecurityConfiguration(secure: true).runTests();
 }