Make an AstProvider available to AssistProcessors

pkg/analysis_server/test/services/correction/assist_test.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.
 
 import 'dart:async';
 
 import 'package:analysis_server/plugin/edit/assist/assist_core.dart';
 import 'package:analysis_server/plugin/edit/assist/assist_dart.dart';
 import 'package:analysis_server/protocol/protocol_generated.dart';
 import 'package:analysis_server/src/plugin/server_plugin.dart';
 import 'package:analysis_server/src/services/correction/assist.dart';
 import 'package:analysis_server/src/services/correction/assist_internal.dart';
 import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/ast/standard_resolution_map.dart';
 import 'package:analyzer/dart/element/element.dart';
 import 'package:analyzer/file_system/file_system.dart';
 import 'package:analyzer/source/package_map_resolver.dart';
+import 'package:analyzer/src/dart/analysis/ast_provider_context.dart';
+import 'package:analyzer/src/dart/analysis/ast_provider_driver.dart';
+import 'package:analyzer/src/dart/element/ast_provider.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:plugin/manager.dart';
 import 'package:plugin/plugin.dart';
 import 'package:test/test.dart';
 import 'package:test_reflective_loader/test_reflective_loader.dart';
 
 import '../../abstract_single_unit.dart';
 import 'flutter_util.dart';
 
@@ skipping 4610 matching lines @@
     List<Position> expectedPositions = _findResultPositions(expectedStrings);
     expect(group.positions, unorderedEquals(expectedPositions));
     if (expectedSuggestions != null) {
       expect(group.suggestions, unorderedEquals(expectedSuggestions));
     }
   }
 
   Future<List<Assist>> _computeAssists() async {
     CompilationUnitElement testUnitElement =
         resolutionMap.elementDeclaredByCompilationUnit(testUnit);
-    DartAssistContext assistContext = new _DartAssistContextForValues(
-        testUnitElement.source,
-        offset,
-        length,
-        testUnitElement.context,
-        testUnit);
+    DartAssistContext assistContext;
+    if (enableNewAnalysisDriver) {
+      assistContext = new _DartAssistContextForValues(
+          testUnitElement.source,
+          offset,
+          length,
+          testUnitElement.context,
+          new AstProviderForDriver(driver),
+          testUnit);
+    } else {
+      assistContext = new _DartAssistContextForValues(
+          testUnitElement.source,
+          offset,
+          length,
+          testUnitElement.context,
+          new AstProviderForContext(testUnitElement.context),
+          testUnit);
+    }
     AssistProcessor processor = new AssistProcessor(assistContext);
     return await processor.compute();
   }
 
   /**
    * Configures the [SourceFactory] to have the `flutter` package in
    * `/packages/flutter/lib` folder.
    */
   void _configureFlutterPkg(Map<String, String> pathToCode) {
     pathToCode.forEach((path, code) {
@@ skipping 52 matching lines @@
   @override
   final int selectionOffset;
 
   @override
   final int selectionLength;
 
   @override
   final AnalysisContext analysisContext;
 
   @override
+  final AstProvider astProvider;
+
+  @override
   final CompilationUnit unit;
 
   _DartAssistContextForValues(this.source, this.selectionOffset,
-      this.selectionLength, this.analysisContext, this.unit);
+      this.selectionLength, this.analysisContext, this.astProvider, this.unit);
 }