mirror of
https://github.com/revanced/smali.git
synced 2025-05-16 14:17:04 +02:00
Refactor out a FindUsagesTest base test class
This commit is contained in:
Ben Gruver
parent
809b934012
commit
fd373aedb7
@ -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();
|
||||
}
|
||||
}
|
||||
|
||||
@ -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();
|
||||
}
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user