Enable re-export in docgen. Work in progress. More to come.

pkg/docgen/test/single_library_test.dart

 library single_library_test;
 
 import 'dart:io';
 
 import 'package:path/path.dart' as path;
 import 'package:unittest/unittest.dart';
 
 import '../lib/docgen.dart';
 
 const String DART_LIBRARY = '''
@@ skipping 32 matching lines @@
   group('Generate docs for', () {
     test('one simple file.', () {
       var temporaryDir = Directory.systemTemp.createTempSync('single_library_');
       var fileName = path.join(temporaryDir.path, 'temp.dart');
       var file = new File(fileName);
       file.writeAsStringSync(DART_LIBRARY);
       getMirrorSystem([new Uri.file(fileName)])
         .then(expectAsync1((mirrorSystem) {
           var testLibraryUri = new Uri(scheme: 'file',
               path: path.absolute(fileName));
-          var library = generateLibrary(mirrorSystem.libraries[testLibraryUri]);
+          var library = new Library(mirrorSystem.libraries[testLibraryUri]);
           expect(library is Library, isTrue);
 
           var classTypes = library.classes;
-          expect(classTypes is ClassGroup, isTrue);
-
           var classes = [];
-          classes.addAll(classTypes.classes.values);
-          classes.addAll(classTypes.errors.values);
+          classes.addAll(classTypes.values);
+          classes.addAll(library.errors.values);
           expect(classes.every((e) => e is Class), isTrue);
 
-          expect(classTypes.typedefs.values.every((e) => e is Typedef), isTrue);
+          expect(library.typedefs.values.every((e) => e is Typedef), isTrue);
 
           var classMethodTypes = [];
           classes.forEach((e) {
             classMethodTypes.add(e.methods);
             classMethodTypes.add(e.inheritedMethods);
           });
-          expect(classMethodTypes.every((e) => e is MethodGroup), isTrue);
+          expect(classMethodTypes.every((e) => e is Map<String, Method>), isTrue);
 
           var classMethods = [];
           classMethodTypes.forEach((e) {
-            classMethods.addAll(e.setters.values);
-            classMethods.addAll(e.getters.values);
-            classMethods.addAll(e.constructors.values);
-            classMethods.addAll(e.operators.values);
-            classMethods.addAll(e.regularMethods.values);
+            classMethods.addAll(e.values);
           });
           expect(classMethods.every((e) => e is Method), isTrue);
 
           var methodParameters = [];
           classMethods.forEach((e) {
             methodParameters.addAll(e.parameters.values);
           });
           expect(methodParameters.every((e) => e is Parameter), isTrue);
 
           var functionTypes = library.functions;
-          expect(functionTypes is MethodGroup, isTrue);
+          expect(functionTypes is Map<String, Method>, isTrue);
 
           var functions = [];
-          functions.addAll(functionTypes.setters.values);
-          functions.addAll(functionTypes.getters.values);
-          functions.addAll(functionTypes.constructors.values);
-          functions.addAll(functionTypes.operators.values);
-          functions.addAll(functionTypes.regularMethods.values);
+          functions.addAll(functionTypes.values);
           expect(functions.every((e) => e is Method), isTrue);
 
           var functionParameters = [];
           functions.forEach((e) {
             functionParameters.addAll(e.parameters.values);
           });
           expect(functionParameters.every((e) => e is Parameter), isTrue);
 
           var variables = library.variables.values;
           expect(variables.every((e) => e is Variable), isTrue);
 
           /// Testing fixReference
           // Testing Doc comment for class [A].
           var libraryMirror = mirrorSystem.libraries[testLibraryUri];
           var classMirror = libraryMirror.classes.values.first;
-          var classDocComment = fixReference('A', libraryMirror,
-              classMirror, null).children.first.text;
-          expect(classDocComment == 'test.A', isTrue);
+          var classDocComment = library.fixReference('A').children.first.text;
+          expect(classDocComment, 'test.A');
 
           // Test for linking to parameter [A]
-          var methodMirror = classMirror.methods['doThis'];
-          var methodParameterDocComment = fixReference('A', libraryMirror,
-              classMirror, methodMirror).children.first.text;
-          // TODO(alanknight) : We're not supporting linking to parameters yet
-          // expect(methodParameterDocComment == 'test.A.doThis#A', isTrue);
+          var method = getDocgenObject(classMirror.methods['doThis']);
+          var methodParameterDocComment = method.fixReference(
+              'A').children.first.text;
+          expect(methodParameterDocComment, 'test.A.doThis.A');
 
           // Testing trying to refer to doThis function
-          var methodDocComment = fixReference('doThis', libraryMirror,
-              classMirror, methodMirror).children.first.text;
-          expect(methodDocComment == 'test.A.doThis', isTrue);
+          var methodDocComment = method.fixReference(
+              'doThis').children.first.text;
+          expect(methodDocComment, 'test.A.doThis');
 
           // Testing something with no reference
-          var libraryDocComment = fixReference('foobar', libraryMirror,
-              classMirror, methodMirror).text;
-          expect(libraryDocComment == 'foobar', isTrue);
+          var libraryDocComment = method.fixReference('foobar').text;
+          expect(libraryDocComment, 'foobar');
         })).whenComplete(() => temporaryDir.deleteSync(recursive: true));
     });
   });
 }