Add support for user-defined HTML files.

test/runner/pub_serve_test.dart

 // Copyright (c) 2015, 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.
 
 @TestOn("vm")
 
 import 'dart:convert';
 import 'dart:io';
 
 import 'package:path/path.dart' as p;
@@ skipping 27 matching lines @@
 
     new Directory(p.join(_sandbox, "test")).createSync();
 
     new File(p.join(_sandbox, "test", "my_test.dart")).writeAsStringSync("""
 import 'package:test/test.dart';
 
 void main() {
   test("test", () => expect(true, isTrue));
 }
 """);
-  });
 
-  tearDown(() {
-    new Directory(_sandbox).deleteSync(recursive: true);
-  });
+    new Directory(p.join(_sandbox, "lib")).createSync();
 
-  group("with transformed tests", () {
-    setUp(() {
-      new Directory(p.join(_sandbox, "lib")).createSync();
-
-      new File(p.join(_sandbox, "lib", "myapp.dart")).writeAsStringSync("""
+    new File(p.join(_sandbox, "lib", "myapp.dart")).writeAsStringSync("""
 import 'package:barback/barback.dart';
 
 class MyTransformer extends Transformer {
   final allowedExtensions = '.dart';
 
   MyTransformer.asPlugin();
 
   Future apply(Transform transform) {
     return transform.primaryInput.readAsString().then((contents) {
       print("contents: \$contents");
       print("new contents: \${contents.replaceAll("isFalse", "isTrue")}");
       transform.addOutput(new Asset.fromString(
           transform.primaryInput.id,
           contents.replaceAll("isFalse", "isTrue")));
     });
   }
 }
 """);
 
-      var pubGetResult = runPub(['get'], workingDirectory: _sandbox);
-      expect(pubGetResult.exitCode, equals(0));
-    });
+    var pubGetResult = runPub(['get'], workingDirectory: _sandbox);
+    expect(pubGetResult.exitCode, equals(0));
+  });
 
+  tearDown(() {
+    new Directory(_sandbox).deleteSync(recursive: true);
+  });
+
+  group("with transformed tests", () {
     test("runs those tests in the VM", () {
       return startPub(['serve', '--port', '0'], workingDirectory: _sandbox)
           .then((process) {
         return _lines.bind(process.stdout)
             .firstWhere(_servingRegExp.hasMatch)
             .then((line) {
           var match = _servingRegExp.firstMatch(line);
 
           try {
             var result = runUnittest(['--pub-serve=${match[1]}'],
                 workingDirectory: _sandbox);
             expect(result.exitCode, equals(0));
             expect(result.stdout, contains('+1: All tests passed!'));
           } finally {
             process.kill();
           }
         });
       });
     });
 
-    test("runs those tests in the browser", () {
+    test("runs those tests on Chrome", () {
       return startPub(['serve', '--port', '0'],
               workingDirectory: _sandbox)
           .then((process) {
         return _lines.bind(process.stdout)
             .firstWhere(_servingRegExp.hasMatch)
             .then((line) {
           var match = _servingRegExp.firstMatch(line);
 
           try {
             var result = runUnittest(
                 ['--pub-serve=${match[1]}', '-p', 'chrome'],
                 workingDirectory: _sandbox);
             expect(result.exitCode, equals(0));
             expect(result.stdout, contains('+1: All tests passed!'));
           } finally {
             process.kill();
           }
         });
       });
     });
 
+    test("runs those tests on content shell", () {
+      return startPub(['serve', '--port', '0'],
+              workingDirectory: _sandbox)
+          .then((process) {
+        return _lines.bind(process.stdout)
+            .firstWhere(_servingRegExp.hasMatch)
+            .then((line) {
+          var match = _servingRegExp.firstMatch(line);
+
+          try {
+            var result = runUnittest(
+                ['--pub-serve=${match[1]}', '-p', 'content-shell'],
+                workingDirectory: _sandbox);
+            expect(result.exitCode, equals(0));
+            expect(result.stdout, contains('+1: All tests passed!'));
+          } finally {
+            process.kill();
+          }
+        });
+      });
+    });
+
     test("gracefully handles pub serve running on the wrong directory for "
         "VM tests", () {
       new Directory(p.join(_sandbox, "web")).createSync();
 
       return startPub(['serve', '--port', '0', 'web'],
               workingDirectory: _sandbox)
           .then((process) {
         return _lines.bind(process.stdout)
             .firstWhere(_servingRegExp.hasMatch)
             .then((line) {
@@ skipping 75 matching lines @@
     \$include: test/**_test.dart
 '''));
           } finally {
             process.kill();
           }
         });
       });
     });
   });
 
+  group("uses a custom HTML file", () {
+    setUp(() {
+      new File(p.join(_sandbox, "test", "test.dart")).writeAsStringSync("""
+import 'dart:html';
+
+import 'package:test/test.dart';
+
+void main() {
+  test("failure", () {
+    expect(document.query('#foo'), isNull);
+  });
+}
+""");
+
+      new File(p.join(_sandbox, "test", "test.html")).writeAsStringSync("""
+<html>
+<head>
+  <link rel='x-dart-test' href='test.dart'>
+  <script src="packages/test/dart.js"></script>
+</head>
+<body>
+  <div id="foo"></div>
+</body>
+""");
+    });
+
+    test("on Chrome", () {
+      return startPub(['serve', '--port', '0'],
+              workingDirectory: _sandbox)
+          .then((process) {
+        return _lines.bind(process.stdout)
+            .firstWhere(_servingRegExp.hasMatch)
+            .then((line) {
+          var match = _servingRegExp.firstMatch(line);
+
+          try {
+            var result = runUnittest(
+                ['--pub-serve=${match[1]}', '-p', 'chrome'],
+                workingDirectory: _sandbox);
+            expect(result.exitCode, equals(0));
+            expect(result.stdout, contains('+1: All tests passed!'));
+          } finally {
+            process.kill();
+          }
+        });
+      });
+    });
+
+    test("on content shell", () {
+      return startPub(['serve', '--port', '0'],
+              workingDirectory: _sandbox)
+          .then((process) {
+        return _lines.bind(process.stdout)
+            .firstWhere(_servingRegExp.hasMatch)
+            .then((line) {
+          var match = _servingRegExp.firstMatch(line);
+
+          try {
+            var result = runUnittest(
+                ['--pub-serve=${match[1]}', '-p', 'content-shell'],
+                workingDirectory: _sandbox);
+            expect(result.exitCode, equals(0));
+            expect(result.stdout, contains('+1: All tests passed!'));
+          } finally {
+            process.kill();
+          }
+        });
+      });
+    });
+  });
+
   test("gracefully handles pub serve not running for VM tests", () {
     var result = runUnittest(['--pub-serve=54321'],
         workingDirectory: _sandbox);
     expect(result.stdout, allOf([
       contains('-1: load error'),
       contains('''
   Failed to load "test/my_test.dart":
   Error getting http://localhost:54321/my_test.dart.vm_test.dart: Connection refused
   Make sure "pub serve" is running.''')
     ]));
@@ skipping 21 matching lines @@
         workingDirectory: _sandbox);
     expect(result.stdout, allOf([
       contains('-1: load error'),
       contains(
           'Failed to load "my_test.dart": When using "pub serve", all test '
               'files must be in test/.\n')
     ]));
     expect(result.exitCode, equals(1));
   });
 }