fix #603, support mock objects

test/codegen_test.dart

 // 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.
 
 /// Tests code generation.
 ///
 /// Runs Dart Dev Compiler on all input in the `codegen` directory and checks
 /// that the output is what we expected.
 library dev_compiler.test.codegen_test;
 
@@ skipping 48 matching lines @@
 // TODO(jmesserly): switch this to a .packages file.
 final packageUrlMappings = {
   'package:expect/expect.dart': path.join(codegenDir, 'expect.dart'),
   'package:async_helper/async_helper.dart':
       path.join(codegenDir, 'async_helper.dart'),
   'package:js/js.dart': path.join(codegenDir, 'packages', 'js', 'js.dart')
 };
 
 final codeCoverage = Platform.environment.containsKey('COVERALLS_TOKEN');
 
+RegExp filePattern;
+
 main(List<String> arguments) {
   if (arguments == null) arguments = [];
   ArgResults args = argParser.parse(arguments);
-  var filePattern = new RegExp(args.rest.length > 0 ? args.rest[0] : '.');
+  filePattern = new RegExp(args.rest.length > 0 ? args.rest[0] : '.');
 
   var sdkDir = path.join(repoDirectory, 'gen', 'patched_sdk');
   var sdkSummaryFile =
       path.join(testDirectory, '..', 'lib', 'runtime', 'dart_sdk.sum');
   var analyzerOptions = new AnalyzerOptions(
       customUrlMappings: packageUrlMappings,
       dartSdkSummaryPath: sdkSummaryFile);
   var compiler = new ModuleCompiler(analyzerOptions);
 
   // Build packages tests depend on.
   _buildAllPackages(compiler);
 
   var testDirs = [
     'language',
     'corelib',
     path.join('lib', 'convert'),
     path.join('lib', 'html'),
     path.join('lib', 'math'),
     path.join('lib', 'typed_data'),
   ];
 
   // Copy all of the test files and expanded multitest files to
   // gen/codegen_tests. We'll compile from there.
-  var testFiles = _setUpTests(testDirs, filePattern);
+  var testFiles = _setUpTests(testDirs);
 
   // Our default compiler options. Individual tests can override these.
   var defaultOptions = ['--no-source-map', '--no-summarize'];
   var compilerArgParser = CompilerOptions.addArguments(new ArgParser());
 
   // Compile each test file to JS and put the result in gen/codegen_output.
   for (var testFile in testFiles) {
     var relativePath = path.relative(testFile, from: codegenTestDir);
 
     // Only compile the top-level files for generating coverage.
@@ skipping 97 matching lines @@
     expectFile.writeAsStringSync("//FAILED TO COMPILE");
   }
 }
 
 void _buildAllPackages(ModuleCompiler compiler) {
   group('dartdevc package', () {
     _buildPackages(compiler, codegenOutputDir, codegenExpectDir);
 
     var packages = ['matcher', 'path', 'stack_trace'];
     for (var package in packages) {
+      if (!filePattern.hasMatch(package)) continue;
       test(package, () {
         _buildPackage(compiler, codegenOutputDir, codegenExpectDir, package);
       });
     }
 
+    if (!filePattern.hasMatch('unittest')) return;
+
     test('unittest', () {
       // Only build files applicable to the web - html_*.dart and its
       // internal dependences.
       _buildPackage(compiler, codegenOutputDir, codegenExpectDir, "unittest",
           packageFiles: [
             'unittest.dart',
             'html_config.dart',
             'html_individual_config.dart',
             'html_enhanced_config.dart'
           ]);
     });
   });
 
+  if (!filePattern.hasMatch('sunflower')) return;
+
   test('dartdevc sunflower', () {
     _buildSunflower(compiler, codegenOutputDir, codegenExpectDir);
   });
 }
 
 void _buildSunflower(
     ModuleCompiler compiler, String outputDir, String expectDir) {
   var baseDir = path.join(codegenDir, 'sunflower');
   var files = ['sunflower', 'circle', 'painter']
       .map((f) => path.join(baseDir, '$f.dart'))
       .toList();
   var input = new BuildUnit('sunflower', baseDir, files, _moduleForLibrary);
   var options = new CompilerOptions(summarizeApi: false);
 
   var built = compiler.compile(input, options);
   _writeModule(path.join(outputDir, 'sunflower', 'sunflower'),
       path.join(expectDir, 'sunflower', 'sunflower'), built);
 }
 
 void _buildPackages(
     ModuleCompiler compiler, String outputDir, String expectDir) {
   // Note: we don't summarize these, as we're going to rely on our in-memory
   // shared analysis context for caching, and `_moduleForLibrary` below
   // understands these are from other modules.
   var options = new CompilerOptions(sourceMap: false, summarizeApi: false);
 
   for (var uri in packageUrlMappings.keys) {
+    if (!filePattern.hasMatch(uri)) return;
+
     assert(uri.startsWith('package:'));
     var uriPath = uri.substring('package:'.length);
     var name = path.basenameWithoutExtension(uriPath);
     test(name, () {
       var input = new BuildUnit(name, codegenDir, [uri], _moduleForLibrary);
       var built = compiler.compile(input, options);
 
       var outPath = path.join(outputDir, path.withoutExtension(uriPath));
       var expectPath = path.join(expectDir, path.withoutExtension(uriPath));
       _writeModule(outPath, expectPath, built);
@@ skipping 37 matching lines @@
 }
 
 String _moduleForLibrary(Source source) {
   var scheme = source.uri.scheme;
   if (scheme == 'package') {
     return source.uri.pathSegments.first;
   }
   throw new Exception('Module not found for library "${source.fullName}"');
 }
 
-List<String> _setUpTests(List<String> testDirs, RegExp filePattern) {
+List<String> _setUpTests(List<String> testDirs) {
   var testFiles = <String>[];
 
   for (var testDir in testDirs) {
-    for (var file in _listFiles(path.join(codegenDir, testDir), filePattern,
-        recursive: true)) {
+    for (var file
+        in _listFiles(path.join(codegenDir, testDir), recursive: true)) {
       var relativePath = path.relative(file, from: codegenDir);
       var outputPath = path.join(codegenTestDir, relativePath);
 
       _ensureDirectory(path.dirname(outputPath));
 
       // Copy it over. We do this even for multitests because import_self_test
       // is a multitest, yet imports its own unexpanded form (!).
       new File(file).copySync(outputPath);
 
       if (file.endsWith("_test.dart")) {
@@ skipping 16 matching lines @@
           });
         } else {
           // It's a single test suite.
           testFiles.add(outputPath);
         }
       }
     }
   }
 
   // Also include the other special files that live at the top level directory.
-  for (var file in _listFiles(codegenDir, filePattern)) {
+  for (var file in _listFiles(codegenDir)) {
     var relativePath = path.relative(file, from: codegenDir);
     var outputPath = path.join(codegenTestDir, relativePath);
 
     new File(file).copySync(outputPath);
     if (file.endsWith(".dart")) {
       testFiles.add(outputPath);
     }
   }
 
   return testFiles;
 }
 
 /// Recursively creates [dir] if it doesn't exist.
 void _ensureDirectory(String dir) {
   new Directory(dir).createSync(recursive: true);
 }
 
 /// Lists all of the files within [dir] that match [filePattern].
-Iterable<String> _listFiles(String dir, RegExp filePattern,
-    {bool recursive: false}) {
+Iterable<String> _listFiles(String dir, {bool recursive: false}) {
   return new Directory(dir)
       .listSync(recursive: recursive, followLinks: false)
       .where((entry) {
     if (entry is! File) return false;
 
     var filePath = entry.path;
     if (!filePattern.hasMatch(filePath)) return false;
 
     return true;
   }).map((file) => file.path);
@@ skipping 33 matching lines @@
 /// Simplified from ParseDartTask.resolveDirective.
 String _resolveDirective(UriBasedDirective directive) {
   StringLiteral uriLiteral = directive.uri;
   String uriContent = uriLiteral.stringValue;
   if (uriContent != null) {
     uriContent = uriContent.trim();
     directive.uriContent = uriContent;
   }
   return directive.validate() == null ? uriContent : null;
 }