refactor and cleanup code completion

pkg/analysis_services/test/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_services/completion/completion_computer.dart';
 import 'package:analysis_services/completion/completion_suggestion.dart';
 import 'package:analysis_services/index/index.dart';
 import 'package:analysis_services/index/local_memory_index.dart';
+import 'package:analysis_services/src/completion/dart_completion_manager.dart';
 import 'package:analysis_services/src/search/search_engine.dart';
 import 'package:analysis_testing/abstract_context.dart';
 import 'package:analysis_testing/mock_sdk.dart';
 import 'package:analyzer/src/generated/ast.dart';
 import 'package:analyzer/src/generated/element.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:unittest/unittest.dart';
 
 class AbstractCompletionTest extends AbstractContextTest {
   Index index;
   SearchEngineImpl searchEngine;
-  CompletionComputer computer;
+  DartCompletionComputer computer;
   String testFile = '/completionTest.dart';
   Source testSource;
   int completionOffset;
-  List<CompletionSuggestion> suggestions;
   bool _computeFastCalled = false;
+  DartCompletionRequest request;
 
   void addResolvedUnit(String file, String code) {
     Source source = addSource(file, code);
     CompilationUnit unit = resolveLibraryUnit(source);
     index.indexUnit(context, unit);
   }
 
   void addTestSource(String content) {
     expect(completionOffset, isNull, reason: 'Call addTestUnit exactly once');
     completionOffset = content.indexOf('^');
     expect(completionOffset, isNot(equals(-1)), reason: 'missing ^');
     int nextOffset = content.indexOf('^', completionOffset + 1);
     expect(nextOffset, equals(-1), reason: 'too many ^');
     content = content.substring(0, completionOffset) +
         content.substring(completionOffset + 1);
     testSource = addSource(testFile, content);
+    request = new DartCompletionRequest(
+        context,
+        searchEngine,
+        testSource,
+        completionOffset);
   }
 
   void assertNotSuggested(String completion) {
-    if (suggestions.any((cs) => cs.completion == completion)) {
+    if (request.suggestions.any((cs) => cs.completion == completion)) {
       fail('did not expect completion: $completion');
     }
   }
 
   void assertSuggest(CompletionSuggestionKind kind, String completion,
       [CompletionRelevance relevance = CompletionRelevance.DEFAULT, bool isDeprecated
       = false, bool isPotential = false]) {
-    var cs;
-    suggestions.forEach((s) {
+    CompletionSuggestion cs;
+    request.suggestions.forEach((s) {
       if (s.completion == completion) {
         if (cs == null) {
           cs = s;
         } else {
-          var match =
-              suggestions.where((s) => s.completion == completion).toList();
-          fail('expected exactly one $completion but found > 1\n $match');
+          List<CompletionSuggestion> matchSuggestions =
+              request.suggestions.where((s) => s.completion == completion).toList();
+          fail(
+              'expected exactly one $completion but found > 1\n $matchSuggestions');
         }
       }
     });
     if (cs == null) {
-      var completions = suggestions.map((s) => s.completion).toList();
+      List<CompletionSuggestion> completions =
+          request.suggestions.map((s) => s.completion).toList();
       fail('expected "$completion" but found\n $completions');
     }
     expect(cs.kind, equals(kind));
     expect(cs.relevance, equals(relevance));
     expect(cs.selectionOffset, equals(completion.length));
     expect(cs.selectionLength, equals(0));
     expect(cs.isDeprecated, equals(isDeprecated));
     expect(cs.isPotential, equals(isPotential));
   }
 
@@ skipping 18 matching lines @@
   }
 
   void assertSuggestLibraryPrefix(String completion,
       [CompletionRelevance relevance = CompletionRelevance.DEFAULT]) {
     assertSuggest(
         CompletionSuggestionKind.LIBRARY_PREFIX,
         completion,
         relevance);
   }
 
+  void assertSuggestLocalVariable(String completion,
+      [CompletionRelevance relevance = CompletionRelevance.DEFAULT]) {
+    assertSuggest(
+        CompletionSuggestionKind.LOCAL_VARIABLE,
+        completion,
+        relevance);
+  }
+
   void assertSuggestMethod(String className, [CompletionRelevance relevance =
       CompletionRelevance.DEFAULT]) {
     assertSuggest(CompletionSuggestionKind.METHOD, className, relevance);
   }
 
   void assertSuggestMethodName(String completion, [CompletionRelevance relevance
       = CompletionRelevance.DEFAULT]) {
     assertSuggest(CompletionSuggestionKind.METHOD_NAME, completion, relevance);
   }
 
   void assertSuggestParameter(String completion, [CompletionRelevance relevance
       = CompletionRelevance.DEFAULT]) {
     assertSuggest(CompletionSuggestionKind.PARAMETER, completion, relevance);
   }
 
   void assertSuggestSetter(String className, [CompletionRelevance relevance =
       CompletionRelevance.DEFAULT]) {
     assertSuggest(CompletionSuggestionKind.SETTER, className, relevance);
   }
 
   void assertSuggestTopLevelVar(String completion,
       [CompletionRelevance relevance = CompletionRelevance.DEFAULT]) {
     assertSuggest(
         CompletionSuggestionKind.TOP_LEVEL_VARIABLE,
         completion,
         relevance);
   }
 
-  void assertSuggestVariable(String completion, [CompletionRelevance relevance =
-      CompletionRelevance.DEFAULT]) {
-    assertSuggest(
-        CompletionSuggestionKind.LOCAL_VARIABLE,
-        completion,
-        relevance);
-  }
-
   bool computeFast() {
     _computeFastCalled = true;
-    computer
-        ..context = context
-        ..searchEngine = searchEngine
-        ..source = testSource
-        ..offset = completionOffset;
-    suggestions = [];
     CompilationUnit unit = context.parseCompilationUnit(testSource);
-    AstNode node = new NodeLocator.con1(completionOffset).searchWithin(unit);
-    return computer.computeFast(unit, node, suggestions);
+    request.unit = unit;
+    request.node = new NodeLocator.con1(completionOffset).searchWithin(unit);
+    return computer.computeFast(request);
   }
 
-  Future<bool> computeFull() {
+  Future<bool> computeFull([bool fullAnalysis = false]) {
     if (!_computeFastCalled) {
       expect(computeFast(), isFalse);
     }
     var result = context.performAnalysisTask();
+    bool resolved = false;
     while (result.hasMoreWork) {
+
+      // 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);
+          resolved = true;
+          if (!fullAnalysis) {
+            break;
+          }
+        }
+      }
+
       result = context.performAnalysisTask();
     }
-    LibraryElement library = context.getLibraryElement(testSource);
-    expect(library, isNotNull);
-    var unit = context.getResolvedCompilationUnit(testSource, library);
-    expect(unit, isNotNull);
-    AstNode node = new NodeLocator.con1(completionOffset).searchWithin(unit);
-    return computer.computeFull(unit, node, suggestions);
+    if (!resolved) {
+      fail('expected unit to be resolved');
+    }
+    return computer.computeFull(request);
   }
 
   @override
   void setUp() {
     super.setUp();
     index = createLocalMemoryIndex();
     searchEngine = new SearchEngineImpl(index);
     addResolvedUnit(MockSdk.LIB_CORE.path, MockSdk.LIB_CORE.content);
   }
 }