tests/standalone/io/http_server_early_client_close_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.

import "dart:async";
import "dart:io";
import "dart:isolate";

void sendData(List<int> data, int port) {
  Socket.connect("127.0.0.1", port).then((socket) {
    socket.listen((data) {
        Expect.fail("No data response was expected");
      });
    socket.add(data);
    socket.close();
    socket.done.then((_) {
      socket.destroy();
    });
  });
}

class EarlyCloseTest {
  EarlyCloseTest(this.data,
                 String this.exception,
                 [bool this.expectRequest = false]);

  Future execute(HttpServer server) {
    Completer c = new Completer();

    bool calledOnRequest = false;
    bool calledOnError = false;
    ReceivePort port = new ReceivePort();
    server.listen(
        (request) {
          Expect.isTrue(expectRequest);
          Expect.isFalse(calledOnError);
          Expect.isFalse(calledOnRequest, "onRequest called multiple times");
          calledOnRequest = true;
          request.listen(
              (_) {},
              onError: (e) {
                Expect.isFalse(calledOnError);
                Expect.equals(exception, e.error.message);
                calledOnError = true;
                port.close();
                c.complete(null);
              });
        },
        onError: (e) {
          Expect.isFalse(calledOnError);
          Expect.equals(exception, e.error.message);
          Expect.equals(expectRequest, calledOnRequest);
          calledOnError = true;
          port.close();
          c.complete(null);
        });

    List<int> d;
    if (data is List<int>) d = data;
    if (data is String) d = data.codeUnits;
    if (d == null) Expect.fail("Invalid data");
    sendData(d, server.port);

    return c.future;
  }

  final data;
  final String exception;
  final bool expectRequest;
}

void testEarlyClose1() {
  List<EarlyCloseTest> tests = new List<EarlyCloseTest>();
  void add(Object data, String exception, {bool expectRequest: false}) {
    tests.add(new EarlyCloseTest(data, exception, expectRequest));
  }
  // The empty packet is valid.

  // Close while sending header
  String message = "Connection closed before full header was received";
  add("G", message);
  add("GET /", message);
  add("GET / HTTP/1.1", message);
  add("GET / HTTP/1.1\r\n", message);

  // Close while sending content
  add("GET / HTTP/1.1\r\nContent-Length: 100\r\n\r\n",
      "Connection closed while receiving data",
      expectRequest: true);
  add("GET / HTTP/1.1\r\nContent-Length: 100\r\n\r\n1",
      "Connection closed while receiving data",
      expectRequest: true);

  void runTest(Iterator it) {
    if (it.moveNext()) {
      HttpServer.bind("127.0.0.1", 0).then((server) {
        it.current.execute(server).then((_) {
          runTest(it);
          server.close();
        });
      });
    }
  }
  runTest(tests.iterator);
}

testEarlyClose2() {
  HttpServer.bind("127.0.0.1", 0).then((server) {
    server.listen(
      (request) {
        String name = new Options().script;
        new File(name).openRead().pipe(request.response);
      },
      onError: (e) { /* ignore */ });

    var count = 0;
    makeRequest() {
      Socket.connect("127.0.0.1", server.port).then((socket) {
        var data = "GET / HTTP/1.1\r\nContent-Length: 0\r\n\r\n".codeUnits;
        socket.add(data);
        socket.close();
        socket.done.then((_) {
          socket.destroy();
          if (++count < 10) {
            makeRequest();
          } else {
            server.close();
          }
        });
      });
    }
    makeRequest();
  });
}

void main() {
  testEarlyClose1();
//  testEarlyClose2();
}