Refactor out a FindUsagesTest base test class
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 a8e2784..810ac5b 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 @@
".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 0000000..8400050
--- /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();
+ }
+}
