sdk/tests/standalone/io/web_socket_protocol_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 "package:expect/expect.dart";
import "dart:async";
import "dart:io";

testEmptyProtocol() {
  HttpServer.bind("127.0.0.1", 0).then((server) {
    server.listen((request) {
      WebSocketTransformer.upgrade(request).then((websocket) {
        websocket.close();
      });
    });
    WebSocket.connect("ws://127.0.0.1:${server.port}/", protocols: []).then((
      client,
    ) {
      Expect.isNull(client.protocol);
      client.close();
      server.close();
    });
  });
}

testProtocol(List<String> protocols, String used) {
  selector(List<String> receivedProtocols) {
    Expect.listEquals(protocols, receivedProtocols);
    return used;
  }

  HttpServer.bind("127.0.0.1", 0).then((server) {
    server.listen((request) {
      WebSocketTransformer.upgrade(request, protocolSelector: selector).then((
        websocket,
      ) {
        Expect.equals(used, websocket.protocol);
        websocket.close();
      });
    });
    WebSocket.connect(
      "ws://127.0.0.1:${server.port}/",
      protocols: protocols,
    ).then((client) {
      Expect.equals(used, client.protocol);
      client.close();
      server.close();
    });
  });
}

testProtocolHandler() {
  // Test throwing an error.
  HttpServer.bind("127.0.0.1", 0).then((server) {
    server.listen((request) {
      selector(List<String> receivedProtocols) {
        throw "error";
      }

      WebSocketTransformer.upgrade(request, protocolSelector: selector).then(
        (websocket) {
          Expect.fail('error expected');
        },
        onError: (error) {
          Expect.equals('error', error);
        },
      );
    });
    WebSocket.connect(
      "ws://127.0.0.1:${server.port}/",
      protocols: ["v1.example.com"],
    ).then(
      (client) {
        Expect.fail('error expected');
      },
      onError: (error) {
        server.close();
      },
    );
  });

  // Test returning another protocol.
  HttpServer.bind("127.0.0.1", 0).then((server) {
    server.listen((request) {
      selector(List<String> receivedProtocols) => "v2.example.com";
      WebSocketTransformer.upgrade(request, protocolSelector: selector).then(
        (websocket) {
          Expect.fail('error expected');
        },
        onError: (error) {
          Expect.isTrue(error is WebSocketException);
        },
      );
    });
    WebSocket.connect(
      "ws://127.0.0.1:${server.port}/",
      protocols: ["v1.example.com"],
    ).then(
      (client) {
        Expect.fail('error expected');
      },
      onError: (error) {
        server.close();
      },
    );
  });
}

void main() {
  testEmptyProtocol();
  testProtocol(["v1.example.com", "v2.example.com"], "v1.example.com");
  testProtocol(["v1.example.com", "v2.example.com"], "v2.example.com");
  testProtocolHandler();
}