Prioritize analysis operations for contexts with priority files.

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';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:typed_mock/typed_mock.dart';
 import 'package:unittest/unittest.dart';
+import '../mocks.dart';
+import 'package:analyzer/src/generated/source.dart';
 
 main() {
   groupSep = ' | ';
 
   group('ServerOperationQueue', () {
     ServerOperationQueue queue;
 
     setUp(() {
       queue = new ServerOperationQueue();
     });
@@ skipping 37 matching lines @@
         queue.add(operationB);
         queue.add(operationC);
         expect(queue.take(), operationC);
         expect(queue.take(), operationB);
         expect(queue.take(), operationA);
         expect(queue.take(), isNull);
       });
 
       test('continue analysis first', () {
         var analysisContext = new AnalysisContextMock();
-        var operationA =
-            new PerformAnalysisOperation(analysisContext, false, false);
-        var operationB =
-            new PerformAnalysisOperation(analysisContext, false, true);
+        var operationA = new PerformAnalysisOperation(analysisContext, false);
+        var operationB = new PerformAnalysisOperation(analysisContext, true);
         queue.add(operationA);
         queue.add(operationB);
         expect(queue.take(), operationB);
         expect(queue.take(), operationA);
         expect(queue.take(), isNull);
       });
 
       test('priority context first', () {
+        var prioritySource = new MockSource();
         var analysisContextA = new AnalysisContextMock();
         var analysisContextB = new AnalysisContextMock();
-        var operationA =
-            new PerformAnalysisOperation(analysisContextA, false, false);
-        var operationB =
-            new PerformAnalysisOperation(analysisContextB, true, false);
+        analysisContextB.prioritySources = [prioritySource];
+        var operationA = new PerformAnalysisOperation(analysisContextA, false);
+        var operationB = new PerformAnalysisOperation(analysisContextB, false);
         queue.add(operationA);
         queue.add(operationB);
         expect(queue.take(), operationB);
         expect(queue.take(), operationA);
         expect(queue.take(), isNull);
+      });
+
+      test('reschedule', () {
+        var prioritySource = new MockSource();
+        var analysisContextA = new AnalysisContextMock();
+        var analysisContextB = new AnalysisContextMock();
+        var operationA = new PerformAnalysisOperation(analysisContextA, false);
+        var operationB = new PerformAnalysisOperation(analysisContextB, false);
+        queue.add(operationA);
+        queue.add(operationB);
+        // update priority sources and reschedule
+        analysisContextB.prioritySources = [prioritySource];
+        queue.reschedule();
+        // verify order
+        expect(queue.take(), operationB);
+        expect(queue.take(), operationA);
+        expect(queue.take(), isNull);
       });
     });
   });
 }
 
 
 /**
  *  Return a [ServerOperation] mock with the given priority.
  */
 ServerOperation mockOperation(ServerOperationPriority priority) {
   ServerOperation operation = new ServerOperationMock();
   when(operation.priority).thenReturn(priority);
   return operation;
 }
 
 
-class AnalysisContextMock extends TypedMock implements AnalysisContext {
+class AnalysisContextMock extends TypedMock implements InternalAnalysisContext {
+  List<Source> prioritySources = <Source>[];
+
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 class AnalysisServerMock extends TypedMock implements AnalysisServer {
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 class ServerContextManagerMock extends TypedMock implements ServerContextManager
     {
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 class ServerOperationMock extends TypedMock implements ServerOperation {
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }