Index Dart/HTML units after analysis.

pkg/analysis_server/test/index/store/split_store_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.index.split_store;
 
 import 'dart:async';
-import 'dart:collection';
 
 import 'package:analysis_server/src/index/store/codec.dart';
 import 'package:analysis_server/src/index/store/split_store.dart';
 import 'package:analyzer/src/generated/element.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/index.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:typed_mock/typed_mock.dart';
 import 'package:unittest/unittest.dart';
 
 import '../../reflective_tests.dart';
+import 'memory_node_manager.dart';
 import 'single_source_container.dart';
 import 'typed_mocks.dart';
 
 
 main() {
   groupSep = ' | ';
   group('FileNodeManager', () {
     runReflectiveTests(_FileNodeManagerTest);
   });
   group('IndexNode', () {
@@ skipping 365 matching lines @@
   bool operator ==(Object other) {
     if (other is _LocationEqualsWrapper) {
       return other.location.offset == location.offset && other.location.length
           == location.length && other.location.element == location.element;
     }
     return false;
   }
 }
 
 
-class _MemoryNodeManager implements NodeManager {
-  ContextCodec _contextCodec = new ContextCodec();
-  ElementCodec _elementCodec;
-  int _locationCount = 0;
-  final Map<String, int> _nodeLocationCounts = new HashMap<String, int>();
-  final Map<String, IndexNode> _nodes = new HashMap<String, IndexNode>();
-  RelationshipCodec _relationshipCodec;
-  StringCodec _stringCodec = new StringCodec();
-
-  _MemoryNodeManager() {
-    _elementCodec = new ElementCodec(_stringCodec);
-    _relationshipCodec = new RelationshipCodec(_stringCodec);
-  }
-
-  @override
-  ContextCodec get contextCodec {
-    return _contextCodec;
-  }
-
-  @override
-  ElementCodec get elementCodec {
-    return _elementCodec;
-  }
-
-  @override
-  int get locationCount {
-    return _locationCount;
-  }
-
-  @override
-  StringCodec get stringCodec {
-    return _stringCodec;
-  }
-
-  @override
-  void clear() {
-    _nodes.clear();
-  }
-
-  int getLocationCount(String name) {
-    int locationCount = _nodeLocationCounts[name];
-    return locationCount != null ? locationCount : 0;
-  }
-
-  @override
-  Future<IndexNode> getNode(String name) {
-    return new Future.value(_nodes[name]);
-  }
-
-  bool isEmpty() {
-    for (IndexNode node in _nodes.values) {
-      Map<RelationKeyData, List<LocationData>> relations = node.relations;
-      if (!relations.isEmpty) {
-        return false;
-      }
-    }
-    return true;
-  }
-
-  @override
-  IndexNode newNode(AnalysisContext context) {
-    return new IndexNode(context, elementCodec, _relationshipCodec);
-  }
-
-  @override
-  void putNode(String name, IndexNode node) {
-    // update location count
-    {
-      _locationCount -= getLocationCount(name);
-      int nodeLocationCount = node.locationCount;
-      _nodeLocationCounts[name] = nodeLocationCount;
-      _locationCount += nodeLocationCount;
-    }
-    // remember the node
-    _nodes[name] = node;
-  }
-
-  @override
-  void removeNode(String name) {
-    _nodes.remove(name);
-  }
-}
-
-
 class _MockFileManager extends TypedMock implements FileManager {
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 
 class _MockIndexNode extends TypedMock implements IndexNode {
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 
@@ skipping 64 matching lines @@
       ['/home/user/sourceB.dart', 'ClassB']);
   ElementLocation elementLocationC = new ElementLocationImpl.con3(
       ['/home/user/sourceC.dart', 'ClassC']);
   ElementLocation elementLocationD = new ElementLocationImpl.con3(
       ['/home/user/sourceD.dart', 'ClassD']);
   HtmlElement htmlElementA = new MockHtmlElement();
   HtmlElement htmlElementB = new MockHtmlElement();
   LibraryElement libraryElement = new MockLibraryElement();
   Source librarySource = new MockSource('librarySource');
   CompilationUnitElement libraryUnitElement = new MockCompilationUnitElement();
-  _MemoryNodeManager nodeManager = new _MemoryNodeManager();
+  MemoryNodeManager nodeManager = new MemoryNodeManager();
   Relationship relationship = Relationship.getRelationship('test-relationship');
   Source sourceA = new MockSource('sourceA');
   Source sourceB = new MockSource('sourceB');
   Source sourceC = new MockSource('sourceC');
   Source sourceD = new MockSource('sourceD');
   SplitIndexStore store;
   CompilationUnitElement unitElementA = new MockCompilationUnitElement();
   CompilationUnitElement unitElementB = new MockCompilationUnitElement();
   CompilationUnitElement unitElementC = new MockCompilationUnitElement();
   CompilationUnitElement unitElementD = new MockCompilationUnitElement();
@@ skipping 435 matching lines @@
       // re-index "unitElementA"
       store.aboutToIndexDart(contextA, unitElementA);
       store.doneIndex();
       return store.getRelationshipsAsync(UniverseElement.INSTANCE,
           relationship).then((List<Location> locations) {
         assertLocations(locations, [locationB]);
       });
     });
   }
 
-  test_universe_removeContext() {
+  test_universe_clear() {
     when(contextA.getElement(elementLocationA)).thenReturn(elementA);
     when(contextB.getElement(elementLocationB)).thenReturn(elementB);
     Location locationA = mockLocation(elementA);
     Location locationB = mockLocation(elementB);
     {
       store.aboutToIndexDart(contextA, unitElementA);
       store.recordRelationship(UniverseElement.INSTANCE, relationship,
           locationA);
       store.doneIndex();
     }
     {
-      store.aboutToIndexDart(contextB, unitElementB);
+      store.aboutToIndexDart(contextA, unitElementB);
       store.recordRelationship(UniverseElement.INSTANCE, relationship,
           locationB);
       store.doneIndex();
     }
     return store.getRelationshipsAsync(UniverseElement.INSTANCE,
         relationship).then((List<Location> locations) {
       assertLocations(locations, [locationA, locationB]);
     }).then((_) {
-      // remove "contextA"
-      store.removeContext(contextA);
+      // clear
+      store.clear();
       return store.getRelationshipsAsync(UniverseElement.INSTANCE,
           relationship).then((List<Location> locations) {
-        assertLocations(locations, [locationB]);
+        expect(locations, isEmpty);
       });
     });
   }
 
-  test_universe_removeSource() {
+  test_universe_removeContext() {
     when(contextA.getElement(elementLocationA)).thenReturn(elementA);
     when(contextB.getElement(elementLocationB)).thenReturn(elementB);
     Location locationA = mockLocation(elementA);
     Location locationB = mockLocation(elementB);
     {
       store.aboutToIndexDart(contextA, unitElementA);
       store.recordRelationship(UniverseElement.INSTANCE, relationship,
           locationA);
       store.doneIndex();
     }
     {
-      store.aboutToIndexDart(contextA, unitElementB);
+      store.aboutToIndexDart(contextB, unitElementB);
       store.recordRelationship(UniverseElement.INSTANCE, relationship,
           locationB);
       store.doneIndex();
     }
     return store.getRelationshipsAsync(UniverseElement.INSTANCE,
         relationship).then((List<Location> locations) {
       assertLocations(locations, [locationA, locationB]);
     }).then((_) {
-      // remove "sourceA"
-      store.removeSource(contextA, sourceA);
+      // remove "contextA"
+      store.removeContext(contextA);
       return store.getRelationshipsAsync(UniverseElement.INSTANCE,
           relationship).then((List<Location> locations) {
         assertLocations(locations, [locationB]);
       });
     });
   }
 
-  test_universe_clear() {
+  test_universe_removeSource() {
     when(contextA.getElement(elementLocationA)).thenReturn(elementA);
     when(contextB.getElement(elementLocationB)).thenReturn(elementB);
     Location locationA = mockLocation(elementA);
     Location locationB = mockLocation(elementB);
     {
       store.aboutToIndexDart(contextA, unitElementA);
       store.recordRelationship(UniverseElement.INSTANCE, relationship,
           locationA);
       store.doneIndex();
     }
     {
       store.aboutToIndexDart(contextA, unitElementB);
       store.recordRelationship(UniverseElement.INSTANCE, relationship,
           locationB);
       store.doneIndex();
     }
     return store.getRelationshipsAsync(UniverseElement.INSTANCE,
         relationship).then((List<Location> locations) {
       assertLocations(locations, [locationA, locationB]);
     }).then((_) {
-      // clear
-      store.clear();
+      // remove "sourceA"
+      store.removeSource(contextA, sourceA);
       return store.getRelationshipsAsync(UniverseElement.INSTANCE,
           relationship).then((List<Location> locations) {
-        expect(locations, isEmpty);
+        assertLocations(locations, [locationB]);
       });
     });
   }
 
   /**
    * Asserts that the [actual] locations have all the [expected] locations and
    * only them.
    */
   static void assertLocations(List<Location> actual, List<Location> expected) {
     List<_LocationEqualsWrapper> actualWrappers = wrapLocations(actual);
@@ skipping 16 matching lines @@
    * Wraps the given locations into [LocationEqualsWrapper].
    */
   static List<_LocationEqualsWrapper> wrapLocations(List<Location> locations) {
     List<_LocationEqualsWrapper> wrappers = <_LocationEqualsWrapper>[];
     for (Location location in locations) {
       wrappers.add(new _LocationEqualsWrapper(location));
     }
     return wrappers;
   }
 }