support part file completions

pkg/analysis_server/test/services/completion/completion_test_util.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 test.services.completion.util;
 
 import 'dart:async';
 
 import 'package:analysis_server/src/protocol.dart' as protocol show Element,
     ElementKind;
@@ skipping 433 matching lines @@
 
       // Update the index
       result.changeNotices.forEach((ChangeNotice notice) {
         CompilationUnit unit = notice.compilationUnit;
         if (unit != null) {
           index.indexUnit(context, unit);
         }
       });
 
       // If the unit has been resolved, then finish the completion
-      LibraryElement library = context.getLibraryElement(testSource);
-      if (library != null) {
-        CompilationUnit unit =
-            context.getResolvedCompilationUnit(testSource, library);
-        if (unit != null) {
-          request.unit = unit;
-          request.node =
-              new NodeLocator.con1(completionOffset).searchWithin(unit);
-          if (request.node is SimpleIdentifier) {
-            request.replacementOffset = request.node.offset;
-            request.replacementLength = request.node.length;
-          } else {
-            request.replacementOffset = request.offset;
-            request.replacementLength = 0;
-          }
-          if (request.replacementOffset == null) {
-            fail('expected non null');
-          }
-          resolved = true;
-          if (!fullAnalysis) {
-            break;
+      List<Source> libSourceList = context.getLibrariesContaining(testSource);
+      if (libSourceList.length > 0) {
+        LibraryElement library = context.getLibraryElement(libSourceList[0]);
+        if (library != null) {
+          CompilationUnit unit =
+              context.getResolvedCompilationUnit(testSource, library);
+          if (unit != null) {
+            request.unit = unit;
+            request.node =
+                new NodeLocator.con1(completionOffset).searchWithin(unit);
+            if (request.node is SimpleIdentifier) {
+              request.replacementOffset = request.node.offset;
+              request.replacementLength = request.node.length;
+            } else {
+              request.replacementOffset = request.offset;
+              request.replacementLength = 0;
+            }
+            if (request.replacementOffset == null) {
+              fail('expected non null');
+            }
+            resolved = true;
+            if (!fullAnalysis) {
+              break;
+            }
           }
         }
       }
 
       result = context.performAnalysisTask();
     }
     if (!resolved) {
       fail('expected unit to be resolved');
     }
     return computer.computeFull(request).then(assertFunction);
@@ skipping 271 matching lines @@
   CompletionSuggestion assertSuggestLocalTopLevelVar(String name,
       String returnType, [CompletionRelevance relevance =
       CompletionRelevance.DEFAULT]) {
     if (computer is LocalComputer) {
       return assertSuggestTopLevelVar(name, returnType, relevance);
     } else {
       return assertNotSuggested(name);
     }
   }
 
+  CompletionSuggestion assertSuggestNonLocalClass(String name,
+      [CompletionRelevance relevance = CompletionRelevance.DEFAULT,
+      CompletionSuggestionKind kind = CompletionSuggestionKind.INVOCATION]) {
+    return assertSuggestImportedClass(name, relevance, kind);
+  }
+
   Future computeFull(assertFunction(bool result), {bool fullAnalysis: true}) {
     return super.computeFull(
         assertFunction,
         fullAnalysis: fullAnalysis).then(assertCachedCompute);
   }
 
   test_ArgumentList() {
     // ArgumentList  MethodInvocation  ExpressionStatement  Block
     addSource('/libA.dart', '''
       library A;
@@ skipping 166 matching lines @@
       // the user may be either (1) entering a type for the assignment
       // or (2) starting a new statement.
       // Consider suggesting only types
       // if only spaces separates the 1st and 2nd identifiers.
       //assertNotSuggested('a');
       //assertNotSuggested('main');
       //assertNotSuggested('identical');
     });
   }
 
+  test_AssignmentExpression_type_newline() {
+    // SimpleIdentifier  TypeName  VariableDeclarationList
+    // VariableDeclarationStatement  Block
+    addTestSource('''
+      class A {} main() {
+        int a;
+        ^
+        b = 1;}''');
+    computeFast();
+    return computeFull((bool result) {
+      assertSuggestLocalClass('A');
+      assertSuggestImportedClass('int');
+      // Allow non-types preceding an identifier on LHS of assignment
+      // if newline follows first identifier
+      // because user is probably starting a new statement
+      assertSuggestLocalVariable('a', 'int');
+      assertSuggestLocalFunction('main', null);
+      assertSuggestImportedFunction('identical', 'bool');
+    });
+  }
+
   test_AssignmentExpression_type_partial() {
     // SimpleIdentifier  TypeName  VariableDeclarationList
     // VariableDeclarationStatement  Block
     addTestSource('''
       class A {} main() {
         int a;
         int^ b = 1;}''');
     computeFast();
     return computeFull((bool result) {
       assertSuggestLocalClass('A');
       assertSuggestImportedClass('int');
       // TODO (danrubel) When entering 1st of 2 identifiers on assignment LHS
       // the user may be either (1) entering a type for the assignment
       // or (2) starting a new statement.
       // Consider suggesting only types
       // if only spaces separates the 1st and 2nd identifiers.
       //assertNotSuggested('a');
       //assertNotSuggested('main');
       //assertNotSuggested('identical');
     });
   }
 
-  test_AssignmentExpression_type_newline() {
-    // SimpleIdentifier  TypeName  VariableDeclarationList
-    // VariableDeclarationStatement  Block
-    addTestSource('''
-      class A {} main() {
-        int a;
-        ^
-        b = 1;}''');
-    computeFast();
-    return computeFull((bool result) {
-      assertSuggestLocalClass('A');
-      assertSuggestImportedClass('int');
-      // Allow non-types preceding an identifier on LHS of assignment
-      // if newline follows first identifier
-      // because user is probably starting a new statement
-      assertSuggestLocalVariable('a', 'int');
-      assertSuggestLocalFunction('main', null);
-      assertSuggestImportedFunction('identical', 'bool');
-    });
-  }
-
   test_AssignmentExpression_type_partial_newline() {
     // SimpleIdentifier  TypeName  VariableDeclarationList
     // VariableDeclarationStatement  Block
     addTestSource('''
       class A {} main() {
         int a;
         i^
         b = 1;}''');
     computeFast();
     return computeFull((bool result) {
@@ skipping 1032 matching lines @@
       assertNotSuggested('m');
       assertNotSuggested('_n');
       assertNotSuggested('a');
       assertNotSuggested('A');
       assertNotSuggested('X');
       assertNotSuggested('Object');
       assertNotSuggested('==');
     });
   }
 
+  test_partFile_TypeName() {
+    // SimpleIdentifier  TypeName  ConstructorName
+    addSource('/testB.dart', '''
+      lib B;
+      int T1;
+      F1() { }
+      class X {X.c(); X._d(); z() {}}''');
+    addSource('/testA.dart', '''
+      library libA;
+      import "/testB.dart";
+      part "$testFile";
+      class A { }
+      var m;''');
+    addTestSource('''
+      part of libA;
+      class B { }
+      main() {new ^}''');
+    computeFast();
+    return computeFull((bool result) {
+      assertSuggestLocalClass('B');
+      assertSuggestImportedClass('Object');
+      assertSuggestImportedClass('X');
+      assertSuggestNonLocalClass('A');
+      assertNotSuggested('F1');
+      assertNotSuggested('T1');
+      assertNotSuggested('_d');
+      assertNotSuggested('z');
+      assertNotSuggested('m');
+    });
+  }
+
+  test_partFile_TypeName2() {
+    // SimpleIdentifier  TypeName  ConstructorName
+    addSource('/testB.dart', '''
+      lib B;
+      int T1;
+      F1() { }
+      class X {X.c(); X._d(); z() {}}''');
+    addSource('/testA.dart','''
+      part of libA;
+      class B { }''');
+    addTestSource( '''
+      library libA;
+      import "/testB.dart";
+      part "/testA.dart";
+      class A { }
+      main() {new ^}
+      var m;''');
+    computeFast();
+    return computeFull((bool result) {
+      assertSuggestLocalClass('A');
+      assertSuggestImportedClass('Object');
+      assertSuggestImportedClass('X');
+      assertSuggestNonLocalClass('B');
+      assertNotSuggested('F1');
+      assertNotSuggested('T1');
+      assertNotSuggested('_d');
+      assertNotSuggested('z');
+      assertNotSuggested('m');
+    });
+  }
+
   test_PrefixedIdentifier_class_const() {
     // SimpleIdentifier PrefixedIdentifier ExpressionStatement Block
     addSource('/testB.dart', '''
       lib B;
       class I {
         static const scI = 'boo';
         X get f => new A();
         get _g => new A();}
       class B implements I {
         static const int scB = 12;
@@ skipping 165 matching lines @@
   test_PrefixedIdentifier_propertyAccess() {
     // PrefixedIdentifier  ExpressionStatement  Block  BlockFunctionBody
     addTestSource('class A {String x; int get foo {x.^}');
     computeFast();
     return computeFull((bool result) {
       assertSuggestInvocationGetter('isEmpty', 'bool');
       assertSuggestInvocationMethod('compareTo', 'Comparable', 'int');
     });
   }
 
+  test_PrefixedIdentifier_propertyAccess_newStmt() {
+    // PrefixedIdentifier  ExpressionStatement  Block  BlockFunctionBody
+    addTestSource('class A {String x; int get foo {x.^ int y = 0;}');
+    computeFast();
+    return computeFull((bool result) {
+      assertSuggestInvocationGetter('isEmpty', 'bool');
+      assertSuggestInvocationMethod('compareTo', 'Comparable', 'int');
+    });
+  }
+
   test_PropertyAccess_expression() {
     // SimpleIdentifier  MethodInvocation  PropertyAccess  ExpressionStatement
     addTestSource('class A {a() {"hello".to^String().length}}');
     computeFast();
     return computeFull((bool result) {
       assertSuggestInvocationGetter('length', 'int');
       assertNotSuggested('A');
       assertNotSuggested('a');
       assertNotSuggested('Object');
       assertNotSuggested('==');
@@ skipping 169 matching lines @@
       assertNotSuggested('bar2');
       assertNotSuggested('_B');
       assertSuggestLocalClass('Y');
       assertSuggestLocalClass('C');
       assertSuggestLocalVariable('f', null);
       assertNotSuggested('x');
       assertNotSuggested('e');
     });
   }
 }