Move split_store.dart to the store/ folder and extract collections/codecs.

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/split_store.dart';
+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 '../../reflective_tests.dart';
+import 'typed_mocks.dart';
 
 
 main() {
   groupSep = ' | ';
-  group('ContextCodec', () {
-    runReflectiveTests(_ContextCodecTest);
-  });
-  group('ElementCodec', () {
-    runReflectiveTests(_ElementCodecTest);
-  });
   group('FileNodeManager', () {
     runReflectiveTests(_FileNodeManagerTest);
   });
   group('IndexNode', () {
     runReflectiveTests(_IndexNodeTest);
   });
-  group('IntToIntSetMap', () {
-    runReflectiveTests(_IntToIntSetMapTest);
-  });
   group('LocationData', () {
     runReflectiveTests(_LocationDataTest);
   });
   group('RelationKeyData', () {
     runReflectiveTests(_RelationKeyDataTest);
   });
-  group('RelationshipCodec', () {
-    runReflectiveTests(_RelationshipCodecTest);
-  });
   group('SplitIndexStore', () {
     runReflectiveTests(_SplitIndexStoreTest);
   });
-  group('StringCodec', () {
-    runReflectiveTests(_StringCodecTest);
-  });
 }
 
 
 void _assertHasLocation(List<Location> locations, Element element, int offset,
     int length) {
   for (Location location in locations) {
     if ((element == null || location.element == element) && location.offset ==
         offset && location.length == length) {
       return;
     }
   }
   fail('Expected to find Location'
       '(element=$element, offset=$offset, length=$length)');
 }
 
 
 @ReflectiveTestCase()
-class _ContextCodecTest {
-  ContextCodec codec = new ContextCodec();
-
-  void test_encode_decode() {
-    AnalysisContext contextA = new _MockAnalysisContext('contextA');
-    AnalysisContext contextB = new _MockAnalysisContext('contextB');
-    int idA = codec.encode(contextA);
-    int idB = codec.encode(contextB);
-    expect(idA, codec.encode(contextA));
-    expect(idB, codec.encode(contextB));
-    expect(codec.decode(idA), contextA);
-    expect(codec.decode(idB), contextB);
-  }
-
-  void test_remove() {
-    // encode
-    {
-      AnalysisContext context = new _MockAnalysisContext('context');
-      // encode
-      int id = codec.encode(context);
-      expect(id, 0);
-      expect(codec.decode(id), context);
-      // remove
-      codec.remove(context);
-      expect(codec.decode(id), isNull);
-    }
-    // encode again
-    {
-      AnalysisContext context = new _MockAnalysisContext('context');
-      // encode
-      int id = codec.encode(context);
-      expect(id, 1);
-      expect(codec.decode(id), context);
-    }
-  }
-}
-
-
-@ReflectiveTestCase()
-class _ElementCodecTest {
-  ElementCodec codec;
-  AnalysisContext context = new _MockAnalysisContext('context');
-  StringCodec stringCodec = new StringCodec();
-
-  void setUp() {
-    codec = new ElementCodec(stringCodec);
-  }
-
-  void test_localLocalVariable() {
-    {
-      Element element = new _MockElement();
-      ElementLocation location = new ElementLocationImpl.con3(['main', 'foo@1',
-          'bar@2']);
-      when(context.getElement(location)).thenReturn(element);
-      when(element.location).thenReturn(location);
-      int id = codec.encode(element);
-      expect(codec.decode(context, id), element);
-    }
-    {
-      Element element = new _MockElement();
-      ElementLocation location = new ElementLocationImpl.con3(['main', 'foo@10',
-          'bar@20']);
-      when(context.getElement(location)).thenReturn(element);
-      when(element.location).thenReturn(location);
-      int id = codec.encode(element);
-      expect(codec.decode(context, id), element);
-    }
-    // check strings, "foo" as a single string, no "foo@1" or "foo@10"
-    expect(stringCodec.nameToIndex, hasLength(3));
-    expect(stringCodec.nameToIndex, containsPair('main', 0));
-    expect(stringCodec.nameToIndex, containsPair('foo', 1));
-    expect(stringCodec.nameToIndex, containsPair('bar', 2));
-  }
-
-  void test_localVariable() {
-    {
-      Element element = new _MockElement();
-      ElementLocation location = new ElementLocationImpl.con3(['main',
-          'foo@42']);
-      when(context.getElement(location)).thenReturn(element);
-      when(element.location).thenReturn(location);
-      int id = codec.encode(element);
-      expect(codec.decode(context, id), element);
-    }
-    {
-      Element element = new _MockElement();
-      ElementLocation location = new ElementLocationImpl.con3(['main',
-          'foo@4200']);
-      when(context.getElement(location)).thenReturn(element);
-      when(element.location).thenReturn(location);
-      int id = codec.encode(element);
-      expect(codec.decode(context, id), element);
-    }
-    // check strings, "foo" as a single string, no "foo@42" or "foo@4200"
-    expect(stringCodec.nameToIndex, hasLength(2));
-    expect(stringCodec.nameToIndex, containsPair('main', 0));
-    expect(stringCodec.nameToIndex, containsPair('foo', 1));
-  }
-
-  void test_notLocal() {
-    Element element = new _MockElement();
-    ElementLocation location = new ElementLocationImpl.con3(['foo', 'bar']);
-    when(element.location).thenReturn(location);
-    when(context.getElement(location)).thenReturn(element);
-    int id = codec.encode(element);
-    expect(codec.encode(element), id);
-    expect(codec.decode(context, id), element);
-    // check strings
-    expect(stringCodec.nameToIndex, hasLength(2));
-    expect(stringCodec.nameToIndex, containsPair('foo', 0));
-    expect(stringCodec.nameToIndex, containsPair('bar', 1));
-  }
-}
-
-
-@ReflectiveTestCase()
 class _FileNodeManagerTest {
-  AnalysisContext context = new _MockAnalysisContext('context');
-  ContextCodec contextCodec = new _MockContextCodec();
+  AnalysisContext context = new MockAnalysisContext('context');
+  ContextCodec contextCodec = new MockContextCodec();
   int contextId = 13;
-  ElementCodec elementCodec = new _MockElementCodec();
+  ElementCodec elementCodec = new MockElementCodec();
   FileManager fileManager = new _MockFileManager();
-  _MockLogger logger = new _MockLogger();
+  MockLogger logger = new MockLogger();
   int nextElementId = 0;
   FileNodeManager nodeManager;
   RelationshipCodec relationshipCodec;
   StringCodec stringCodec = new StringCodec();
 
   void setUp() {
     relationshipCodec = new RelationshipCodec(stringCodec);
     nodeManager = new FileNodeManager(fileManager, logger, stringCodec,
         contextCodec, elementCodec, relationshipCodec);
     when(contextCodec.encode(context)).thenReturn(contextId);
@@ skipping 155 matching lines @@
   }
 
   void test_removeNode() {
     String name = '42.index';
     nodeManager.removeNode(name);
     verify(fileManager.delete(name)).once();
   }
 
   Element _mockElement() {
     int elementId = nextElementId++;
-    Element element = new _MockElement();
+    Element element = new MockElement();
     when(elementCodec.encode(element)).thenReturn(elementId);
     when(elementCodec.decode(context, elementId)).thenReturn(element);
     return element;
   }
 }
 
 
 @ReflectiveTestCase()
 class _IndexNodeTest {
-  AnalysisContext context = new _MockAnalysisContext('context');
-  ElementCodec elementCodec = new _MockElementCodec();
+  AnalysisContext context = new MockAnalysisContext('context');
+  ElementCodec elementCodec = new MockElementCodec();
   int nextElementId = 0;
   IndexNode node;
   RelationshipCodec relationshipCodec;
   StringCodec stringCodec = new StringCodec();
 
   void setUp() {
     relationshipCodec = new RelationshipCodec(stringCodec);
     node = new IndexNode(context, elementCodec, relationshipCodec);
   }
 
@@ skipping 53 matching lines @@
     }
     // request
     List<Location> locations = node.getRelationships(elementA, relationship);
     expect(locations, hasLength(2));
     _assertHasLocation(locations, elementB, 1, 10);
     _assertHasLocation(locations, elementC, 2, 20);
   }
 
   Element _mockElement() {
     int elementId = nextElementId++;
-    Element element = new _MockElement();
+    Element element = new MockElement();
     when(elementCodec.encode(element)).thenReturn(elementId);
     when(elementCodec.decode(context, elementId)).thenReturn(element);
     return element;
   }
 }
 
 
 @ReflectiveTestCase()
-class _IntToIntSetMapTest {
-  IntToIntSetMap map = new IntToIntSetMap(32, 0.75);
-
-  void test_clear() {
-    map.add(1, 10);
-    map.add(2, 20);
-    expect(map.length, 2);
-    map.clear();
-    expect(map.length, 0);
-  }
-
-  void test_get() {
-    map.add(1, 10);
-    map.add(1, 11);
-    map.add(1, 12);
-    map.add(2, 20);
-    map.add(2, 21);
-    expect(map.get(1), unorderedEquals([10, 11, 12]));
-    expect(map.get(2), unorderedEquals([20, 21]));
-  }
-
-  void test_get_no() {
-    expect(map.get(3), []);
-  }
-
-  void test_length() {
-    expect(map.length, 0);
-    map.add(1, 10);
-    expect(map.length, 1);
-    map.add(1, 11);
-    expect(map.length, 2);
-    map.add(1, 12);
-    expect(map.length, 3);
-    map.add(2, 20);
-    expect(map.length, 4);
-    map.add(2, 21);
-    expect(map.length, 5);
-  }
-}
-
-
-@ReflectiveTestCase()
 class _LocationDataTest {
-  AnalysisContext context = new _MockAnalysisContext('context');
-  ElementCodec elementCodec = new _MockElementCodec();
+  AnalysisContext context = new MockAnalysisContext('context');
+  ElementCodec elementCodec = new MockElementCodec();
   StringCodec stringCodec = new StringCodec();
 
   void test_newForData() {
-    Element element = new _MockElement();
+    Element element = new MockElement();
     when(elementCodec.decode(context, 0)).thenReturn(element);
     LocationData locationData = new LocationData.forData(0, 1, 2);
     Location location = locationData.getLocation(context, elementCodec);
     expect(location.element, element);
     expect(location.offset, 1);
     expect(location.length, 2);
   }
 
   void test_newForObject() {
     // prepare Element
-    Element element = new _MockElement();
+    Element element = new MockElement();
     when(elementCodec.encode(element)).thenReturn(42);
     when(elementCodec.decode(context, 42)).thenReturn(element);
     // create
     Location location = new Location(element, 1, 2);
     LocationData locationData = new LocationData.forObject(elementCodec,
         location);
     // touch 'hashCode'
     locationData.hashCode;
     // ==
     expect(locationData == new LocationData.forData(42, 1, 2), isTrue);
@@ skipping 38 matching lines @@
     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
@@ skipping 59 matching lines @@
     _nodes[name] = node;
   }
 
   @override
   void removeNode(String name) {
     _nodes.remove(name);
   }
 }
 
 
-class _MockAnalysisContext extends TypedMock implements AnalysisContext {
-  String _name;
-  _MockAnalysisContext(this._name);
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-  String toString() => _name;
-}
-
-
-class _MockCompilationUnitElement extends TypedMock implements
-    CompilationUnitElement {
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-}
-
-
-class _MockContextCodec extends TypedMock implements ContextCodec {
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-}
-
-
-class _MockElement extends TypedMock implements Element {
-  String _name;
-  _MockElement([this._name = '<element>']);
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-  String toString() => _name;
-}
-
-
-class _MockElementCodec extends TypedMock implements ElementCodec {
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-}
-
-
 class _MockFileManager extends TypedMock implements FileManager {
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 
-class _MockHtmlElement extends TypedMock implements HtmlElement {
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-}
-
-
 class _MockIndexNode extends TypedMock implements IndexNode {
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 
-class _MockInstrumentedAnalysisContextImpl extends TypedMock implements
-    InstrumentedAnalysisContextImpl {
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-}
-
-
-class _MockLibraryElement extends TypedMock implements LibraryElement {
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-}
-
-
-class _MockLocation extends TypedMock implements Location {
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-}
-
-
-class _MockLogger extends TypedMock implements Logger {
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-}
-
-
-class _MockRelationshipCodec extends TypedMock implements RelationshipCodec {
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-}
-
-
-class _MockSource extends TypedMock implements Source {
-  String _name;
-  _MockSource(this._name);
-  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
-  String toString() => _name;
-}
-
-
 @ReflectiveTestCase()
 class _RelationKeyDataTest {
-  AnalysisContext context = new _MockAnalysisContext('context');
-  ElementCodec elementCodec = new _MockElementCodec();
-  RelationshipCodec relationshipCodec = new _MockRelationshipCodec();
+  AnalysisContext context = new MockAnalysisContext('context');
+  ElementCodec elementCodec = new MockElementCodec();
+  RelationshipCodec relationshipCodec = new MockRelationshipCodec();
   StringCodec stringCodec = new StringCodec();
 
   void test_newFromData() {
     RelationKeyData keyData = new RelationKeyData.forData(1, 2);
     // equals
     expect(keyData == this, isFalse);
     expect(keyData == new RelationKeyData.forData(10, 20), isFalse);
     expect(keyData == keyData, isTrue);
     expect(keyData == new RelationKeyData.forData(1, 2), isTrue);
   }
 
   void test_newFromObjects() {
     // prepare Element
     Element element;
     int elementId = 2;
     {
-      element = new _MockElement();
+      element = new MockElement();
       ElementLocation location = new ElementLocationImpl.con3(['foo', 'bar']);
       when(element.location).thenReturn(location);
       when(context.getElement(location)).thenReturn(element);
       when(elementCodec.encode(element)).thenReturn(elementId);
     }
     // prepare relationship
     Relationship relationship = Relationship.getRelationship('my-relationship');
     int relationshipId = 1;
     when(relationshipCodec.encode(relationship)).thenReturn(relationshipId);
     // create RelationKeyData
     RelationKeyData keyData = new RelationKeyData.forObject(elementCodec,
         relationshipCodec, element, relationship);
     // touch
     keyData.hashCode;
     // equals
     expect(keyData == this, isFalse);
     expect(keyData == new RelationKeyData.forData(10, 20), isFalse);
     expect(keyData == keyData, isTrue);
     expect(keyData == new RelationKeyData.forData(elementId, relationshipId),
         isTrue);
   }
 }
 
 
-@ReflectiveTestCase()
-class _RelationshipCodecTest {
-  RelationshipCodec codec;
-  StringCodec stringCodec = new StringCodec();
-
-  void setUp() {
-    codec = new RelationshipCodec(stringCodec);
-  }
-
-  void test_all() {
-    Relationship relationship = Relationship.getRelationship('my-relationship');
-    int id = codec.encode(relationship);
-    expect(codec.decode(id), relationship);
-  }
-}
-
-
 class _SingleSourceContainer implements SourceContainer {
   final Source _source;
   _SingleSourceContainer(this._source);
   @override
   bool contains(Source source) => source == _source;
 }
 
 
 @ReflectiveTestCase()
 class _SplitIndexStoreTest {
-  AnalysisContext contextA = new _MockAnalysisContext('contextA');
+  AnalysisContext contextA = new MockAnalysisContext('contextA');
 
-  AnalysisContext contextB = new _MockAnalysisContext('contextB');
+  AnalysisContext contextB = new MockAnalysisContext('contextB');
 
-  AnalysisContext contextC = new _MockAnalysisContext('contextC');
+  AnalysisContext contextC = new MockAnalysisContext('contextC');
 
-  Element elementA = new _MockElement('elementA');
-  Element elementB = new _MockElement('elementB');
+  Element elementA = new MockElement('elementA');
+  Element elementB = new MockElement('elementB');
 
-  Element elementC = new _MockElement('elementC');
-  Element elementD = new _MockElement('elementD');
+  Element elementC = new MockElement('elementC');
+  Element elementD = new MockElement('elementD');
   ElementLocation elementLocationA = new ElementLocationImpl.con3(
       ['/home/user/sourceA.dart', 'ClassA']);
   ElementLocation elementLocationB = new ElementLocationImpl.con3(
       ['/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();
+  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();
   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');
+  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();
+  CompilationUnitElement unitElementA = new MockCompilationUnitElement();
+  CompilationUnitElement unitElementB = new MockCompilationUnitElement();
+  CompilationUnitElement unitElementC = new MockCompilationUnitElement();
+  CompilationUnitElement unitElementD = new MockCompilationUnitElement();
   void setUp() {
     store = new SplitIndexStore(nodeManager);
     when(contextA.isDisposed).thenReturn(false);
     when(contextB.isDisposed).thenReturn(false);
     when(contextC.isDisposed).thenReturn(false);
     when(contextA.getElement(elementLocationA)).thenReturn(elementA);
     when(contextA.getElement(elementLocationB)).thenReturn(elementB);
     when(contextA.getElement(elementLocationC)).thenReturn(elementC);
     when(contextA.getElement(elementLocationD)).thenReturn(elementD);
     when(sourceA.fullName).thenReturn('/home/user/sourceA.dart');
@@ skipping 36 matching lines @@
     when(libraryElement.source).thenReturn(librarySource);
     when(libraryElement.definingCompilationUnit).thenReturn(libraryUnitElement);
   }
   void test_aboutToIndexDart_disposedContext() {
     when(contextA.isDisposed).thenReturn(true);
     expect(store.aboutToIndexDart(contextA, unitElementA), isFalse);
   }
   void test_aboutToIndexDart_disposedContext_wrapped() {
     when(contextA.isDisposed).thenReturn(true);
     InstrumentedAnalysisContextImpl instrumentedContext =
-        new _MockInstrumentedAnalysisContextImpl();
+        new MockInstrumentedAnalysisContextImpl();
     when(instrumentedContext.basis).thenReturn(contextA);
     expect(store.aboutToIndexDart(instrumentedContext, unitElementA), isFalse);
   }
 
   void test_aboutToIndexDart_library_first() {
     when(libraryElement.parts).thenReturn(<CompilationUnitElement>[unitElementA,
         unitElementB]);
     {
       store.aboutToIndexDart(contextA, libraryUnitElement);
       store.doneIndex();
@@ skipping 410 matching lines @@
   static void assertLocations(List<Location> actual, List<Location> expected) {
     List<_LocationEqualsWrapper> actualWrappers = wrapLocations(actual);
     List<_LocationEqualsWrapper> expectedWrappers = wrapLocations(expected);
     expect(actualWrappers, unorderedEquals(expectedWrappers));
   }
 
   /**
    * @return the new [Location] mock.
    */
   static Location mockLocation(Element element) {
-    Location location = new _MockLocation();
+    Location location = new MockLocation();
     when(location.element).thenReturn(element);
     when(location.offset).thenReturn(0);
     when(location.length).thenReturn(0);
     return location;
   }
 
   /**
    * 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;
   }
 }
-
-
-@ReflectiveTestCase()
-class _StringCodecTest {
-  StringCodec codec = new StringCodec();
-
-  void test_all() {
-    int idA = codec.encode('aaa');
-    int idB = codec.encode('bbb');
-    expect(codec.decode(idA), 'aaa');
-    expect(codec.decode(idB), 'bbb');
-  }
-}