Fix for searching top-level declarations in parts.

pkg/analyzer/test/src/dart/analysis/driver_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.
 
 library analyzer.test.driver;
 
 import 'dart:async';
 import 'dart:convert';
 
 import 'package:analyzer/dart/ast/ast.dart';
@@ skipping 878 matching lines @@
     provider.newFile(a, 'class A {}');
     provider.newFile(b, 'export "a.dart", class B {}');
     provider.newFile(c, 'import "d.dart"; class C {}');
     provider.newFile(d, 'class D {}');
 
     driver.addFile(a);
     driver.addFile(b);
     driver.addFile(c);
     // Don't add d.dart, it is referenced implicitly.
 
-    void assertDeclarations(List<TopLevelDeclarationInSource> declarations,
-        List<String> expectedFiles, List<bool> expectedIsExported) {
-      expect(expectedFiles, hasLength(expectedIsExported.length));
-      for (int i = 0; i < expectedFiles.length; i++) {
-        expect(declarations,
-            contains(predicate((TopLevelDeclarationInSource declaration) {
-          return declaration.source.fullName == expectedFiles[i] &&
-              declaration.isExported == expectedIsExported[i];
-        })));
-      }
-    }
-
-    assertDeclarations(
+    _assertTopLevelDeclarations(
         await driver.getTopLevelNameDeclarations('A'), [a, b], [false, true]);
 
-    assertDeclarations(
+    _assertTopLevelDeclarations(
         await driver.getTopLevelNameDeclarations('B'), [b], [false]);
 
-    assertDeclarations(
+    _assertTopLevelDeclarations(
         await driver.getTopLevelNameDeclarations('C'), [c], [false]);
 
-    assertDeclarations(
+    _assertTopLevelDeclarations(
         await driver.getTopLevelNameDeclarations('D'), [d], [false]);
 
-    assertDeclarations(await driver.getTopLevelNameDeclarations('X'), [], []);
+    _assertTopLevelDeclarations(
+        await driver.getTopLevelNameDeclarations('X'), [], []);
+  }
+
+  test_getTopLevelNameDeclarations_parts() async {
+    var a = _p('/test/lib/a.dart');
+    var b = _p('/test/lib/b.dart');
+    var c = _p('/test/lib/c.dart');
+
+    provider.newFile(
+        a,
+        r'''
+library lib;
+part 'b.dart';
+part 'c.dart';
+class A {}
+''');
+    provider.newFile(b, 'part of lib; class B {}');
+    provider.newFile(c, 'part of lib; class C {}');
+
+    driver.addFile(a);
+    driver.addFile(b);
+    driver.addFile(c);
+
+    _assertTopLevelDeclarations(
+        await driver.getTopLevelNameDeclarations('A'), [a], [false]);
+
+    _assertTopLevelDeclarations(
+        await driver.getTopLevelNameDeclarations('B'), [a], [false]);
+
+    _assertTopLevelDeclarations(
+        await driver.getTopLevelNameDeclarations('C'), [a], [false]);
+
+    _assertTopLevelDeclarations(
+        await driver.getTopLevelNameDeclarations('X'), [], []);
   }
 
   test_getUnitElement() async {
     String content = r'''
 foo(int p) {}
 main() {
   foo(42);
 }
 ''';
     addTestFile(content);
@@ skipping 489 matching lines @@
     addTestFile('int f() => 42;');
     await _waitForIdle();
 
     expect(allStatuses, hasLength(2));
     expect(allStatuses[0].isAnalyzing, isTrue);
     expect(allStatuses[0].isIdle, isFalse);
     expect(allStatuses[1].isAnalyzing, isFalse);
     expect(allStatuses[1].isIdle, isTrue);
   }
 
+  void _assertTopLevelDeclarations(
+      List<TopLevelDeclarationInSource> declarations,
+      List<String> expectedFiles,
+      List<bool> expectedIsExported) {
+    expect(expectedFiles, hasLength(expectedIsExported.length));
+    for (int i = 0; i < expectedFiles.length; i++) {
+      expect(declarations,
+          contains(predicate((TopLevelDeclarationInSource declaration) {
+        return declaration.source.fullName == expectedFiles[i] &&
+            declaration.isExported == expectedIsExported[i];
+      })));
+    }
+  }
+
   ClassDeclaration _getClass(CompilationUnit unit, String name) {
     for (CompilationUnitMember declaration in unit.declarations) {
       if (declaration is ClassDeclaration) {
         if (declaration.name.name == name) {
           return declaration;
         }
       }
     }
     fail('Cannot find the class $name in\n$unit');
     return null;
@@ skipping 84 matching lines @@
   String _p(String path) => provider.convertPath(path);
 
   Future<Null> _waitForIdle() async {
     await idleStatusMonitor.signal;
   }
 
   static String _md5(String content) {
     return hex.encode(md5.convert(UTF8.encode(content)).bytes);
   }
 }