Version 0.8.10.8

dart/editor/tools/plugins/com.google.dart.engine_test/src/com/google/dart/engine/internal/index/MemoryIndexStoreImplTest.java

 /*
  * Copyright (c) 2012, the Dart project authors.
  * 
  * Licensed under the Eclipse Public License v1.0 (the "License"); you may not use this file except
  * in compliance with the License. You may obtain a copy of the License at
  * 
  * http://www.eclipse.org/legal/epl-v10.html
  * 
  * Unless required by applicable law or agreed to in writing, software distributed under the License
  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
  * or implied. See the License for the specific language governing permissions and limitations under
  * the License.
  */
 package com.google.dart.engine.internal.index;
 
 import com.google.common.base.Objects;
 import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.Lists;
 import com.google.dart.engine.EngineTestCase;
 import com.google.dart.engine.context.AnalysisContext;
 import com.google.dart.engine.element.CompilationUnitElement;
 import com.google.dart.engine.element.Element;
 import com.google.dart.engine.element.ElementLocation;
 import com.google.dart.engine.index.Location;
 import com.google.dart.engine.index.Relationship;
 import com.google.dart.engine.internal.context.InstrumentedAnalysisContextImpl;
 import com.google.dart.engine.internal.element.ElementLocationImpl;
+import com.google.dart.engine.internal.element.member.Member;
+import com.google.dart.engine.source.DirectoryBasedSourceContainer;
 import com.google.dart.engine.source.Source;
 import com.google.dart.engine.source.SourceContainer;
 
 import org.mockito.invocation.InvocationOnMock;
 import org.mockito.stubbing.Answer;
 
 import static org.fest.assertions.Assertions.assertThat;
 import static org.mockito.Matchers.any;
 import static org.mockito.Matchers.eq;
 import static org.mockito.Mockito.mock;
@@ skipping 174 matching lines @@
   }
 
   public void test_recordRelationship() throws Exception {
     // no relationships initially
     assertEquals(0, store.internalGetLocationCount());
     // record relationship
     store.recordRelationship(elementA, relationship, location);
     assertEquals(1, store.internalGetLocationCount());
   }
 
+  public void test_recordRelationship_member() throws Exception {
+    Member member = mock(Member.class);
+    when(member.getBaseElement()).thenReturn(elementA);
+    // no relationships initially
+    assertEquals(0, store.internalGetLocationCount());
+    // record relationship
+    store.recordRelationship(member, relationship, location);
+    // no location for "member"
+    {
+      Location[] locations = store.getRelationships(member, relationship);
+      assertLocations(locations);
+    }
+    // has location for "elementA"
+    {
+      Location[] locations = store.getRelationships(elementA, relationship);
+      assertLocations(locations, location);
+    }
+  }
+
   public void test_recordRelationship_noElement() throws Exception {
     store.recordRelationship(null, relationship, location);
     assertEquals(0, store.internalGetLocationCount());
   }
 
+  public void test_recordRelationship_noElementContext() throws Exception {
+    when(elementA.getContext()).thenReturn(null);
+    store.recordRelationship(elementA, relationship, location);
+    assertEquals(0, store.internalGetLocationCount());
+  }
+
+  public void test_recordRelationship_noElementSource() throws Exception {
+    when(elementA.getSource()).thenReturn(null);
+    store.recordRelationship(elementA, relationship, location);
+    assertEquals(0, store.internalGetLocationCount());
+  }
+
   public void test_recordRelationship_noLocation() throws Exception {
     store.recordRelationship(elementA, relationship, null);
     assertEquals(0, store.internalGetLocationCount());
   }
 
-  public void test_recordRelationship_noLocationElement() throws Exception {
-    Element elementWithoutEnclosing = mock(Element.class);
-    Location location = new Location(elementWithoutEnclosing, 0, 0);
+  public void test_recordRelationship_noLocationContext() throws Exception {
+    when(location.getElement().getContext()).thenReturn(null);
+    store.recordRelationship(elementA, relationship, location);
+    assertEquals(0, store.internalGetLocationCount());
+  }
+
+  public void test_recordRelationship_noLocationSource() throws Exception {
+    when(location.getElement().getSource()).thenReturn(null);
     store.recordRelationship(elementA, relationship, location);
     assertEquals(0, store.internalGetLocationCount());
   }
 
   public void test_removeContext_instrumented() throws Exception {
     InstrumentedAnalysisContextImpl instrumentedContext = mock(InstrumentedAnalysisContextImpl.class);
     when(instrumentedContext.getBasis()).thenReturn(contextA);
     // configure B
     when(elementB.getContext()).thenReturn(contextA);
     Location locationB = mockLocation(elementB);
@@ skipping 169 matching lines @@
     store.removeSource(contextC, sourceB);
     assertEquals(0, store.internalGetLocationCount());
     assertEquals(0, store.internalGetLocationCount(contextB));
     assertEquals(0, store.internalGetLocationCount(contextC));
     {
       Location[] locations = store.getRelationships(elementA, relationship);
       assertThat(locations).isEmpty();
     }
   }
 
+  public void test_removeSources_nullContext() throws Exception {
+    // record
+    {
+      store.recordRelationship(IndexConstants.UNIVERSE, relationship, location);
+      assertEquals(1, store.internalGetLocationCount());
+    }
+    // remove "null" context, should never happen - ignored
+    SourceContainer sourceContainer = new DirectoryBasedSourceContainer("/path/");
+    store.removeSources(null, sourceContainer);
+    assertEquals(1, store.internalGetLocationCount());
+  }
+
   public void test_removeSources_withDeclaration() throws Exception {
     Location locationB = mockLocation(elementB);
     Location locationC = mockLocation(elementC);
     // record: A, [B -> A],  [C -> A] and [B -> C]
     {
       store.recordRelationship(elementA, relationship, locationB);
       store.recordRelationship(elementA, relationship, locationC);
       store.recordRelationship(elementC, relationship, locationB);
       assertEquals(3, store.internalGetLocationCount());
       Location[] locations = store.getRelationships(elementA, relationship);
@@ skipping 139 matching lines @@
     when(elementA.getSource()).thenReturn(sourceA);
     when(elementB.getSource()).thenReturn(sourceB);
     when(elementC.getSource()).thenReturn(sourceC);
     when(elementD.getSource()).thenReturn(sourceD);
     when(unitElementA.getSource()).thenReturn(sourceA);
     when(unitElementB.getSource()).thenReturn(sourceB);
     when(unitElementC.getSource()).thenReturn(sourceC);
     when(unitElementD.getSource()).thenReturn(sourceD);
   }
 }