Cherry-pick r44373, r44378, and r44275.

pkg/analysis_server/test/operation/operation_queue_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.operation.queue;
 
 import 'package:analysis_server/src/analysis_server.dart';
 import 'package:analysis_server/src/operation/operation.dart';
 import 'package:analysis_server/src/operation/operation_analysis.dart';
 import 'package:analysis_server/src/operation/operation_queue.dart';
@@ skipping 43 matching lines @@
     var operationB = mockOperation(ServerOperationPriority.ANALYSIS_CONTINUE);
     queue.add(operationA);
     queue.add(operationB);
     // there are some operations
     expect(queue.isEmpty, false);
     // clear - no operations
     queue.clear();
     expect(queue.isEmpty, true);
   }
 
+  void test_contextRemoved() {
+    var contextA = new AnalysisContextMock();
+    var contextB = new AnalysisContextMock();
+    var opA1 = new _ServerOperationMock(contextA);
+    var opA2 = new _ServerOperationMock(contextA);
+    var opB1 = new _ServerOperationMock(contextB);
+    var opB2 = new _ServerOperationMock(contextB);
+    when(opA1.priority)
+        .thenReturn(ServerOperationPriority.ANALYSIS_NOTIFICATION);
+    when(opA2.priority)
+        .thenReturn(ServerOperationPriority.ANALYSIS_NOTIFICATION);
+    when(opB1.priority)
+        .thenReturn(ServerOperationPriority.ANALYSIS_NOTIFICATION);
+    when(opB2.priority)
+        .thenReturn(ServerOperationPriority.ANALYSIS_NOTIFICATION);
+    queue.add(opA1);
+    queue.add(opB1);
+    queue.add(opA2);
+    queue.add(opB2);
+    queue.contextRemoved(contextA);
+    expect(queue.take(), same(opB1));
+    expect(queue.take(), same(opB2));
+  }
+
   void test_isEmpty_false() {
     var operation = mockOperation(ServerOperationPriority.ANALYSIS);
     queue.add(operation);
     expect(queue.isEmpty, isFalse);
   }
 
   void test_isEmpty_true() {
     expect(queue.isEmpty, isTrue);
   }
 
@@ skipping 67 matching lines @@
     queue.add(operationB);
     queue.add(operationC);
     expect(queue.take(), operationC);
     expect(queue.take(), operationB);
     expect(queue.take(), operationA);
     expect(queue.take(), isNull);
   }
 }
 
 class _ServerOperationMock extends TypedMock implements ServerOperation {
+  final AnalysisContext context;
+
+  _ServerOperationMock([this.context]);
+
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 class _SourceMock extends TypedMock implements Source {
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 class _SourceSensitiveOperationMock extends TypedMock
     implements SourceSensitiveOperation {
   final Source source;
 
   _SourceSensitiveOperationMock(this.source);
 
   @override
   ServerOperationPriority get priority {
     return ServerOperationPriority.ANALYSIS_NOTIFICATION;
   }
 
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 
   @override
   bool shouldBeDiscardedOnSourceChange(Source source) {
     return source == this.source;
   }
 }