Allow multiple-entry libraries in code_transformers

pkg/code_transformers/test/resolver_test.dart

 // Copyright (c) 2014, 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 code_transformers.test.resolver_test;
 
 import 'dart:async';
 import 'dart:io' show File, Platform;
 
 import 'package:barback/barback.dart';
 import 'package:code_transformers/resolver.dart';
 import 'package:code_transformers/tests.dart';
 import 'package:path/path.dart' as path;
 import 'package:unittest/compact_vm_config.dart';
 import 'package:unittest/unittest.dart';
 
 main() {
   useCompactVMConfiguration();
-
-  var sdkDir = dartSdkDirectory;
-  if (sdkDir == null) {
-    // If we cannot find the SDK dir, then assume this is being run from Dart's
-    // source directory and this script is the main script.
-    sdkDir = path.join(
-        path.dirname(path.fromUri(Platform.script)), '..', '..', '..', 'sdk');
-  }
-
   var entryPoint = new AssetId('a', 'web/main.dart');
-  var resolvers = new Resolvers(sdkDir);
+  var resolvers = new Resolvers(testingDartSdkDirectory);
 
   Future validateResolver({Map<String, String> inputs, void validator(Resolver),
       List<String> messages: const[]}) {
     return applyTransformers(
           [[new TestTransformer(resolvers, entryPoint, validator)]],
           inputs: inputs,
           messages: messages);
   }
 
   group('Resolver', () {
 
     test('should handle empty files', () {
       return validateResolver(
           inputs: {
             'a|web/main.dart': '',
           },
           validator: (resolver) {
             var source = resolver.sources[entryPoint];
             expect(source.modificationStamp, 1);
 
-            var lib = resolver.entryLibrary;
+            var lib = resolver.getLibrary(entryPoint);
             expect(lib, isNotNull);
             expect(lib.entryPoint, isNull);
           });
     });
 
     test('should update when sources change', () {
       return validateResolver(
           inputs: {
             'a|web/main.dart': ''' main() {} ''',
           },
           validator: (resolver) {
             var source = resolver.sources[entryPoint];
             expect(source.modificationStamp, 2);
 
-            var lib = resolver.entryLibrary;
+            var lib = resolver.getLibrary(entryPoint);
             expect(lib, isNotNull);
             expect(lib.entryPoint, isNotNull);
           });
     });
 
     test('should follow imports', () {
       return validateResolver(
           inputs: {
             'a|web/main.dart': '''
               import 'a.dart';
 
               main() {
               } ''',
             'a|web/a.dart': '''
               library a;
               ''',
           },
           validator: (resolver) {
-            var lib = resolver.entryLibrary;
+            var lib = resolver.getLibrary(entryPoint);
             expect(lib.importedLibraries.length, 2);
             var libA = lib.importedLibraries.where((l) => l.name == 'a').single;
             expect(libA.getType('Foo'), isNull);
           });
     });
 
     test('should update changed imports', () {
       return validateResolver(
           inputs: {
             'a|web/main.dart': '''
               import 'a.dart';
 
               main() {
               } ''',
             'a|web/a.dart': '''
               library a;
               class Foo {}
               ''',
           },
           validator: (resolver) {
-            var lib = resolver.entryLibrary;
+            var lib = resolver.getLibrary(entryPoint);
             expect(lib.importedLibraries.length, 2);
             var libA = lib.importedLibraries.where((l) => l.name == 'a').single;
             expect(libA.getType('Foo'), isNotNull);
           });
     });
 
     test('should follow package imports', () {
       return validateResolver(
           inputs: {
             'a|web/main.dart': '''
               import 'package:b/b.dart';
 
               main() {
               } ''',
             'b|lib/b.dart': '''
               library b;
               ''',
           },
           validator: (resolver) {
-            var lib = resolver.entryLibrary;
+            var lib = resolver.getLibrary(entryPoint);
             expect(lib.importedLibraries.length, 2);
             var libB = lib.importedLibraries.where((l) => l.name == 'b').single;
             expect(libB.getType('Foo'), isNull);
           });
     });
 
     test('should update on changed package imports', () {
       return validateResolver(
           inputs: {
             'a|web/main.dart': '''
               import 'package:b/b.dart';
 
               main() {
               } ''',
             'b|lib/b.dart': '''
               library b;
               class Bar {}
               ''',
           },
           validator: (resolver) {
-            var lib = resolver.entryLibrary;
+            var lib = resolver.getLibrary(entryPoint);
             expect(lib.importedLibraries.length, 2);
             var libB = lib.importedLibraries.where((l) => l.name == 'b').single;
             expect(libB.getType('Bar'), isNotNull);
           });
     });
 
     test('should handle deleted files', () {
       return validateResolver(
           inputs: {
             'a|web/main.dart': '''
               import 'package:b/b.dart';
 
               main() {
               } ''',
           },
           messages: [
             'error: Unable to find asset for "package:b/b.dart"',
             'error: Unable to find asset for "package:b/b.dart"',
           ],
           validator: (resolver) {
-            var lib = resolver.entryLibrary;
+            var lib = resolver.getLibrary(entryPoint);
             expect(lib.importedLibraries.length, 1);
           });
     });
 
     test('should fail on absolute URIs', () {
       return validateResolver(
           inputs: {
             'a|web/main.dart': '''
               import '/b.dart';
 
               main() {
               } ''',
           },
           messages: [
             // First from the AST walker
-            'error: absolute paths not allowed: "/b.dart" (main.dart 0 14)',
+            'error: absolute paths not allowed: "/b.dart" (web/main.dart 0 14)',
             // Then two from the resolver.
             'error: absolute paths not allowed: "/b.dart"',
             'error: absolute paths not allowed: "/b.dart"',
           ],
           validator: (resolver) {
-            var lib = resolver.entryLibrary;
+            var lib = resolver.getLibrary(entryPoint);
             expect(lib.importedLibraries.length, 1);
           });
     });
 
     test('should list all libraries', () {
       return validateResolver(
           inputs: {
             'a|web/main.dart': '''
               library a.main;
               import 'package:a/a.dart';
@@ skipping 122 matching lines @@
     });
 
     test('handles parallel resolves', () {
       return Future.wait([
         validateResolver(
           inputs: {
             'a|web/main.dart': '''
                 library foo;'''
           },
           validator: (resolver) {
-            expect(resolver.entryLibrary.name, 'foo');
+            expect(resolver.getLibrary(entryPoint).name, 'foo');
           }),
         validateResolver(
           inputs: {
             'a|web/main.dart': '''
                 library bar;'''
           },
           validator: (resolver) {
-            expect(resolver.entryLibrary.name, 'bar');
+            expect(resolver.getLibrary(entryPoint).name, 'bar');
           }),
       ]);
     });
   });
 }
 
 class TestTransformer extends Transformer with ResolverTransformer {
   final AssetId primary;
   final Function validator;
 
   TestTransformer(Resolvers resolvers, this.primary, this.validator) {
     this.resolvers = resolvers;
   }
 
   Future<bool> isPrimary(Asset input) =>
       new Future.value(input.id == primary);
 
   applyResolver(Transform transform, Resolver resolver) {
     return validator(resolver);
   }
 }