1 files changed, 15 insertions, 16 deletions

diff --git a/src/test/java/cuchaz/enigma/TestJarIndexConstructorReferences.java b/src/test/java/cuchaz/enigma/TestJarIndexConstructorReferences.java
index 763639a..c3f3b66 100644
--- a/src/test/java/cuchaz/enigma/TestJarIndexConstructorReferences.java
+++ b/src/test/java/cuchaz/enigma/TestJarIndexConstructorReferences.java
@@ -12,12 +12,11 @@
 package cuchaz.enigma;
 
 import cuchaz.enigma.analysis.EntryReference;
-import cuchaz.enigma.analysis.JarIndex;
 import cuchaz.enigma.analysis.ParsedJar;
-import cuchaz.enigma.mapping.entry.ClassEntry;
-import cuchaz.enigma.mapping.entry.MethodDefEntry;
-import cuchaz.enigma.mapping.entry.MethodEntry;
-import cuchaz.enigma.mapping.entry.ReferencedEntryPool;
+import cuchaz.enigma.analysis.index.JarIndex;
+import cuchaz.enigma.translation.representation.entry.ClassEntry;
+import cuchaz.enigma.translation.representation.entry.MethodDefEntry;
+import cuchaz.enigma.translation.representation.entry.MethodEntry;
 import org.junit.Test;
 
 import java.io.File;
@@ -41,13 +40,13 @@ public class TestJarIndexConstructorReferences {
         public TestJarIndexConstructorReferences()
                         throws Exception {
                 File jarFile = new File("build/test-obf/constructors.jar");
-                index = new JarIndex(new ReferencedEntryPool());
-                index.indexJar(new ParsedJar(new JarFile(jarFile)), false);
+                index = JarIndex.empty();
+                index.indexJar(new ParsedJar(new JarFile(jarFile)), s -> {});
         }
 
         @Test
         public void obfEntries() {
-                assertThat(index.getObfClassEntries(), containsInAnyOrder(newClass("cuchaz/enigma/inputs/Keep"), baseClass,
+                assertThat(index.getEntryIndex().getClasses(), containsInAnyOrder(newClass("cuchaz/enigma/inputs/Keep"), baseClass,
                                 subClass, subsubClass, defaultClass, callerClass));
         }
 
@@ -55,7 +54,7 @@ public class TestJarIndexConstructorReferences {
         @SuppressWarnings("unchecked")
         public void baseDefault() {
                 MethodEntry source = newMethod(baseClass, "<init>", "()V");
-                Collection<EntryReference<MethodEntry, MethodDefEntry>> references = index.getMethodsReferencing(source);
+                Collection<EntryReference<MethodEntry, MethodDefEntry>> references = index.getReferenceIndex().getReferencesToMethod(source);
                 assertThat(references, containsInAnyOrder(
                                 newBehaviorReferenceByMethod(source, callerClass.getName(), "a", "()V"),
                                 newBehaviorReferenceByMethod(source, subClass.getName(), "<init>", "()V"),
@@ -67,7 +66,7 @@ public class TestJarIndexConstructorReferences {
         @SuppressWarnings("unchecked")
         public void baseInt() {
                 MethodEntry source = newMethod(baseClass, "<init>", "(I)V");
-                assertThat(index.getMethodsReferencing(source), containsInAnyOrder(
+                assertThat(index.getReferenceIndex().getReferencesToMethod(source), containsInAnyOrder(
                                 newBehaviorReferenceByMethod(source, callerClass.getName(), "b", "()V")
                 ));
         }
@@ -76,7 +75,7 @@ public class TestJarIndexConstructorReferences {
         @SuppressWarnings("unchecked")
         public void subDefault() {
                 MethodEntry source = newMethod(subClass, "<init>", "()V");
-                assertThat(index.getMethodsReferencing(source), containsInAnyOrder(
+                assertThat(index.getReferenceIndex().getReferencesToMethod(source), containsInAnyOrder(
                                 newBehaviorReferenceByMethod(source, callerClass.getName(), "c", "()V"),
                                 newBehaviorReferenceByMethod(source, subClass.getName(), "<init>", "(I)V")
                 ));
@@ -86,7 +85,7 @@ public class TestJarIndexConstructorReferences {
         @SuppressWarnings("unchecked")
         public void subInt() {
                 MethodEntry source = newMethod(subClass, "<init>", "(I)V");
-                assertThat(index.getMethodsReferencing(source), containsInAnyOrder(
+                assertThat(index.getReferenceIndex().getReferencesToMethod(source), containsInAnyOrder(
                                 newBehaviorReferenceByMethod(source, callerClass.getName(), "d", "()V"),
                                 newBehaviorReferenceByMethod(source, subClass.getName(), "<init>", "(II)V"),
                                 newBehaviorReferenceByMethod(source, subsubClass.getName(), "<init>", "(I)V")
@@ -97,7 +96,7 @@ public class TestJarIndexConstructorReferences {
         @SuppressWarnings("unchecked")
         public void subIntInt() {
                 MethodEntry source = newMethod(subClass, "<init>", "(II)V");
-                assertThat(index.getMethodsReferencing(source), containsInAnyOrder(
+                assertThat(index.getReferenceIndex().getReferencesToMethod(source), containsInAnyOrder(
                                 newBehaviorReferenceByMethod(source, callerClass.getName(), "e", "()V")
                 ));
         }
@@ -105,14 +104,14 @@ public class TestJarIndexConstructorReferences {
         @Test
         public void subIntIntInt() {
                 MethodEntry source = newMethod(subClass, "<init>", "(III)V");
-                assertThat(index.getMethodsReferencing(source), is(empty()));
+                assertThat(index.getReferenceIndex().getReferencesToMethod(source), is(empty()));
         }
 
         @Test
         @SuppressWarnings("unchecked")
         public void subsubInt() {
                 MethodEntry source = newMethod(subsubClass, "<init>", "(I)V");
-                assertThat(index.getMethodsReferencing(source), containsInAnyOrder(
+                assertThat(index.getReferenceIndex().getReferencesToMethod(source), containsInAnyOrder(
                                 newBehaviorReferenceByMethod(source, callerClass.getName(), "f", "()V")
                 ));
         }
@@ -121,7 +120,7 @@ public class TestJarIndexConstructorReferences {
         @SuppressWarnings("unchecked")
         public void defaultConstructable() {
                 MethodEntry source = newMethod(defaultClass, "<init>", "()V");
-                assertThat(index.getMethodsReferencing(source), containsInAnyOrder(
+                assertThat(index.getReferenceIndex().getReferencesToMethod(source), containsInAnyOrder(
                                 newBehaviorReferenceByMethod(source, callerClass.getName(), "g", "()V")
                 ));
         }