Refactor out a FindUsagesTest base test class

2 files changed, 172 insertions, 138 deletions

diff --git a/smalidea/src/test/java/org/jf/smalidea/findUsages/FindClassUsagesTest.java b/smalidea/src/test/java/org/jf/smalidea/findUsages/FindClassUsagesTest.java
index a8e27843..810ac5b1 100644
--- a/smalidea/src/test/java/org/jf/smalidea/findUsages/FindClassUsagesTest.java
+++ b/smalidea/src/test/java/org/jf/smalidea/findUsages/FindClassUsagesTest.java
@@ -31,75 +31,7 @@
 
 package org.jf.smalidea.findUsages;
 
-import com.google.common.collect.Lists;
-import com.intellij.find.FindManager;
-import com.intellij.find.findUsages.FindUsagesHandler;
-import com.intellij.find.findUsages.FindUsagesManager;
-import com.intellij.find.findUsages.FindUsagesOptions;
-import com.intellij.find.impl.FindManagerImpl;
-import com.intellij.psi.PsiElement;
-import com.intellij.psi.PsiFile;
-import com.intellij.psi.PsiReference;
-import com.intellij.testFramework.PsiTestCase;
-import com.intellij.usageView.UsageInfo;
-import com.intellij.util.CommonProcessors;
-import org.jetbrains.annotations.NotNull;
-import org.junit.Assert;
-
-import java.util.Collection;
-import java.util.List;
-import java.util.regex.Matcher;
-import java.util.regex.Pattern;
-
-public class FindClassUsagesTest extends PsiTestCase {
-    private static final String USAGE_TAG = "<usage>";
-    private static final String REF_TAG = "<ref>";
-
-    private class TestFile {
-        @NotNull public final String fileName;
-        @NotNull public final String fileText;
-        @NotNull public final PsiFile psiFile;
-
-        public TestFile(@NotNull String fileName, @NotNull String fileText) throws Exception {
-            this.fileName = fileName;
-            this.fileText = fileText;
-            this.psiFile = createFile(fileName, getText());
-        }
-
-        @NotNull
-        public String getText() {
-            return fileText.replace(REF_TAG, "").replace(USAGE_TAG, "");
-        }
-
-        public int getRefIndex() {
-            return fileText.replace(USAGE_TAG, "").indexOf(REF_TAG);
-        }
-
-        public List<Integer> getUsageIndices() {
-            Matcher matcher = Pattern.compile(USAGE_TAG).matcher(fileText.replace(REF_TAG, ""));
-            List<Integer> matches = Lists.newArrayList();
-
-            int adjustment = 0;
-            while (matcher.find()) {
-                matches.add(matcher.start() - adjustment);
-                adjustment += USAGE_TAG.length();
-            }
-            return matches;
-        }
-    }
-
-    private List<TestFile> testFiles;
-
-    @Override
-    public void setUp() throws Exception {
-        testFiles = Lists.newArrayList();
-        super.setUp();
-    }
-
-    public void addFile(String fileName, String fileText) throws Exception {
-        testFiles.add(new TestFile(fileName, fileText));
-    }
-
+public class FindClassUsagesTest extends FindUsagesTest {
     public void testSmaliUsageInSmaliFile() throws Exception {
         addFile("blah.smali", ".class public Lbl<ref><usage>ah; .super Ljava/lang/Object;");
         addFile("blarg.smali", "" +
@@ -141,73 +73,4 @@ public class FindClassUsagesTest extends PsiTestCase {
                 ".end method");
         doTest();
     }
-
-    private void doTest() {
-        PsiReference reference = null;
-        for (TestFile testFile: testFiles) {
-            int refIndex = testFile.getRefIndex();
-            if (refIndex != -1) {
-                reference = testFile.psiFile.findReferenceAt(refIndex);
-                break;
-            }
-        }
-
-        Assert.assertNotNull(reference);
-
-        Collection<UsageInfo> usages = findUsages(reference);
-        for (TestFile testFile: testFiles) {
-            assertUsages(testFile, usages);
-        }
-    }
-
-    private void assertUsages(@NotNull TestFile testFile, @NotNull Collection<UsageInfo> usages) {
-        List<UsageInfo> fileUsages = Lists.newArrayList();
-        for (UsageInfo usage: usages) {
-            if (usage.getFile().getName().equals(testFile.fileName)) {
-                fileUsages.add(usage);
-            }
-        }
-
-        for (Integer usageIndex: testFile.getUsageIndices()) {
-            boolean found = false;
-            for (UsageInfo usage: fileUsages) {
-                int startOffset = usage.getElement().getNode().getStartOffset();
-                int length = usage.getElement().getTextLength();
-
-                if (usageIndex >= startOffset && usageIndex < startOffset + length) {
-                    fileUsages.remove(usage);
-                    found = true;
-                    break;
-                }
-            }
-            Assert.assertTrue(found);
-        }
-        Assert.assertEquals(0, fileUsages.size());
-    }
-
-    private Collection<UsageInfo> findUsages(@NotNull PsiReference reference) {
-        PsiElement resolved = reference.resolve();
-        Assert.assertNotNull(resolved);
-
-        FindUsagesManager findUsagesManager =
-                        ((FindManagerImpl)FindManager.getInstance(getProject())).getFindUsagesManager();
-
-        FindUsagesHandler findUsagesHandler =
-                findUsagesManager.getFindUsagesHandler(resolved, false);
-        Assert.assertNotNull(findUsagesHandler);
-
-        final FindUsagesOptions options = findUsagesHandler.getFindUsagesOptions();
-        final CommonProcessors.CollectProcessor<UsageInfo> processor =
-                new CommonProcessors.CollectProcessor<UsageInfo>();
-
-        for (PsiElement element : findUsagesHandler.getPrimaryElements()) {
-            findUsagesHandler.processElementUsages(element, processor, options);
-        }
-
-        for (PsiElement element : findUsagesHandler.getSecondaryElements()) {
-            findUsagesHandler.processElementUsages(element, processor, options);
-        }
-
-        return processor.getResults();
-    }
 }
diff --git a/smalidea/src/test/java/org/jf/smalidea/findUsages/FindUsagesTest.java b/smalidea/src/test/java/org/jf/smalidea/findUsages/FindUsagesTest.java
new file mode 100644
index 00000000..84000505
--- /dev/null
+++ b/smalidea/src/test/java/org/jf/smalidea/findUsages/FindUsagesTest.java
@@ -0,0 +1,171 @@
+/*
+ * Copyright 2015, Google Inc.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ *     * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *     * Redistributions in binary form must reproduce the above
+ * copyright notice, this list of conditions and the following disclaimer
+ * in the documentation and/or other materials provided with the
+ * distribution.
+ *     * Neither the name of Google Inc. nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+package org.jf.smalidea.findUsages;
+
+import com.google.common.collect.Lists;
+import com.intellij.find.FindManager;
+import com.intellij.find.findUsages.FindUsagesHandler;
+import com.intellij.find.findUsages.FindUsagesManager;
+import com.intellij.find.findUsages.FindUsagesOptions;
+import com.intellij.find.impl.FindManagerImpl;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiFile;
+import com.intellij.psi.PsiReference;
+import com.intellij.testFramework.PsiTestCase;
+import com.intellij.usageView.UsageInfo;
+import com.intellij.util.CommonProcessors;
+import org.jetbrains.annotations.NotNull;
+import org.junit.Assert;
+
+import java.util.Collection;
+import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+public abstract class FindUsagesTest extends PsiTestCase {
+    public static final String USAGE_TAG = "<usage>";
+    public static final String REF_TAG = "<ref>";
+
+    private class TestFile {
+        @NotNull public final String fileName;
+        @NotNull public final String fileText;
+        @NotNull public final PsiFile psiFile;
+
+        public TestFile(@NotNull String fileName, @NotNull String fileText) throws Exception {
+            this.fileName = fileName;
+            this.fileText = fileText;
+            this.psiFile = createFile(fileName, getText());
+        }
+
+        @NotNull
+        public String getText() {
+            return fileText.replace(REF_TAG, "").replace(USAGE_TAG, "");
+        }
+
+        public int getRefIndex() {
+            return fileText.replace(USAGE_TAG, "").indexOf(REF_TAG);
+        }
+
+        public List<Integer> getUsageIndices() {
+            Matcher matcher = Pattern.compile(USAGE_TAG).matcher(fileText.replace(REF_TAG, ""));
+            List<Integer> matches = Lists.newArrayList();
+
+            int adjustment = 0;
+            while (matcher.find()) {
+                matches.add(matcher.start() - adjustment);
+                adjustment += USAGE_TAG.length();
+            }
+            return matches;
+        }
+    }
+
+    private List<TestFile> testFiles;
+
+    @Override
+    public void setUp() throws Exception {
+        testFiles = Lists.newArrayList();
+        super.setUp();
+    }
+
+    protected void addFile(String fileName, String fileText) throws Exception {
+        testFiles.add(new TestFile(fileName, fileText));
+    }
+
+    protected void doTest() {
+        PsiReference reference = null;
+        for (TestFile testFile: testFiles) {
+            int refIndex = testFile.getRefIndex();
+            if (refIndex != -1) {
+                reference = testFile.psiFile.findReferenceAt(refIndex);
+                break;
+            }
+        }
+
+        Assert.assertNotNull(reference);
+
+        Collection<UsageInfo> usages = findUsages(reference);
+        for (TestFile testFile: testFiles) {
+            assertUsages(testFile, usages);
+        }
+    }
+
+    private void assertUsages(@NotNull TestFile testFile, @NotNull Collection<UsageInfo> usages) {
+        List<UsageInfo> fileUsages = Lists.newArrayList();
+        for (UsageInfo usage: usages) {
+            if (usage.getFile().getName().equals(testFile.fileName)) {
+                fileUsages.add(usage);
+            }
+        }
+
+        for (Integer usageIndex: testFile.getUsageIndices()) {
+            boolean found = false;
+            for (UsageInfo usage: fileUsages) {
+                int startOffset = usage.getElement().getNode().getStartOffset();
+                int length = usage.getElement().getTextLength();
+
+                if (usageIndex >= startOffset && usageIndex < startOffset + length) {
+                    fileUsages.remove(usage);
+                    found = true;
+                    break;
+                }
+            }
+            Assert.assertTrue(found);
+        }
+        Assert.assertEquals(0, fileUsages.size());
+    }
+
+    private Collection<UsageInfo> findUsages(@NotNull PsiReference reference) {
+        PsiElement resolved = reference.resolve();
+        Assert.assertNotNull(resolved);
+
+        FindUsagesManager findUsagesManager =
+                ((FindManagerImpl)FindManager.getInstance(getProject())).getFindUsagesManager();
+
+        FindUsagesHandler findUsagesHandler =
+                findUsagesManager.getFindUsagesHandler(resolved, false);
+        Assert.assertNotNull(findUsagesHandler);
+
+        final FindUsagesOptions options = findUsagesHandler.getFindUsagesOptions();
+        final CommonProcessors.CollectProcessor<UsageInfo> processor =
+                new CommonProcessors.CollectProcessor<UsageInfo>();
+
+        for (PsiElement element : findUsagesHandler.getPrimaryElements()) {
+            findUsagesHandler.processElementUsages(element, processor, options);
+        }
+
+        for (PsiElement element : findUsagesHandler.getSecondaryElements()) {
+            findUsagesHandler.processElementUsages(element, processor, options);
+        }
+
+        return processor.getResults();
+    }
+}