Decide if unit without directives is a part or a library.

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 19 matching lines @@
 import '../../reflective_tests.dart';
 
 main() {
   groupSep = ' | ';
   runReflectiveTests(DartWorkManagerTest);
 }
 
 @reflectiveTest
 class DartWorkManagerTest {
   InternalAnalysisContext context = new _InternalAnalysisContextMock();
+  AnalysisCache cache;
   DartWorkManager manager;
 
   CaughtException caughtException = new CaughtException(null, null);
 
   Source source1 = new TestSource('1.dart');
   Source source2 = new TestSource('2.dart');
   Source source3 = new TestSource('3.dart');
   Source source4 = new TestSource('4.dart');
   CacheEntry entry1;
   CacheEntry entry2;
   CacheEntry entry3;
   CacheEntry entry4;
 
   void expect_librarySourceQueue(List<Source> sources) {
     expect(manager.librarySourceQueue, unorderedEquals(sources));
   }
 
   void expect_unknownSourceQueue(List<Source> sources) {
     expect(manager.unknownSourceQueue, unorderedEquals(sources));
   }
 
   void setUp() {
+    cache = context.analysisCache;
     manager = new DartWorkManager(context);
     entry1 = context.getCacheEntry(source1);
     entry2 = context.getCacheEntry(source2);
     entry3 = context.getCacheEntry(source3);
     entry4 = context.getCacheEntry(source4);
   }
 
   void test_applyChange_add() {
     // add source1
     manager.applyChange([source1], [], []);
@@ skipping 364 matching lines @@
     entry1.setValue(PARSE_ERRORS, <AnalysisError>[error2], []);
     // PARSED_UNIT is ready, set errors
     manager.resultsComputed(
         source1, {PARSED_UNIT: AstFactory.compilationUnit()});
     // 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() {
+  void test_resultsComputed_includedParts_turnLibraryIntoPart() {
+    when(context.shouldErrorsBeAnalyzed(anyObject, null)).thenReturn(true);
+    Source library = new TestSource('library.dart');
+    Source part = new TestSource('part.dart');
+    // library "library" (real)
+    manager.resultsComputed(library, {SOURCE_KIND: SourceKind.LIBRARY});
+    CacheEntry libraryEntry = new CacheEntry(library);
+    cache.put(libraryEntry);
+    libraryEntry.setValue(SOURCE_KIND, SourceKind.LIBRARY, []);
+    // library "part" (speculative)
+    manager.resultsComputed(part, {SOURCE_KIND: SourceKind.LIBRARY});
+    CacheEntry partEntry = new CacheEntry(part);
+    cache.put(partEntry);
+    partEntry.setValue(SOURCE_KIND, SourceKind.LIBRARY, []);
+    // unit "part in part" (speculative)
+    AnalysisTarget partInPartTarget = new LibrarySpecificUnit(part, part);
+    CacheEntry partInPartEntry = new CacheEntry(partInPartTarget);
+    cache.put(partInPartEntry);
+    // so far we consider "part" a library
+    expect(manager.librarySourceQueue, contains(part));
+    expect(partEntry.getValue(SOURCE_KIND), SourceKind.LIBRARY);
+    expect(cache.get(partInPartTarget), isNotNull);
+    // "part" is used as a part in "library"
+    manager.resultsComputed(library, {INCLUDED_PARTS: [part]});
+    expect(manager.partLibrariesMap[part], [library]);
+    expect(manager.libraryPartsMap[library], [part]);
+    // now we know that "part" is not a library
+    expect(manager.librarySourceQueue, isNot(contains(part)));
+    expect(partEntry.getValue(SOURCE_KIND), SourceKind.PART);
+    expect(cache.get(partInPartTarget), isNull);
+  }
+
+  void test_resultsComputed_includedParts_updatePartLibraries() {
     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);
@@ skipping 85 matching lines @@
   }
 
   @override
   ChangeNoticeImpl getNotice(Source source) {
     return _pendingNotices.putIfAbsent(
         source, () => new ChangeNoticeImpl(source));
   }
 
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }