Run refactoring tests with the new analysis driver.

pkg/analysis_server/test/services/refactoring/abstract_refactoring.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.refactoring;
 
 import 'dart:async';
 
 import 'package:analysis_server/plugin/protocol/protocol.dart'
     show
         RefactoringProblem,
         RefactoringProblemSeverity,
         SourceChange,
         SourceEdit,
         SourceFileEdit;
 import 'package:analysis_server/src/services/correction/status.dart';
 import 'package:analysis_server/src/services/index/index.dart';
 import 'package:analysis_server/src/services/refactoring/refactoring.dart';
+import 'package:analysis_server/src/services/search/search_engine.dart';
 import 'package:analysis_server/src/services/search/search_engine_internal.dart';
+import 'package:analysis_server/src/services/search/search_engine_internal2.dart';
 import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/element/element.dart' show Element;
 import 'package:analyzer/file_system/file_system.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:test/test.dart';
 
 import '../../abstract_single_unit.dart';
 
 int findIdentifierLength(String search) {
   int length = 0;
   while (length < search.length) {
     int c = search.codeUnitAt(length);
     if (!(c >= 'a'.codeUnitAt(0) && c <= 'z'.codeUnitAt(0) ||
         c >= 'A'.codeUnitAt(0) && c <= 'Z'.codeUnitAt(0) ||
         c >= '0'.codeUnitAt(0) && c <= '9'.codeUnitAt(0))) {
       break;
     }
     length++;
   }
   return length;
 }
 
 /**
  * The base class for all [Refactoring] tests.
  */
 abstract class RefactoringTest extends AbstractSingleUnitTest {
   Index index;
-  SearchEngineImpl searchEngine;
+  SearchEngine searchEngine;
 
   SourceChange refactoringChange;
 
   Refactoring get refactoring;
 
   /**
    * Asserts that [refactoringChange] contains a [FileEdit] for the file
    * with the given [path], and it results the [expectedCode].
    */
   void assertFileChangeResult(String path, String expectedCode) {
@@ skipping 97 matching lines @@
   /**
    * Completes with a fully resolved unit that contains the [element].
    */
   Future<CompilationUnit> getResolvedUnitWithElement(Element element) async {
     return element.context
         .resolveCompilationUnit(element.source, element.library);
   }
 
   Future<Null> indexTestUnit(String code) async {
     await resolveTestUnit(code);
-    index.indexUnit(testUnit);
+    if (!enableNewAnalysisDriver) {
+      index.indexUnit(testUnit);
+    }
   }
 
   Future<Null> indexUnit(String file, String code) async {
     Source source = addSource(file, code);
-    CompilationUnit unit = await resolveLibraryUnit(source);
-    index.indexUnit(unit);
+    if (!enableNewAnalysisDriver) {
+      CompilationUnit unit = await resolveLibraryUnit(source);
+      index.indexUnit(unit);
+    }
   }
 
   void setUp() {
     super.setUp();
-    index = createMemoryIndex();
-    searchEngine = new SearchEngineImpl(index);
+    if (enableNewAnalysisDriver) {
+      searchEngine = new SearchEngineImpl2([driver]);
+    } else {
+      index = createMemoryIndex();
+      searchEngine = new SearchEngineImpl(index);
+    }
   }
 }