Add new VirtualHost class to http_server package, for handling multiple hosts on multiple sources (…

pkg/http_server/test/virtual_host_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 "package:unittest/unittest.dart";
+import "package:http_server/http_server.dart";
+
+
+const CERTIFICATE = "localhost_cert";
+
+Future<int> getStatusCode(int port,
+                          String path,
+                          {String host,
+                           bool secure: false}) {
+  var uri = (secure ? "https" : "http") + '://localhost:$port$path';

[Søren Gjesse, 2013/06/24 07:58:37]

Mow about using the new Uri.http / Uri.https constructors?

[Anders Johnsen, 2013/06/24 10:07:06]

Done.

+  return new HttpClient().getUrl(Uri.parse(uri))
+      .then((request) {
+        if (host != null) request.headers.host = host;
+        return request.close();
+      })
+      .then((response) => response.drain().then(
+          (_) => response.statusCode));
+}
+
+
+void main() {
+  Path scriptDir = new Path(new Options().script).directoryPath;
+  Path certificateDatabase = scriptDir.append('pkcert');
+  SecureSocket.initialize(database: certificateDatabase.toNativePath(),
+                          password: 'dartdart');
+
+  test('empty-host', () {
+    expect(HttpServer.bind('localhost', 0).then((server) {
+      var virHost = new VirtualHost(server);
+      return getStatusCode(server.port, '/')
+          .whenComplete(server.close);
+    }), completion(equals(HttpStatus.FORBIDDEN)));
+  });
+
+  test('empty-host-unhandled', () {
+    expect(HttpServer.bind('localhost', 0).then((server) {
+      var virHost = new VirtualHost(server);
+      expect(virHost.unhandled.first.then((request) {
+        request.response.close();
+      }), completion(isNull));
+      return getStatusCode(server.port, '/')
+          .whenComplete(server.close);
+    }), completion(equals(HttpStatus.OK)));
+  });
+
+  test('single-host', () {
+    expect(HttpServer.bind('localhost', 0).then((server) {
+      var virHost = new VirtualHost(server);
+      expect(virHost.addHost('*.host.com').first.then((request) {
+        request.response.close();
+      }), completion(isNull));
+      return getStatusCode(server.port, '/', host: 'my.host.com')
+          .whenComplete(server.close);
+    }), completion(equals(HttpStatus.OK)));
+  });
+
+  test('multiple-host', () {
+    expect(HttpServer.bind('localhost', 0).then((server) {
+      var virHost = new VirtualHost(server);
+      expect(virHost.addHost('*.host1.com').first.then((request) {
+        request.response.close();
+      }), completion(isNull));
+      expect(virHost.addHost('*.host2.com').first.then((request) {
+        request.response.close();
+      }), completion(isNull));
+      expect(virHost.addHost('*.host3.com').first.then((request) {
+        request.response.close();
+      }), completion(isNull));
+      return Future.wait([
+          getStatusCode(server.port, '/', host: 'my.host1.com'),
+          getStatusCode(server.port, '/', host: 'my.host2.com'),
+          getStatusCode(server.port, '/', host: 'my.host3.com')])
+          .whenComplete(server.close);
+    }), completion(equals([HttpStatus.OK, HttpStatus.OK, HttpStatus.OK])));
+  });
+
+  test('multiple-source-https', () {
+    expect(Future.wait([
+        HttpServer.bind('localhost', 0),
+        HttpServer.bindSecure('localhost', 0, certificateName: CERTIFICATE)])
+        .then((servers) {
+      var virHost = new VirtualHost();
+      virHost.addSource(servers[0]);
+      virHost.addSource(servers[1]);
+      virHost.unhandled.listen((request) {
+        request.response.close();
+      });
+      return Future.wait([
+          getStatusCode(servers[0].port, '/', host: 'myhost1.com'),
+          getStatusCode(servers[1].port, '/', host: 'myhost2.com', secure: true)])

[ricow1, 2013/06/24 07:41:30]

long line

[Anders Johnsen, 2013/06/24 10:07:06]

Done.

+          .whenComplete(() => servers.forEach((s) => s.close()));
+    }), completion(equals([HttpStatus.OK, HttpStatus.OK])));
+  });
+
+  group('domain', () {
+    test('specific-sub-domain', () {
+      expect(HttpServer.bind('localhost', 0).then((server) {
+        var virHost = new VirtualHost(server);
+        expect(virHost.addHost('my1.host.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        expect(virHost.addHost('my2.host.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        expect(virHost.addHost('my3.host.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        return Future.wait([
+            getStatusCode(server.port, '/', host: 'my1.host.com'),
+            getStatusCode(server.port, '/', host: 'my2.host.com'),
+            getStatusCode(server.port, '/', host: 'my3.host.com')])
+            .whenComplete(server.close);
+      }), completion(equals([HttpStatus.OK, HttpStatus.OK, HttpStatus.OK])));
+    });
+
+    test('wildcard-sub-domain', () {
+      expect(HttpServer.bind('localhost', 0).then((server) {
+        var virHost = new VirtualHost(server);
+        expect(virHost.addHost('*.host1.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        expect(virHost.addHost('*.host2.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        expect(virHost.addHost('*.host3.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        return Future.wait([
+            getStatusCode(server.port, '/', host: 'my.host1.com'),
+            getStatusCode(server.port, '/', host: 'my.host2.com'),
+            getStatusCode(server.port, '/', host: 'my.host3.com')])
+            .whenComplete(server.close);
+      }), completion(equals([HttpStatus.OK, HttpStatus.OK, HttpStatus.OK])));
+    });
+
+    test('mix-sub-domain', () {
+      expect(HttpServer.bind('localhost', 0).then((server) {
+        var virHost = new VirtualHost(server);
+        expect(virHost.addHost('my1.host.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        expect(virHost.addHost('my2.host.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        expect(virHost.addHost('*.host.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        return Future.wait([
+            getStatusCode(server.port, '/', host: 'my1.host.com'),
+            getStatusCode(server.port, '/', host: 'my2.host.com'),
+            getStatusCode(server.port, '/', host: 'my3.host.com')])
+            .whenComplete(server.close);
+      }), completion(equals([HttpStatus.OK, HttpStatus.OK, HttpStatus.OK])));
+    });
+
+
+    test('wildcard', () {
+      expect(HttpServer.bind('localhost', 0).then((server) {
+        var virHost = new VirtualHost(server);
+        expect(virHost.addHost('*').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        expect(virHost.addHost('*.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        expect(virHost.addHost('*.host.com').first.then((request) {
+          request.response.close();
+        }), completion(isNull));
+        return Future.wait([
+            getStatusCode(server.port, '/', host: 'some.host.dk'),
+            getStatusCode(server.port, '/', host: 'my.host2.com'),
+            getStatusCode(server.port, '/', host: 'long.sub.of.host.com')])
+            .whenComplete(server.close);
+      }), completion(equals([HttpStatus.OK, HttpStatus.OK, HttpStatus.OK])));
+    });
+
+    test('duplicate-domain', () {
+      var virHost = new VirtualHost();
+      virHost.addHost('my1.host.com');
+      expect(() => (virHost.addHost('my1.host.com')), throwsArgumentError);
+      virHost.addHost('*.host.com');
+      expect(() => (virHost.addHost('*.host.com')), throwsArgumentError);
+      virHost.addHost('my2.host.com');
+      virHost.addHost('my3.host.com');
+      virHost.addHost('*.com');
+      virHost.addHost('*');
+      expect(() => (virHost.addHost('*')), throwsArgumentError);
+    });
+  });
+}
+