add some PackageGraph unit tests

Author: jakemac53

.gitignore

@@ -4,4 +4,5 @@ build/
 .pub/
 pubspec.lock
 
-generated
+# Include .packages files from tests which are hand coded
+!test/package_graph/**/.packages

lib/src/package_graph/package_graph.dart

@@ -3,6 +3,7 @@
 // BSD-style license that can be found in the LICENSE file.
 import 'dart:io';
 
+import 'package:path/path.dart' as path;
 import 'package:yaml/yaml.dart';
 
 /// A graph of the package dependencies for an application.
@@ -16,19 +17,19 @@ class PackageGraph {
   PackageGraph._(this.root, Map<String, PackageNode> allPackages)
       : allPackages = new Map.unmodifiable(allPackages);
 
-  /// Creates a [PackageGraph] for the package in which you are currently
-  /// running.
-  factory PackageGraph.forThisPackage() {
+  /// Creates a [PackageGraph] for the package whose top level directory lives
+  /// at [packagePath] (no trailing slash).
+  factory PackageGraph.forPath(String packagePath) {
     /// Read in the pubspec file and parse it as yaml.
-    var pubspec = new File('pubspec.yaml');
+    var pubspec = new File(path.join(packagePath, 'pubspec.yaml'));
     if (!pubspec.existsSync()) {
       throw 'Unable to generate package graph, no `pubspec.yaml` found. '
           'This program must be ran from the root directory of your package.';
     }
-    var yaml = loadYaml(pubspec.readAsStringSync());
+    var rootYaml = loadYaml(pubspec.readAsStringSync());
 
     /// Read in the `.packages` file to get the locations of all packages.
-    var packagesFile = new File('.packages');
+    var packagesFile = new File(path.join(packagePath, '.packages'));
     if (!packagesFile.existsSync()) {
       throw 'Unable to generate package graph, no `.packages` found. '
           'This program must be ran from the root directory of your package.';
@@ -37,18 +38,26 @@ class PackageGraph {
     packagesFile.readAsLinesSync().skip(1).forEach((line) {
       var firstColon = line.indexOf(':');
       var name = line.substring(0, firstColon);
-      var uriString = line.substring(firstColon + 1);
+      assert(line.endsWith('lib${Platform.pathSeparator}'));
+      // Start after package_name:, and strip out trailing `lib` dir.
+      var uriString = line.substring(firstColon + 1, line.length - 4);
+      var uri;
       try {
-        packageLocations[name] = Uri.parse(uriString);
+        uri = Uri.parse(uriString);
       } on FormatException catch (_) {
         /// Some types of deps don't have a scheme, and just point to a relative
         /// path.
-        packageLocations[name] = new Uri.file(uriString);
+        uri = new Uri.file(uriString);
       }
+      if (!uri.isAbsolute) {
+        uri = new Uri.file(path.join(packagePath, uri.path));
+      }
+      packageLocations[name] = uri;
     });
 
     /// Create all [PackageNode]s for all deps.
     var nodes = <String, PackageNode>{};
+    Map<String, dynamic> rootDeps;
     PackageNode addNodeAndDeps(YamlMap yaml, PackageDependencyType type,
         {bool isRoot: false}) {
       var name = yaml['name'];
@@ -57,24 +66,31 @@ class PackageGraph {
           name, yaml['version'], type, packageLocations[name]);
       nodes[name] = node;
 
-      _depsFromYaml(yaml, withOverrides: isRoot).forEach((name, source) {
+      var deps = _depsFromYaml(yaml, withOverrides: isRoot);
+      if (isRoot) rootDeps = deps;
+      deps.forEach((name, source) {
         var dep = nodes[name];
         if (dep == null) {
           var pubspec = _pubspecForUri(packageLocations[name]);
-          dep = addNodeAndDeps(pubspec, _dependencyType(source));
+          dep = addNodeAndDeps(pubspec, _dependencyType(rootDeps[name]));
         }
         node._dependencies.add(dep);
       });
 
       return node;
     }
 
-    var root = addNodeAndDeps(yaml, PackageDependencyType.Path, isRoot: true);
+    var root =
+        addNodeAndDeps(rootYaml, PackageDependencyType.Path, isRoot: true);
     return new PackageGraph._(root, nodes);
   }
 
+  /// Creates a [PackageGraph] for the package in which you are currently
+  /// running.
+  factory PackageGraph.forThisPackage() => new PackageGraph.forPath('.');
+
   /// Shorthand to get a package by name.
-  PackageNode operator[] (String packageName) => allPackages[packageName];
+  PackageNode operator [](String packageName) => allPackages[packageName];
 
   String toString() {
     var buffer = new StringBuffer();
@@ -118,7 +134,7 @@ class PackageNode {
 enum PackageDependencyType { Pub, Github, Path, }
 
 PackageDependencyType _dependencyType(source) {
-  if (source is String) return PackageDependencyType.Pub;
+  if (source is String || source == null) return PackageDependencyType.Pub;
 
   assert(source is YamlMap);
   assert(source.keys.length == 1);
@@ -145,14 +161,9 @@ Map<String, YamlMap> _depsFromYaml(YamlMap yaml, {bool withOverrides: false}) {
   return deps;
 }
 
-/// [uri] should be directly from a `.packages` file, and points to the `lib`
-/// dir.
+/// [uri] should point to the top level directory for the package.
 YamlMap _pubspecForUri(Uri uri) {
-  var libPath = uri.toFilePath();
-  assert(libPath.endsWith('lib/'));
-  var pubspecPath =
-      libPath.replaceRange(libPath.length - 4, libPath.length, 'pubspec.yaml');
-
+  var pubspecPath = path.join(uri.toFilePath(), 'pubspec.yaml');
   var pubspec = new File(pubspecPath);
   if (!pubspec.existsSync()) {
     throw 'Unable to generate package graph, no `$pubspecPath` found.';

test/package_graph/basic_pkg/.packages

@@ -0,0 +1,6 @@
+# Generated by hand on 2016-02-01
+a:pkg/a/lib/
+b:pkg/b/lib/
+c:pkg/c/lib/
+d:pkg/d/lib/
+basic_pkg:lib/

test/package_graph/basic_pkg/pkg/a/pubspec.yaml

@@ -0,0 +1,5 @@
+name: a
+version: 2.0.0
+dependencies:
+  b: ^3.0.0
+  c: any

test/package_graph/basic_pkg/pkg/b/pubspec.yaml

@@ -0,0 +1,4 @@
+name: b
+version: 3.0.0
+dependencies:
+  c: ^0.4.0

test/package_graph/basic_pkg/pkg/c/pubspec.yaml

@@ -0,0 +1,4 @@
+name: c
+version: 4.0.0
+dependencies:
+  basic_pkg: ^1.0.0

test/package_graph/basic_pkg/pubspec.yaml

@@ -0,0 +1,8 @@
+name: basic_pkg
+version: 1.0.0
+dependencies:
+  a: 2.0.0
+  b:
+    git: git://github.com/b/b.git
+  c:
+    path: pkg/c

test/package_graph/no_packages_file/pubspec.yaml

@@ -0,0 +1,2 @@
+name: a
+version: 1.0.0

test/package_graph/no_pubspec/.packages

@@ -0,0 +1 @@
+# Created by hand, but empty

test/package_graph/package_graph_test.dart

@@ -0,0 +1,78 @@
+// Copyright (c) 2016, 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 'package:test/test.dart';
+
+import 'package:build/build.dart';
+
+main() async {
+  PackageGraph graph;
+
+  group('forThisPackage ', () {
+    setUp(() async {
+      graph = await new PackageGraph.forThisPackage();
+    });
+
+    test('root', () {
+      expectPkg(
+          graph.root, 'build', isNotEmpty, PackageDependencyType.Path, './');
+    });
+  });
+
+  group('basic package ', () {
+    var basicPkgPath = 'test/package_graph/basic_pkg';
+
+    setUp(() async {
+      graph = await new PackageGraph.forPath(basicPkgPath);
+    });
+
+    test('allPackages', () {
+      expect(graph.allPackages, {
+        'a': graph['a'],
+        'b': graph['b'],
+        'c': graph['c'],
+        'basic_pkg': graph['basic_pkg'],
+      });
+    });
+
+    test('root', () {
+      expectPkg(graph.root, 'basic_pkg', '1.0.0', PackageDependencyType.Path,
+          basicPkgPath, [graph['a'], graph['b'], graph['c']]);
+    });
+
+    test('pub dependency', () {
+      expectPkg(graph['a'], 'a', '2.0.0', PackageDependencyType.Pub,
+          '$basicPkgPath/pkg/a/', [graph['b'], graph['c']]);
+    });
+
+    test('git dependency', () {
+      expectPkg(graph['b'], 'b', '3.0.0', PackageDependencyType.Github,
+          '$basicPkgPath/pkg/b/', [graph['c']]);
+    });
+
+    test('path dependency', () {
+      expectPkg(graph['c'], 'c', '4.0.0', PackageDependencyType.Path,
+          '$basicPkgPath/pkg/c/', [graph['basic_pkg']]);
+    });
+  });
+
+  test('missing pubspec throws on create', () {
+    expect(() => new PackageGraph.forPath('no_pubspec'), throws);
+  });
+
+  test('missing .packages file throws on create', () {
+    expect(() => new PackageGraph.forPath('no_packages_file'), throws);
+  });
+}
+
+void expectPkg(PackageNode node, name, version, type, location,
+    [Iterable<PackageNode> dependencies]) {
+  expect(node.name, name);
+  expect(node.version, version);
+  expect(node.dependencyType, type);
+  expect(node.location.path, location);
+  if (dependencies != null) {
+    expect(node.dependencies, unorderedEquals(dependencies));
+  }
+}