Add tests for AnalysisServer.getAnalysisContextForSource.

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/src/analysis_server.dart';
 import 'package:analysis_server/src/constants.dart';
 import 'package:analysis_server/src/domain_server.dart';
 import 'package:analysis_server/src/operation/operation.dart';
 import 'package:analysis_server/src/protocol.dart';
+import 'package:analyzer/file_system/file_system.dart';
 import 'package:analyzer/file_system/memory_file_system.dart';
 import 'package:analyzer/instrumentation/instrumentation.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/java_engine.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:typed_mock/typed_mock.dart';
 import 'package:unittest/unittest.dart';
 
 import 'mock_sdk.dart';
 import 'mocks.dart';
 import 'reflective_tests.dart';
 
 main() {
   groupSep = ' | ';
   runReflectiveTests(AnalysisServerTest);
 }
 
 @ReflectiveTestCase()
 class AnalysisServerTest {
   MockServerChannel channel;
   AnalysisServer server;
   MemoryResourceProvider resourceProvider;
 
+  /**
+   * Verify that getAnalysisContextForSource returns the correct contexts even
+   * for sources that are included by multiple contexts.
+   *
+   * See dartbug.com/21898
+   */
+  Future fail_getAnalysisContextForSource_crossImports() {
+    // Subscribe to STATUS so we'll know when analysis is done.
+    server.serverServices = [ServerService.STATUS].toSet();
+    // Analyze project foo containing foo.dart and project bar containing
+    // bar.dart.
+    resourceProvider.newFolder('/foo');
+    resourceProvider.newFolder('/bar');
+    File foo = resourceProvider.newFile('/foo/foo.dart', '''
+libary foo;
+import "../bar/bar.dart";
+''');
+    Source fooSource = foo.createSource();
+    File bar = resourceProvider.newFile('/bar/bar.dart', '''
+library bar;
+import "../foo/foo.dart";
+''');
+    Source barSource = bar.createSource();
+    server.setAnalysisRoots('0', ['/foo', '/bar'], [], {});
+    return pumpEventQueue(40).then((_) {
+      expect(server.statusAnalyzing, isFalse);
+      // Make sure getAnalysisContext returns the proper context for each.
+      AnalysisContext fooContext =
+          server.getAnalysisContextForSource(fooSource);
+      expect(fooContext, isNotNull);
+      AnalysisContext barContext =
+          server.getAnalysisContextForSource(barSource);
+      expect(barContext, isNotNull);
+      expect(fooContext, isNot(same(barContext)));
+      expect(fooContext.getKindOf(fooSource), SourceKind.LIBRARY);
+      expect(fooContext.getKindOf(barSource), SourceKind.UNKNOWN);
+      expect(barContext.getKindOf(fooSource), SourceKind.UNKNOWN);
+      expect(barContext.getKindOf(barSource), SourceKind.LIBRARY);
+    });
+  }
+
+  /**
+   * Verify that getAnalysisContextForSource returns the correct contexts even
+   * for sources that haven't been analyzed yet.
+   *
+   * See dartbug.com/21898
+   */
+  Future fail_getAnalysisContextForSource_unanalyzed() {
+    // Subscribe to STATUS so we'll know when analysis is done.
+    server.serverServices = [ServerService.STATUS].toSet();
+    // Analyze project foo containing foo.dart and project bar containing
+    // bar.dart.
+    resourceProvider.newFolder('/foo');
+    resourceProvider.newFolder('/bar');
+    File foo = resourceProvider.newFile('/foo/foo.dart', 'library lib;');
+    Source fooSource = foo.createSource();
+    File bar = resourceProvider.newFile('/bar/bar.dart', 'library lib;');
+    Source barSource = bar.createSource();
+    server.setAnalysisRoots('0', ['/foo', '/bar'], [], {});
+    AnalysisContext fooContext = server.getAnalysisContextForSource(fooSource);
+    expect(fooContext, isNotNull);
+    AnalysisContext barContext = server.getAnalysisContextForSource(barSource);
+    expect(barContext, isNotNull);
+    expect(fooContext, isNot(same(barContext)));
+    return pumpEventQueue(40).then((_) {
+      expect(server.statusAnalyzing, isFalse);
+      // Make sure getAnalysisContext returned the proper context for each.
+      expect(fooContext.getKindOf(fooSource), SourceKind.LIBRARY);
+      expect(fooContext.getKindOf(barSource), SourceKind.UNKNOWN);
+      expect(barContext.getKindOf(fooSource), SourceKind.UNKNOWN);
+      expect(barContext.getKindOf(barSource), SourceKind.LIBRARY);
+    });
+  }
+
   void setUp() {
     channel = new MockServerChannel();
     resourceProvider = new MemoryResourceProvider();
     server = new AnalysisServer(
         channel,
         resourceProvider,
         new MockPackageMapProvider(),
         null,
         new AnalysisServerOptions(),
         new MockSdk(),
@@ skipping 54 matching lines @@
 
   Future test_echo() {
     server.handlers = [new EchoHandler()];
     var request = new Request('my22', 'echo');
     return channel.sendRequest(request).then((Response response) {
       expect(response.id, equals('my22'));
       expect(response.error, isNull);
     });
   }
 
+  Future test_getAnalysisContextForSource() {
+    // Subscribe to STATUS so we'll know when analysis is done.
+    server.serverServices = [ServerService.STATUS].toSet();
+    // Analyze project foo containing foo.dart and project bar containing
+    // bar.dart.
+    resourceProvider.newFolder('/foo');
+    resourceProvider.newFolder('/bar');
+    File foo = resourceProvider.newFile('/foo/foo.dart', 'library lib;');
+    Source fooSource = foo.createSource();
+    File bar = resourceProvider.newFile('/bar/bar.dart', 'library lib;');
+    Source barSource = bar.createSource();
+    server.setAnalysisRoots('0', ['/foo', '/bar'], [], {});
+    return pumpEventQueue(40).then((_) {
+      expect(server.statusAnalyzing, isFalse);
+      // Make sure getAnalysisContext returns the proper context for each.
+      AnalysisContext fooContext =
+          server.getAnalysisContextForSource(fooSource);
+      expect(fooContext, isNotNull);
+      AnalysisContext barContext =
+          server.getAnalysisContextForSource(barSource);
+      expect(barContext, isNotNull);
+      expect(fooContext, isNot(same(barContext)));
+      expect(fooContext.getKindOf(fooSource), SourceKind.LIBRARY);
+      expect(fooContext.getKindOf(barSource), SourceKind.UNKNOWN);
+      expect(barContext.getKindOf(fooSource), SourceKind.UNKNOWN);
+      expect(barContext.getKindOf(barSource), SourceKind.LIBRARY);
+    });
+  }
+
   Future test_prioritySourcesChangedEvent() {
     resourceProvider.newFolder('/foo');
 
     int eventCount = 0;
     Source firstSource = null;
     server.onPriorityChange.listen((PriorityChangeEvent event) {
       ++eventCount;
       firstSource = event.firstSource;
     });
 
@@ skipping 94 matching lines @@
   @override
   Response handleRequest(Request request) {
     if (request.method == 'echo') {
       return new Response(request.id, result: {
         'echo': true
       });
     }
     return null;
   }
 }