Add a number of convenience methods to test_pub.

utils/tests/pub/test_pub.dart

 // Copyright (c) 2012, 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.
 
 /**
  * Test infrastructure for testing pub. Unlike typical unit tests, most pub
  * tests are integration tests that stage some stuff on the file system, run
  * pub, and then validate the results. This library provides an API to build
  * tests like that.
  */
 #library('test_pub');
 
 #import('dart:io');
 #import('dart:isolate');
 #import('dart:json');
 #import('dart:math');
 #import('dart:uri');
 
 #import('../../../pkg/unittest/unittest.dart');
 #import('../../lib/file_system.dart', prefix: 'fs');
+#import('../../pub/git_source.dart');
 #import('../../pub/io.dart');
+#import('../../pub/repo_source.dart');
+#import('../../pub/sdk_source.dart');
 #import('../../pub/utils.dart');
 #import('../../pub/yaml/yaml.dart');
 
 /**
  * Creates a new [FileDescriptor] with [name] and [contents].
  */
 FileDescriptor file(Pattern name, String contents) =>
     new FileDescriptor(name, contents);
 
 /**
@@ skipping 84 matching lines @@
   _server.close();
   _server = null;
   // TODO(nweiz): Remove this once issue 4155 is fixed. Pumping the event loop
   // *seems* to be enough to ensure that the server is actually closed, but I'm
   // putting this at 10ms to be safe.
   return sleep(10);
 }
 
 /**
  * Creates an HTTP server that replicates the structure of pub.dartlang.org.
- * [pubspecs] is a list of YAML-format pubspecs representing the packages to
+ * [pubspecs] is a list of unserialized pubspecs representing the packages to
  * serve.
  */
-void servePackages(String host, int port, List<String> pubspecs) {
+void servePackages(String host, int port, List<Map> pubspecs) {
   var packages = <String, Map<String, String>>{};
-  pubspecs.forEach((spec) {
-    var parsed = loadYaml(spec);
-    var name = parsed['name'];
-    var version = parsed['version'];
-    packages.putIfAbsent(name, () => <String, String>{})[version] = spec;
-  });
+  for (var spec in pubspecs) {
+    var name = spec['name'];
+    var version = spec['version'];
+    packages.putIfAbsent(name, () => <String, String>{})[version] = yaml(spec);
+  }
 
   serve(host, port, [
     dir('packages', flatten(packages.getKeys().map((name) {
       return [
         file('$name.json',
           JSON.stringify({'versions': packages[name].getKeys()})),
         dir(name, [
           dir('versions', flatten(packages[name].getKeys().map((version) {
             return [
               file('$version.yaml', packages[name][version]),
               tar('$version.tar.gz', [
                 file('pubspec.yaml', packages[name][version]),
                 file('$name.dart', 'main() => print("$name $version");')
               ])
             ];
           })))
         ])
       ];
     })))
   ]);
 }
 
+/** Converts [value] into a YAML string. */
+String yaml(value) => JSON.stringify(value);
+
+/**
+ * Describes a file named `pubspec.yaml` with the given YAML-serialized
+ * [contents], which should be a serializable object.
+ *
+ * [contents] may contain [Future]s that resolve to serializable objects, which
+ * may in turn contain [Future]s recursively.
+ */
+Descriptor pubspec(Map contents) {
+  return async(_awaitObject(contents).transform((resolvedContents) =>
+      file("pubspec.yaml", yaml(resolvedContents))));
+}
+
+/**
+ * Describes a file named `pubspec.yaml` for an application package with the
+ * given [dependencies].
+ */
+Descriptor appPubspec(List dependencies) =>
+  pubspec({"dependencies": _dependencyListToMap(dependencies)});
+
+/**
+ * Describes a file named `pubspec.yaml` for a library package with the given
+ * [name], [version], and [dependencies].
+ */
+Descriptor libPubspec(String name, String version, [List dependencies]) =>
+  pubspec(package(name, version, dependencies));
+
+/**
+ * Describes a map representing a library package with the given [name],
+ * [version], and [dependencies].
+ */
+Map package(String name, String version, [List dependencies]) {
+  var package = {"name": name, "version": version};
+  if (dependencies != null) {
+    package["dependencies"] = _dependencyListToMap(dependencies);
+  }
+  return package;
+}
+
+/**
+ * Describes a map representing a dependency on a package in the package
+ * repository.
+ */
+Map dependency(String name, [String versionConstraint]) {
+  var dependency = {"repo": {"name": name, "url": "http://localhost:3123"}};
+  if (versionConstraint != null) dependency["version"] = versionConstraint;
+  return dependency;
+}
+
+/**
+ * Describes a directory for a package installed from the mock package repo.
+ * This directory is of the form found in the `packages/` directory.
+ */
+DirectoryDescriptor packageDir(String name, String version) {
+  return dir(name, [
+    file("$name.dart", 'main() => print("$name $version");')
+  ]);
+}
+
+/**
+ * Describes a directory for a package installed from the mock package server.
+ * This directory is of the form found in the global package cache.
+ */
+DirectoryDescriptor packageCacheDir(String name, String version) {
+  return dir("$name-$version", [
+    file("$name.dart", 'main() => print("$name $version");')
+  ]);
+}
+
+/**
+ * Describes a directory for a Git package. This directory is of the form found
+ * in the global package cache.
+ */
+DirectoryDescriptor gitPackageCacheDir(String name, [int modifier]) {
+  var value = name;
+  if (modifier != null) value = "$name $modifier";
+  return dir(new RegExp("$name${@'-[a-f0-9]+'}"), [
+    file('$name.dart', 'main() => "$value";')
+  ]);
+}
+
+/**
+ * Describes the `packages/` directory containing all the given [packages],
+ * which should be name/version pairs. The packages will be validated against
+ * the format produced by the mock package server.
+ */
+DirectoryDescriptor packagesDir(Map<String, String> packages) {
+  var contents = <Descriptor>[];
+  packages.forEach((name, version) {
+    contents.add(packageDir(name, version));
+  });
+  return dir(packagesPath, contents);
+}
+
+/**
+ * Describes the global package cache directory containing all the given
+ * [packages], which should be name/version pairs. The packages will be
+ * validated against the format produced by the mock package server.
+ *
+ * A package's value may also be a list of versions, in which case all versions
+ * are expected to be installed.
+ */
+DirectoryDescriptor cacheDir(Map packages) {
+  var contents = <Descriptor>[];
+  packages.forEach((name, versions) {
+    if (versions is! List) versions = [versions];
+    for (var version in versions) {
+      contents.add(packageCacheDir(name, version));
+    }
+  });
+  return dir(cachePath, [
+    dir('repo', [dir('localhost%583123', contents)])
+  ]);
+}
+
+/**
+ * Describes the application directory, containing only a pubspec specifying the
+ * given [dependencies].
+ */
+DirectoryDescriptor appDir(List dependencies) =>
+  dir(appPath, [appPubspec(dependencies)]);
+
+/**
+ * Converts a list of dependencies as passed to [package] into a hash as used in
+ * a pubspec.
+ */
+Map _dependencyListToMap(List<Map> dependencies) {
+  var result = <String, Map>{};
+  dependencies.map((dependency) {
+    var keys = dependency.getKeys().filter((key) => key != "version");
+    var sourceName = only(keys);
+    var source;
+    switch (sourceName) {
+    case "git":
+      source = new GitSource();
+      break;
+    case "repo":
+      source = new RepoSource();
+      break;
+    case "sdk":
+      source = new SdkSource('');
+      break;
+    default:
+      throw 'Unknown source "$sourceName"';
+    }
+
+    result[source.packageName(dependency[sourceName])] = dependency;
+  });
+  return result;
+}
+
 /**
  * The path of the package cache directory used for tests. Relative to the
  * sandbox directory.
  */
 final String cachePath = "cache";
 
 /**
  * The path of the mock SDK directory used for tests. Relative to the sandbox
  * directory.
  */
@@ skipping 308 matching lines @@
     }
 
     return listDir(dir).chain((files) {
       var matches = files.filter((file) => endsWithPattern(file, name));
       if (matches.length == 0) {
         Expect.fail('No files in $dir match pattern $name.');
       }
       if (matches.length == 1) return validate(matches[0]);
 
       var failures = [];
+      var successes = 0;
       var completer = new Completer();
+      checkComplete() {
+        if (failures.length + successes != matches.length) return;
+        if (successes > 0) {
+          completer.complete(null);
+          return;
+        }
+
+        var error = new StringBuffer();
+        error.add("No files named $name in $dir were valid:\n");
+        for (var failure in failures) {
+          error.add("  ").add(failure).add("\n");
+        }
+        completer.completeException(new ExpectException(error.toString()));
+      }
+
       for (var match in matches) {
         var future = validate(match);
 
         future.handleException((e) {
           failures.add(e);
-          if (failures.length != matches.length) return true;
-
-          var error = new StringBuffer();
-          error.add("No files named $name in $dir were valid:\n");
-          for (var failure in failures) {
-            error.add("  ").add(failure).add("\n");
-          }
-          completer.completeException(new ExpectException(error.toString()));
+          checkComplete();
           return true;
         });
 
-        future.then(completer.complete);
+        future.then((_) {
+          successes++;
+          checkComplete();
+        });
       }
       return completer.future;
     });
   }
 }
 
 /**
  * Describes a file. These are used both for setting up an expected directory
  * tree before running a test, and for validating that the file system matches
  * some expectations after running it.
@@ skipping 63 matching lines @@
    */
   final List<Descriptor> contents;
 
   DirectoryDescriptor(Pattern name, this.contents) : super(name);
 
   /**
    * Creates the file within [dir]. Returns a [Future] that is completed after
    * the creation is done.
    */
   Future<Directory> create(parentDir) {
-    final completer = new Completer<Directory>();
+    // Create the directory.
+    return ensureDir(join(parentDir, _stringName)).chain((dir) {
+      if (contents == null) return new Future<Directory>.immediate(dir);
 
-    // Create the directory.
-    ensureDir(join(parentDir, _stringName)).then((dir) {
-      if (contents == null) {
-        completer.complete(dir);
-      } else {
-        // Recursively create all of its children.
-        final childFutures = contents.map((child) => child.create(dir));
-        Futures.wait(childFutures).then((_) {
-          // Only complete once all of the children have been created too.
-          completer.complete(dir);
-        });
-      }
+      // Recursively create all of its children.
+      final childFutures = contents.map((child) => child.create(dir));
+      // Only complete once all of the children have been created too.
+      return Futures.wait(childFutures).transform((_) => dir);
     });
-
-    return completer.future;
   }
 
   /**
    * Deletes the directory within [dir]. Returns a [Future] that is completed
    * after the deletion is done.
    */
   Future delete(dir) {
     return deleteDir(join(dir, _stringName));
   }
 
@@ skipping 182 matching lines @@
     }).then((tar) {
       var sourceStream = tar.openInputStream();
       pipeInputToInput(
           sourceStream, sinkStream, onClosed: tempDir.deleteRecursively);
     });
     return sinkStream;
   }
 }
 
 /**
+ * Takes a simple data structure (composed of [Map]s, [List]s, scalar objects,
+ * and [Future]s) and recursively resolves all the [Future]s contained within.
+ * Completes with the fully resolved structure.
+ */
+Future _awaitObject(object) {
+  // Unroll nested futures.
+  if (object is Future) return object.chain(_awaitObject);
+  if (object is Collection) return Futures.wait(object.map(_awaitObject));
+  if (object is! Map) return new Future.immediate(object);
+
+  var pairs = <Future<Pair>>[];
+  object.forEach((key, value) {
+    pairs.add(_awaitObject(value)
+        .transform((resolved) => new Pair(key, resolved)));
+  });
+  return Futures.wait(pairs).transform((resolvedPairs) {
+    var map = {};
+    for (var pair in resolvedPairs) {
+      map[pair.first] = pair.last;
+    }
+    return map;
+  });
+}
+
+/**
  * Schedules a callback to be called as part of the test case.
  */
 void _schedule(_ScheduledEvent event) {
   if (_scheduled == null) _scheduled = [];
   _scheduled.add(event);
 }
 
 /**
  * Schedules a callback to be called after Pub is run with [runPub], even if it
  * fails.
  */
 void _scheduleCleanup(_ScheduledEvent event) {
   if (_scheduledCleanup == null) _scheduledCleanup = [];
   _scheduledCleanup.add(event);
 }