Register files with new drivers (issue 29641)

pkg/analyzer_plugin/test/plugin/plugin_test.dart

 // Copyright (c) 2017, 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:analyzer/file_system/file_system.dart';
 import 'package:analyzer/file_system/memory_file_system.dart';
 import 'package:analyzer/src/dart/analysis/driver.dart';
 import 'package:analyzer_plugin/plugin/plugin.dart';
 import 'package:analyzer_plugin/protocol/protocol_common.dart';
 import 'package:analyzer_plugin/protocol/protocol_generated.dart';
 import 'package:path/src/context.dart';
 import 'package:test/test.dart';
 import 'package:test_reflective_loader/test_reflective_loader.dart';
 
 void main() {
   defineReflectiveTests(ServerPluginTest);
 }
 
 class MockAnalysisDriver extends AnalysisDriverGeneric {
+  /**
+   * The files that have been added to this driver.
+   */
+  List<String> addedFiles = <String>[];
+
   @override
   bool get hasFilesToAnalyze => false;
 
   @override
   set priorityFiles(List<String> priorityPaths) {}
 
   @override
   AnalysisDriverPriority get workPriority => AnalysisDriverPriority.nothing;
 
   @override
+  void addFile(String path) {
+    addedFiles.add(path);
+  }
+
+  @override
   void dispose() {}
 
   @override
   Future<Null> performWork() => new Future.value(null);
 }
 
 @reflectiveTest
 class ServerPluginTest {
   MemoryResourceProvider resourceProvider = new MemoryResourceProvider();
 
@@ skipping 71 matching lines @@
     var result = await plugin.handleAnalysisSetContextBuilderOptions(
         new AnalysisSetContextBuilderOptionsParams(
             new ContextBuilderOptions()));
     expect(result, isNotNull);
   }
 
   test_handleAnalysisSetContextRoots() async {
     var result = await plugin.handleAnalysisSetContextRoots(
         new AnalysisSetContextRootsParams([contextRoot1]));
     expect(result, isNotNull);
-    expect(plugin.driverMap[contextRoot1], isNotNull);
+    AnalysisDriverGeneric driver = plugin.driverMap[contextRoot1];
+    expect(driver, isNotNull);
+    expect((driver as MockAnalysisDriver).addedFiles, hasLength(1));
   }
 
   test_handleAnalysisSetPriorityFiles() async {
     await plugin.handleAnalysisSetContextRoots(
         new AnalysisSetContextRootsParams([contextRoot1]));
 
     var result = await plugin.handleAnalysisSetPriorityFiles(
         new AnalysisSetPriorityFilesParams([filePath1]));
     expect(result, isNotNull);
   }
@@ skipping 137 matching lines @@
   AnalysisDriverGeneric createAnalysisDriver(ContextRoot contextRoot) {
     return new MockAnalysisDriver();
   }
 
   @override
   void sendNotificationsForSubscriptions(
       Map<String, List<AnalysisService>> subscriptions) {
     latestSubscriptions = subscriptions;
   }
 }