Implement AnalysisServer.getContainingContext() using ContextManager.

pkg/analysis_server/test/analysis_server_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.
 
 library test.analysis_server;
 
 import 'dart:async';
 
 import 'package:analysis_server/plugin/protocol/protocol.dart';
 import 'package:analysis_server/src/analysis_server.dart';
@@ skipping 224 matching lines @@
       expect(fooContext.getKindOf(barSource), SourceKind.UNKNOWN);
       expect(barContext.getKindOf(fooSource), SourceKind.UNKNOWN);
       expect(barContext.getKindOf(barSource), SourceKind.LIBRARY);
     });
   }
 
   test_getContextSourcePair_nested() {
     String dir1Path = '/dir1';
     String dir2Path = dir1Path + '/dir2';
     String filePath = dir2Path + '/file.dart';
-    Folder dir1 = resourceProvider.newFolder(dir1Path);
-    Folder dir2 = resourceProvider.newFolder(dir2Path);
+    resourceProvider.newFile('$dir1Path/.packages', '');
+    resourceProvider.newFile('$dir2Path/.packages', '');
     resourceProvider.newFile(filePath, 'library lib;');
-
-    AnalysisContext context1 = AnalysisEngine.instance.createAnalysisContext();
-    AnalysisContext context2 = AnalysisEngine.instance.createAnalysisContext();
-    _configureSourceFactory(context1);
-    _configureSourceFactory(context2);
-    server.folderMap[dir1] = context1;
-    server.folderMap[dir2] = context2;
-
+    // create contexts
+    server.setAnalysisRoots('0', [dir1Path], [], {});
+    // get pair
     ContextSourcePair pair = server.getContextSourcePair(filePath);
     Source source = pair.source;
-    expect(pair.context, same(context2));
+    _assertContextOfFolder(pair.context, dir2Path);
     expect(source, isNotNull);
     expect(source.uri.scheme, 'file');
     expect(source.fullName, filePath);
   }
 
   test_getContextSourcePair_nonFile() {
     String dirPath = '/dir';
     Folder dir = resourceProvider.newFolder(dirPath);
 
     AnalysisContext context = AnalysisEngine.instance.createAnalysisContext();
     _configureSourceFactory(context);
     server.folderMap[dir] = context;
 
     ContextSourcePair pair = server.getContextSourcePair(dirPath);
     expect(pair, isNotNull);
     expect(pair.context, isNull);
     expect(pair.source, isNull);
   }
 
   test_getContextSourcePair_package_inRoot() {
     String rootPath = '/my_package';
     String filePath = rootPath + '/lib/file.dart';
     Folder rootFolder = resourceProvider.newFolder(rootPath);
     resourceProvider.newFile(filePath, 'library lib;');
 
     packageMapProvider.packageMap = <String, List<Folder>>{
       'my_package': <Folder>[rootFolder]
     };
-
-    AnalysisContext context = AnalysisEngine.instance.createAnalysisContext();
-    _configureSourceFactory(context);
-    server.folderMap[rootFolder] = context;
-
+    // create contexts
+    server.setAnalysisRoots('0', [rootPath], [], {});
+    // get pair
     ContextSourcePair pair = server.getContextSourcePair(filePath);
     Source source = pair.source;
-    expect(pair.context, same(context));
+    _assertContextOfFolder(pair.context, rootPath);
     expect(source, isNotNull);
     expect(source.uri.scheme, 'package');
     expect(source.fullName, filePath);
   }
 
   test_getContextSourcePair_simple() {
     String dirPath = '/dir';
     String filePath = dirPath + '/file.dart';
-    Folder dir = resourceProvider.newFolder(dirPath);
     resourceProvider.newFile(filePath, 'library lib;');
-
-    AnalysisContext context = AnalysisEngine.instance.createAnalysisContext();
-    _configureSourceFactory(context);
-    server.folderMap[dir] = context;
-
+    // create contexts
+    server.setAnalysisRoots('0', [dirPath], [], {});
+    // get pair
     ContextSourcePair pair = server.getContextSourcePair(filePath);
     Source source = pair.source;
-    expect(pair.context, same(context));
+    _assertContextOfFolder(pair.context, dirPath);
     expect(source, isNotNull);
     expect(source.uri.scheme, 'file');
     expect(source.fullName, filePath);
   }
 
   /**
    * Test that having multiple analysis contexts analyze the same file doesn't
    * cause that file to receive duplicate notifications when it's modified.
    */
   Future test_no_duplicate_notifications() async {
@@ skipping 176 matching lines @@
     // update the file
     channel.notificationsReceived.clear();
     resourceProvider.modifyFile(filePath, '// 333');
     await pumpEventQueue();
     // the file has an overlay, so the file-system change was ignored
     expect(channel.notificationsReceived.any((notification) {
       return notification.event == SERVER_STATUS;
     }), isFalse);
   }
 
+  void _assertContextOfFolder(
+      AnalysisContext context, String expectedFolderPath) {
+    Folder expectedFolder = resourceProvider.newFolder(expectedFolderPath);
+    ContextInfo expectedContextInfo = (server.contextManager
+        as ContextManagerImpl).getContextInfoFor(expectedFolder);
+    expect(expectedContextInfo, isNotNull);
+    expect(context, same(expectedContextInfo.context));
+  }
+
   void _configureSourceFactory(AnalysisContext context) {
     var resourceUriResolver = new ResourceUriResolver(resourceProvider);
     var packageUriResolver = new PackageMapUriResolver(
         resourceProvider, packageMapProvider.packageMap);
     context.sourceFactory =
         new SourceFactory([packageUriResolver, resourceUriResolver]);
   }
 }
 
 class EchoHandler implements RequestHandler {
@@ skipping 17 matching lines @@
   _MockServerOperation(this.context);
 
   @override
   ServerOperationPriority get priority => ServerOperationPriority.ANALYSIS;
 
   @override
   void perform(AnalysisServer server) {
     isComplete = true;
   }
 }