Add the ability to dynamically modify transforms to barback.

pkg/barback/test/utils.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.
 
 library barback.test.utils;
 
 import 'dart:async';
 import 'dart:collection';
 import 'dart:io';
 
@@ skipping 43 matching lines @@
 /// [AssetId] and the value should be a string defining the contents of that
 /// asset.
 ///
 /// [transformers] is a map from package names to the transformers for each
 /// package.
 void initGraph([assets,
     Map<String, Iterable<Iterable<Transformer>>> transformers]) {
   if (assets == null) assets = [];
   if (transformers == null) transformers = {};
 
-  _provider = new MockProvider(assets, transformers);
+  var assetList;
+  if (assets is Map) {
+    assetList = assets.keys.map((asset) {
+      var id = new AssetId.parse(asset);
+      return new _MockAsset(id, assets[asset]);
+    });
+  } else if (assets is Iterable) {
+    assetList = assets.map((asset) {
+      var id = new AssetId.parse(asset);
+      var contents = pathos.basenameWithoutExtension(id.path);
+      return new _MockAsset(id, contents);
+    });
+  }
+
+  var assetMap = mapMapValues(groupBy(assetList, (asset) => asset.id.package),
+      (package, assets) => new AssetSet.from(assets));
+
+  // Make sure that packages that have transformers but no assets are considered
+  // by MockProvider to exist.
+  for (var package in transformers.keys) {
+    assetMap.putIfAbsent(package, () => new AssetSet());
+  }
+
+  _provider = new MockProvider(assetMap);
   _barback = new Barback(_provider);
   _nextBuildResult = 0;
+
+  transformers.forEach(_barback.updateTransformers);
+
+  // There should be one successful build after adding all the transformers but
+  // before adding any sources.
+  if (!transformers.isEmpty) buildShouldSucceed();
 }
 
 /// Updates [assets] in the current [PackageProvider].
 ///
 /// Each item in the list may either be an [AssetId] or a string that can be
 /// parsed as one.
 void updateSources(Iterable assets) {
   assets = _parseAssets(assets);
   schedule(() => _barback.updateSources(assets),
       "updating ${assets.join(', ')}");
@@ skipping 18 matching lines @@
 }
 
 /// Removes [assets] from the current [PackageProvider].
 ///
 /// Each item in the list may either be an [AssetId] or a string that can be
 /// parsed as one. Unlike [removeSources], this is not automatically scheduled
 /// and will be run synchronously when called.
 void removeSourcesSync(Iterable assets) =>
     _barback.removeSources(_parseAssets(assets));
 
+/// Sets the transformers for [package] to [transformers].
+void updateTransformers(String package,
+    Iterable<Iterable<Transformers>> transformers) {
+  schedule(() => _barback.updateTransformers(package, transformers),
+      "updating transformers for $package");
+}
+
 /// Parse a list of strings or [AssetId]s into a list of [AssetId]s.
 List<AssetId> _parseAssets(Iterable assets) {
   return assets.map((asset) {
     if (asset is String) return new AssetId.parse(asset);
     return asset;
   }).toList();
 }
 
 /// Schedules a change to the contents of an asset identified by [name] to
 /// [contents].
@@ skipping 218 matching lines @@
         trace);
   }).catchError((error) {
     currentSchedule.signalError(error);
   });
 
   return delay.then((_) => cancelable.cancel());
 }
 
 /// An [AssetProvider] that provides the given set of assets.
 class MockProvider implements PackageProvider {
-  Iterable<String> get packages => _packages.keys;
+  Iterable<String> get packages => _assets.keys;
 
-  Map<String, _MockPackage> _packages;
+  Map<String, AssetSet> _assets;
 
   /// The set of assets for which [MockLoadException]s should be emitted if
   /// they're loaded.
   final _errors = new Set<AssetId>();
 
   /// The completer that [getAsset()] is waiting on to complete when paused.
   ///
   /// If `null` it will return the asset immediately.
   Completer _pauseCompleter;
 
   /// Tells the provider to wait during [getAsset] until [complete()]
   /// is called.
   ///
   /// Lets you test the asynchronous behavior of loading.
   void _pause() {
     _pauseCompleter = new Completer();
   }
 
   void _resume() {
     _pauseCompleter.complete();
     _pauseCompleter = null;
   }
 
-  MockProvider(assets,
-      Map<String, Iterable<Iterable<Transformer>>> transformers) {
-    var assetList;
-    if (assets is Map) {
-      assetList = assets.keys.map((asset) {
-        var id = new AssetId.parse(asset);
-        return new _MockAsset(id, assets[asset]);
-      });
-    } else if (assets is Iterable) {
-      assetList = assets.map((asset) {
-        var id = new AssetId.parse(asset);
-        var contents = pathos.basenameWithoutExtension(id.path);
-        return new _MockAsset(id, contents);
-      });
-    }
-
-    _packages = mapMapValues(groupBy(assetList, (asset) => asset.id.package),
-        (package, assets) {
-      var packageTransformers = transformers[package];
-      if (packageTransformers == null) packageTransformers = [];
-      return new _MockPackage(
-          new AssetSet.from(assets), packageTransformers.toList());
-    });
-
+  MockProvider(this._assets) {
     // If there are no assets or transformers, add a dummy package. This better
     // simulates the real world, where there'll always be at least the
     // entrypoint package.
-    if (_packages.isEmpty) {
-      _packages = {"app": new _MockPackage(new AssetSet(), [])};
+    if (_assets.isEmpty) {
+      _assets = {"app": new AssetSet()};
     }
   }
 
   void _modifyAsset(String name, String contents) {
     var id = new AssetId.parse(name);
     _errors.remove(id);
-    _packages[id.package].assets[id].contents = contents;
+    _assets[id.package][id].contents = contents;
   }
 
   void _setAssetError(String name) => _errors.add(new AssetId.parse(name));
 
   List<AssetId> listAssets(String package, {String within}) {
     if (within != null) {
       throw new UnimplementedError("Doesn't handle 'within' yet.");
     }
 
-    return _packages[package].assets.map((asset) => asset.id);
-  }
-
-  Iterable<Iterable<Transformer>> getTransformers(String package) {
-    var mockPackage = _packages[package];
-    if (mockPackage == null) {
-      throw new ArgumentError("No package named $package.");
-    }
-    return mockPackage.transformers;
+    return _assets[package].map((asset) => asset.id);
   }
 
   Future<Asset> getAsset(AssetId id) {
     // Eagerly load the asset so we can test an asset's value changing between
     // when a load starts and when it finishes.
-    var package = _packages[id.package];
+    var assets = _assets[id.package];
     var asset;
-    if (package != null) asset = package.assets[id];
+    if (assets != null) asset = assets[id];
 
     var hasError = _errors.contains(id);
 
     var future;
     if (_pauseCompleter != null) {
       future = _pauseCompleter.future;
     } else {
       future = new Future.value();
     }
 
     return future.then((_) {
       if (hasError) throw new MockLoadException(id);
       if (asset == null) throw new AssetNotFoundException(id);
       return asset;
     });
   }
 }
 
 /// Error thrown for assets with [setAssetError] set.
 class MockLoadException implements Exception {
   final AssetId id;
 
   MockLoadException(this.id);
 
   String toString() => "Error loading $id.";
 }
 
-/// Used by [MockProvider] to keep track of which assets and transformers exist
-/// for each package.
-class _MockPackage {
-  final AssetSet assets;
-  final List<List<Transformer>> transformers;
-
-  _MockPackage(this.assets, Iterable<Iterable<Transformer>> transformers)
-      : transformers = transformers.map((phase) => phase.toList()).toList();
-}
-
 /// An implementation of [Asset] that never hits the file system.
 class _MockAsset implements Asset {
   final AssetId id;
   String contents;
 
   _MockAsset(this.id, this.contents);
 
   Future<String> readAsString({Encoding encoding}) =>
       new Future.value(contents);
 
   Stream<List<int>> read() => throw new UnimplementedError();
 
   String toString() => "MockAsset $id $contents";
 }