| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2014, the Dart project authors. Please see the AUTHORS file | |
| 2 // for details. All rights reserved. Use of this source code is governed by a | |
| 3 // BSD-style license that can be found in the LICENSE file. | |
| 4 | |
| 5 library test.services.completion.suggestion; | |
| 6 | |
| 7 import 'dart:async'; | |
| 8 | |
| 9 import 'package:analysis_server/plugin/protocol/protocol.dart'; | |
| 10 import 'package:analysis_server/src/provisional/completion/completion_core.dart' | |
| 11 show CompletionRequest, CompletionResult; | |
| 12 import 'package:analysis_server/src/services/completion/completion_core.dart'; | |
| 13 import 'package:analysis_server/src/services/completion/completion_manager.dart'
; | |
| 14 import 'package:analysis_server/src/services/completion/dart_completion_manager.
dart'; | |
| 15 import 'package:analysis_server/src/services/index/index.dart'; | |
| 16 import 'package:analysis_server/src/services/index/local_memory_index.dart'; | |
| 17 import 'package:analysis_server/src/services/search/search_engine.dart'; | |
| 18 import 'package:analysis_server/src/services/search/search_engine_internal.dart'
; | |
| 19 import 'package:analyzer/src/generated/engine.dart'; | |
| 20 import 'package:analyzer/src/generated/source.dart'; | |
| 21 import 'package:test_reflective_loader/test_reflective_loader.dart'; | |
| 22 import 'package:unittest/unittest.dart'; | |
| 23 | |
| 24 import '../../abstract_single_unit.dart'; | |
| 25 import '../../utils.dart'; | |
| 26 | |
| 27 main() { | |
| 28 initializeTestEnvironment(); | |
| 29 defineReflectiveTests(DartCompletionManagerTest); | |
| 30 } | |
| 31 | |
| 32 /** | |
| 33 * Returns a [Future] that completes after pumping the event queue [times] | |
| 34 * times. By default, this should pump the event queue enough times to allow | |
| 35 * any code to run, as long as it's not waiting on some external event. | |
| 36 */ | |
| 37 Future pumpEventQueue([int times = 20]) { | |
| 38 if (times == 0) return new Future.value(); | |
| 39 // We use a delayed future to allow microtask events to finish. The | |
| 40 // Future.value or Future() constructors use scheduleMicrotask themselves and | |
| 41 // would therefore not wait for microtask callbacks that are scheduled after | |
| 42 // invoking this method. | |
| 43 return new Future.delayed(Duration.ZERO, () => pumpEventQueue(times - 1)); | |
| 44 } | |
| 45 | |
| 46 @reflectiveTest | |
| 47 class DartCompletionManagerTest extends AbstractSingleUnitTest { | |
| 48 Index index; | |
| 49 SearchEngineImpl searchEngine; | |
| 50 Source source; | |
| 51 DartCompletionManager manager; | |
| 52 MockCompletionContributor contributor1; | |
| 53 MockCompletionContributor contributor2; | |
| 54 CompletionSuggestion suggestion1; | |
| 55 CompletionSuggestion suggestion2; | |
| 56 bool _continuePerformingAnalysis = true; | |
| 57 | |
| 58 void resolveLibrary() { | |
| 59 context.resolveCompilationUnit( | |
| 60 source, context.computeLibraryElement(source)); | |
| 61 } | |
| 62 | |
| 63 @override | |
| 64 void setUp() { | |
| 65 super.setUp(); | |
| 66 index = createLocalMemoryIndex(); | |
| 67 searchEngine = new SearchEngineImpl(index); | |
| 68 source = addSource('/does/not/exist.dart', ''); | |
| 69 manager = | |
| 70 new DartCompletionManager.create(context, searchEngine, source, []); | |
| 71 suggestion1 = new CompletionSuggestion(CompletionSuggestionKind.INVOCATION, | |
| 72 DART_RELEVANCE_DEFAULT, "suggestion1", 1, 1, false, false); | |
| 73 suggestion2 = new CompletionSuggestion(CompletionSuggestionKind.IDENTIFIER, | |
| 74 DART_RELEVANCE_DEFAULT, "suggestion2", 2, 2, false, false); | |
| 75 new Future(_performAnalysis); | |
| 76 } | |
| 77 | |
| 78 @override | |
| 79 void tearDown() { | |
| 80 _continuePerformingAnalysis = false; | |
| 81 } | |
| 82 | |
| 83 test_compute_fastAndFull() { | |
| 84 contributor1 = new MockCompletionContributor(suggestion1, null); | |
| 85 contributor2 = new MockCompletionContributor(null, suggestion2); | |
| 86 manager.contributors = [contributor1, contributor2]; | |
| 87 int count = 0; | |
| 88 bool done = false; | |
| 89 CompletionRequest completionRequest = | |
| 90 new CompletionRequestImpl(context, provider, searchEngine, source, 0); | |
| 91 manager.results(completionRequest).listen((CompletionResult r) { | |
| 92 bool isLast = r is CompletionResultImpl ? r.isLast : true; | |
| 93 switch (++count) { | |
| 94 case 1: | |
| 95 contributor1.assertCalls(context, source, 0, searchEngine); | |
| 96 expect(contributor1.fastCount, equals(1)); | |
| 97 expect(contributor1.fullCount, equals(0)); | |
| 98 contributor2.assertCalls(context, source, 0, searchEngine); | |
| 99 expect(contributor2.fastCount, equals(1)); | |
| 100 expect(contributor2.fullCount, equals(1)); | |
| 101 expect(isLast, isTrue); | |
| 102 expect(r.suggestions, hasLength(2)); | |
| 103 expect(r.suggestions, contains(suggestion1)); | |
| 104 expect(r.suggestions, contains(suggestion2)); | |
| 105 resolveLibrary(); | |
| 106 break; | |
| 107 default: | |
| 108 fail('unexpected'); | |
| 109 } | |
| 110 }, onDone: () { | |
| 111 done = true; | |
| 112 // There is only one notification | |
| 113 expect(count, equals(1)); | |
| 114 }); | |
| 115 return pumpEventQueue(250).then((_) { | |
| 116 expect(done, isTrue); | |
| 117 }); | |
| 118 } | |
| 119 | |
| 120 test_compute_fastOnly() { | |
| 121 contributor1 = new MockCompletionContributor(suggestion1, null); | |
| 122 contributor2 = new MockCompletionContributor(suggestion2, null); | |
| 123 manager.contributors = [contributor1, contributor2]; | |
| 124 int count = 0; | |
| 125 bool done = false; | |
| 126 CompletionRequest completionRequest = | |
| 127 new CompletionRequestImpl(context, provider, searchEngine, source, 0); | |
| 128 manager.results(completionRequest).listen((CompletionResult r) { | |
| 129 bool isLast = r is CompletionResultImpl ? r.isLast : true; | |
| 130 switch (++count) { | |
| 131 case 1: | |
| 132 contributor1.assertCalls(context, source, 0, searchEngine); | |
| 133 expect(contributor1.fastCount, equals(1)); | |
| 134 expect(contributor1.fullCount, equals(0)); | |
| 135 contributor2.assertCalls(context, source, 0, searchEngine); | |
| 136 expect(contributor2.fastCount, equals(1)); | |
| 137 expect(contributor2.fullCount, equals(0)); | |
| 138 expect(isLast, isTrue); | |
| 139 expect(r.suggestions, hasLength(2)); | |
| 140 expect(r.suggestions, contains(suggestion1)); | |
| 141 expect(r.suggestions, contains(suggestion2)); | |
| 142 break; | |
| 143 default: | |
| 144 fail('unexpected'); | |
| 145 } | |
| 146 }, onDone: () { | |
| 147 done = true; | |
| 148 expect(count, equals(1)); | |
| 149 }); | |
| 150 return pumpEventQueue().then((_) { | |
| 151 expect(done, isTrue); | |
| 152 }); | |
| 153 } | |
| 154 | |
| 155 void _performAnalysis() { | |
| 156 if (!_continuePerformingAnalysis) { | |
| 157 return; | |
| 158 } | |
| 159 context.performAnalysisTask(); | |
| 160 new Future(_performAnalysis); | |
| 161 } | |
| 162 } | |
| 163 | |
| 164 class MockCompletionContributor extends DartCompletionContributor { | |
| 165 final CompletionSuggestion fastSuggestion; | |
| 166 final CompletionSuggestion fullSuggestion; | |
| 167 int fastCount = 0; | |
| 168 int fullCount = 0; | |
| 169 DartCompletionRequest request; | |
| 170 | |
| 171 MockCompletionContributor(this.fastSuggestion, this.fullSuggestion); | |
| 172 | |
| 173 assertCalls(AnalysisContext context, Source source, int offset, | |
| 174 SearchEngine searchEngine) { | |
| 175 expect(request.context, equals(context)); | |
| 176 expect(request.source, equals(source)); | |
| 177 expect(request.offset, equals(offset)); | |
| 178 expect(request.searchEngine, equals(searchEngine)); | |
| 179 } | |
| 180 | |
| 181 assertFull(int fullCount) { | |
| 182 expect(this.fastCount, equals(1)); | |
| 183 expect(this.fullCount, equals(fullCount)); | |
| 184 } | |
| 185 | |
| 186 @override | |
| 187 bool computeFast(DartCompletionRequest request) { | |
| 188 this.request = request; | |
| 189 fastCount++; | |
| 190 if (fastSuggestion != null) { | |
| 191 request.addSuggestion(fastSuggestion); | |
| 192 } | |
| 193 return fastSuggestion != null; | |
| 194 } | |
| 195 | |
| 196 @override | |
| 197 Future<bool> computeFull(DartCompletionRequest request) { | |
| 198 this.request = request; | |
| 199 fullCount++; | |
| 200 if (fullSuggestion != null) { | |
| 201 request.addSuggestion(fullSuggestion); | |
| 202 } | |
| 203 return new Future.value(fullSuggestion != null); | |
| 204 } | |
| 205 } | |
| OLD | NEW |