Manage 'libraries containing part' in DartWorkManager.

pkg/analyzer/test/src/task/dart_work_manager_test.dart

 // Copyright (c) 2015, 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.src.task.dart_work_manager_test;
 
 import 'package:analyzer/src/context/cache.dart';
 import 'package:analyzer/src/generated/ast.dart';
 import 'package:analyzer/src/generated/engine.dart'
     show
@@ skipping 114 matching lines @@
     entry1.setValue(SOURCE_KIND, SourceKind.LIBRARY, []);
     entry2.setValue(SOURCE_KIND, SourceKind.PART, []);
     entry3.setValue(SOURCE_KIND, SourceKind.LIBRARY, []);
     entry1.setValue(LIBRARY_ERRORS_READY, false, []);
     entry3.setValue(LIBRARY_ERRORS_READY, false, []);
     // change source2, schedule source1 and source3
     manager.applyChange([], [source2], []);
     expect_librarySourceQueue([source1, source3]);
   }
 
+  void test_applyChange_updatePartsLibraries_changeLibrary() {
+    Source part1 = new TestSource('part1.dart');
+    Source part2 = new TestSource('part2.dart');
+    Source part3 = new TestSource('part3.dart');
+    Source library1 = new TestSource('library1.dart');
+    Source library2 = new TestSource('library2.dart');
+    manager.partLibrariesMap[part1] = [library1, library2];
+    manager.partLibrariesMap[part2] = [library2];
+    manager.partLibrariesMap[part3] = [library1];
+    manager.libraryPartsMap[library1] = [part1, part3];
+    manager.libraryPartsMap[library2] = [part1, part2];
+    // change library1
+    manager.applyChange([], [library1], []);
+    expect(manager.partLibrariesMap[part1], unorderedEquals([library2]));
+    expect(manager.partLibrariesMap[part2], unorderedEquals([library2]));
+    expect(manager.partLibrariesMap[part3], unorderedEquals([]));
+    expect(manager.libraryPartsMap[library1], isNull);
+    expect(manager.libraryPartsMap[library2], [part1, part2]);
+  }
+
+  void test_applyChange_updatePartsLibraries_changePart() {
+    Source part1 = new TestSource('part1.dart');
+    Source part2 = new TestSource('part2.dart');
+    Source part3 = new TestSource('part3.dart');
+    Source library1 = new TestSource('library1.dart');
+    Source library2 = new TestSource('library2.dart');
+    manager.partLibrariesMap[part1] = [library1, library2];
+    manager.partLibrariesMap[part2] = [library2];
+    manager.partLibrariesMap[part3] = [library1];
+    manager.libraryPartsMap[library1] = [part1, part3];
+    manager.libraryPartsMap[library2] = [part1, part2];
+    // change part1
+    manager.applyChange([], [part1], []);
+    expect(manager.partLibrariesMap[part1], isNull);
+    expect(manager.partLibrariesMap[part2], unorderedEquals([library2]));
+    expect(manager.partLibrariesMap[part3], unorderedEquals([library1]));
+    expect(manager.libraryPartsMap[library1], [part3]);
+    expect(manager.libraryPartsMap[library2], [part2]);
+  }
+
+  void test_applyChange_updatePartsLibraries_removeLibrary() {
+    Source part1 = new TestSource('part1.dart');
+    Source part2 = new TestSource('part2.dart');
+    Source part3 = new TestSource('part3.dart');
+    Source library1 = new TestSource('library1.dart');
+    Source library2 = new TestSource('library2.dart');
+    manager.partLibrariesMap[part1] = [library1, library2];
+    manager.partLibrariesMap[part2] = [library2];
+    manager.partLibrariesMap[part3] = [library1];
+    manager.libraryPartsMap[library1] = [part1, part3];
+    manager.libraryPartsMap[library2] = [part1, part2];
+    // remove library1
+    manager.applyChange([], [], [library1]);
+    expect(manager.partLibrariesMap[part1], unorderedEquals([library2]));
+    expect(manager.partLibrariesMap[part2], unorderedEquals([library2]));
+    expect(manager.partLibrariesMap[part3], unorderedEquals([]));
+    expect(manager.libraryPartsMap[library1], isNull);
+    expect(manager.libraryPartsMap[library2], [part1, part2]);
+  }
+
+  void test_applyChange_updatePartsLibraries_removePart() {
+    Source part1 = new TestSource('part1.dart');
+    Source part2 = new TestSource('part2.dart');
+    Source part3 = new TestSource('part3.dart');
+    Source library1 = new TestSource('library1.dart');
+    Source library2 = new TestSource('library2.dart');
+    manager.partLibrariesMap[part1] = [library1, library2];
+    manager.partLibrariesMap[part2] = [library2];
+    manager.partLibrariesMap[part3] = [library1];
+    manager.libraryPartsMap[library1] = [part1, part3];
+    manager.libraryPartsMap[library2] = [part1, part2];
+    // remove part1
+    manager.applyChange([], [], [part1]);
+    expect(manager.partLibrariesMap[part1], isNull);
+    expect(manager.partLibrariesMap[part2], unorderedEquals([library2]));
+    expect(manager.partLibrariesMap[part3], unorderedEquals([library1]));
+    expect(manager.libraryPartsMap[library1], [part3]);
+    expect(manager.libraryPartsMap[library2], [part2]);
+  }
+
   void test_applyPriorityTargets_library() {
     entry1.setValue(SOURCE_KIND, SourceKind.LIBRARY, []);
     entry2.setValue(SOURCE_KIND, SourceKind.LIBRARY, []);
     entry3.setValue(SOURCE_KIND, SourceKind.LIBRARY, []);
     manager.priorityResultQueue
         .add(new TargetedResult(source1, LIBRARY_ERRORS_READY));
     manager.priorityResultQueue
         .add(new TargetedResult(source2, LIBRARY_ERRORS_READY));
     // -source1 +source3
     manager.applyPriorityTargets([source2, source3]);
@@ skipping 34 matching lines @@
     LineInfo lineInfo = new LineInfo([0]);
     entry1.setValue(LINE_INFO, lineInfo, []);
     entry1.setValue(SCAN_ERRORS, <AnalysisError>[error1], []);
     context.getCacheEntry(new LibrarySpecificUnit(source2, source1)).setValue(
         VERIFY_ERRORS, <AnalysisError>[error2], []);
     AnalysisErrorInfo errorInfo = manager.getErrors(source1);
     expect(errorInfo.errors, unorderedEquals([error1, error2]));
     expect(errorInfo.lineInfo, lineInfo);
   }
 
+  void test_getLibrariesContainingPart() {
+    Source part1 = new TestSource('part1.dart');
+    Source part2 = new TestSource('part2.dart');
+    Source part3 = new TestSource('part3.dart');
+    Source library1 = new TestSource('library1.dart');
+    Source library2 = new TestSource('library2.dart');
+    manager.partLibrariesMap[part1] = [library1, library2];
+    manager.partLibrariesMap[part2] = [library2];
+    manager.libraryPartsMap[library1] = [part1];
+    manager.libraryPartsMap[library2] = [part1, part2];
+    // getLibrariesContainingPart
+    expect(manager.getLibrariesContainingPart(part1),
+        unorderedEquals([library1, library2]));
+    expect(
+        manager.getLibrariesContainingPart(part2), unorderedEquals([library2]));
+    expect(manager.getLibrariesContainingPart(part3), isEmpty);
+  }
+
   void test_getNextResult_hasLibraries_firstIsError() {
     entry1.setErrorState(caughtException, [LIBRARY_ERRORS_READY]);
     manager.librarySourceQueue.addAll([source1, source2]);
     TargetedResult request = manager.getNextResult();
     expect(request.target, source2);
     expect(request.result, LIBRARY_ERRORS_READY);
     // source1 is out, source2 is waiting
     expect_librarySourceQueue([source2]);
   }
 
@@ skipping 140 matching lines @@
     context.getCacheEntry(unitTarget).setValue(
         VERIFY_ERRORS, <AnalysisError>[error2], []);
     // notify about Source specific errors
     manager.resultsComputed(source1, {SCAN_ERRORS: []});
     // all of the errors are included
     ChangeNoticeImpl notice = context.getNotice(source1);
     expect(notice.errors, unorderedEquals([error1, error2]));
     expect(notice.lineInfo, lineInfo);
   }
 
+  void test_resultsComputed_includedParts() {
+    Source part1 = new TestSource('part1.dart');
+    Source part2 = new TestSource('part2.dart');
+    Source part3 = new TestSource('part3.dart');
+    Source library1 = new TestSource('library1.dart');
+    Source library2 = new TestSource('library2.dart');
+    // library1 parts
+    manager.resultsComputed(library1, {INCLUDED_PARTS: [part1, part2]});
+    expect(manager.partLibrariesMap[part1], [library1]);
+    expect(manager.partLibrariesMap[part2], [library1]);
+    expect(manager.partLibrariesMap[part3], isNull);
+    expect(manager.libraryPartsMap[library1], [part1, part2]);
+    expect(manager.libraryPartsMap[library2], isNull);
+    // library2 parts
+    manager.resultsComputed(library2, {INCLUDED_PARTS: [part2, part3]});
+    expect(manager.partLibrariesMap[part1], [library1]);
+    expect(manager.partLibrariesMap[part2], [library1, library2]);
+    expect(manager.partLibrariesMap[part3], [library2]);
+    expect(manager.libraryPartsMap[library1], [part1, part2]);
+    expect(manager.libraryPartsMap[library2], [part2, part3]);
+  }
+
   void test_resultsComputed_noSourceKind() {
     manager.unknownSourceQueue.addAll([source1, source2]);
     manager.resultsComputed(source1, {});
     expect_librarySourceQueue([]);
     expect_unknownSourceQueue([source1, source2]);
   }
 
   void test_resultsComputed_notDart() {
     manager.unknownSourceQueue.addAll([source1, source2]);
     manager.resultsComputed(new TestSource('test.html'), {});
@@ skipping 63 matching lines @@
   }
 
   @override
   ChangeNoticeImpl getNotice(Source source) {
     return _pendingNotices.putIfAbsent(
         source, () => new ChangeNoticeImpl(source));
   }
 
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }