Add support for highlights notification

pkg/analyzer_plugin/test/plugin/highlights_mixin_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/highlights_mixin.dart';
+import 'package:analyzer_plugin/protocol/protocol.dart';
+import 'package:analyzer_plugin/protocol/protocol_common.dart';
+import 'package:analyzer_plugin/protocol/protocol_generated.dart';
+import 'package:analyzer_plugin/src/utilities/highlights/highlights.dart';
+import 'package:analyzer_plugin/utilities/highlights/highlights.dart';
+import 'package:path/src/context.dart';
+import 'package:test/test.dart';
+import 'package:test_reflective_loader/test_reflective_loader.dart';
+
+import 'mocks.dart';
+
+void main() {
+  defineReflectiveTests(HighlightsMixinTest);
+}
+
+@reflectiveTest
+class HighlightsMixinTest {
+  MemoryResourceProvider resourceProvider = new MemoryResourceProvider();
+
+  String packagePath1;
+  String filePath1;
+  ContextRoot contextRoot1;
+
+  MockChannel channel;
+  _TestServerPlugin plugin;
+
+  void setUp() {
+    Context pathContext = resourceProvider.pathContext;
+
+    packagePath1 = resourceProvider.convertPath('/package1');
+    filePath1 = pathContext.join(packagePath1, 'lib', 'test.dart');
+    resourceProvider.newFile(filePath1, '');
+    contextRoot1 = new ContextRoot(packagePath1, <String>[]);
+
+    channel = new MockChannel();
+    plugin = new _TestServerPlugin(resourceProvider);
+    plugin.start(channel);
+  }
+
+  test_sendHighlightsNotification() async {
+    await plugin.handleAnalysisSetContextRoots(
+        new AnalysisSetContextRootsParams([contextRoot1]));
+
+    Completer<Null> notificationReceived = new Completer<Null>();
+    channel.listen(null, onNotification: (Notification notification) {
+      expect(notification, isNotNull);
+      AnalysisHighlightsParams params =
+          new AnalysisHighlightsParams.fromNotification(notification);
+      expect(params.file, filePath1);
+      expect(params.regions, hasLength(5));
+      notificationReceived.complete();
+    });
+    await plugin.sendHighlightsNotification(filePath1);
+    await notificationReceived.future;
+  }
+}
+
+class _TestHighlightsContributor implements HighlightsContributor {
+  int elementCount;
+
+  _TestHighlightsContributor(this.elementCount);
+
+  @override
+  void computeHighlights(
+      HighlightsRequest request, HighlightsCollector collector) {
+    for (int i = 0; i < elementCount; i++) {
+      collector.addRegion(i, 20, HighlightRegionType.METHOD_DECLARATION);
+    }
+  }
+}
+
+class _TestServerPlugin extends MockServerPlugin with HighlightsMixin {
+  _TestServerPlugin(ResourceProvider resourceProvider)
+      : super(resourceProvider);
+
+  @override
+  List<HighlightsContributor> getHighlightsContributors(String path) {
+    return <HighlightsContributor>[
+      new _TestHighlightsContributor(2),
+      new _TestHighlightsContributor(3)
+    ];
+  }
+
+  @override
+  Future<HighlightsRequest> getHighlightsRequest(String path) async {
+    AnalysisResult result = new AnalysisResult(
+        null, null, path, null, null, null, null, null, null, null, null);
+    return new DartHighlightsRequestImpl(resourceProvider, result);
+  }
+}