/*
* Copyright 2000-2017 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.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.apache.org/licenses/LICENSE-2.0
*
* 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.intellij.psi;
import com.intellij.JavaTestUtil;
import com.intellij.ide.highlighter.JavaFileType;
import com.intellij.openapi.editor.Document;
import com.intellij.openapi.fileEditor.FileDocumentManager;
import com.intellij.openapi.util.io.FileUtil;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vfs.StandardFileSystems;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.codeStyle.CodeStyleSettingsManager;
import com.intellij.psi.codeStyle.CommonCodeStyleSettings;
import com.intellij.psi.impl.compiled.ClsElementImpl;
import com.intellij.psi.impl.compiled.ClsFileImpl;
import com.intellij.psi.impl.compiled.InnerClassSourceStrategy;
import com.intellij.psi.impl.compiled.StubBuildingVisitor;
import com.intellij.psi.impl.java.stubs.impl.PsiJavaFileStubImpl;
import com.intellij.testFramework.LightIdeaTestCase;
import com.intellij.testFramework.PlatformTestUtil;
import org.jetbrains.org.objectweb.asm.ClassReader;
import java.io.File;
import java.io.IOException;
public class ClsMirrorBuildingTest extends LightIdeaTestCase {
public void testSimpleEnum() { doTest(); }
public void testEnumWithFields() { doTest(); }
public void testNormalClass() { doTest(); }
public void testNested() { doTest(); }
public void testDeprecated() { doTest(); }
public void testAnnotations() { doTest(); }
public void testMoreAnnotations() { doTest(); }
public void testMemberAnnotations() { doTest(); }
public void testParameterNames() { doTest(); }
public void testEmptyEnum() { doTest(); }
public void test$BuckClass() { doTest(); }
public void testBuckClass$() { doTest(); }
public void testExtMethods() { doTest(); }
public void testMethodReceiver() { doTest(); }
public void testPackageInfo() { doTest("package-info"); }
public void testEA40568() { doTest(); }
public void testPrimitives() { doTest(); }
public void testClassRefs() { doTest(); }
public void testEA46236() { doTest("ValuedEnum"); }
public void testKotlinFunList() { doTest(); }
public void testMiddle$Buck() { doTest(); }
public void testDefaultPackage() { doTest(); }
public void testLocalClass() { doTest(); }
public void testBounds() { doTest(); }
public void testGrEnum() { doTest(); }
public void testSuspiciousParameterNames() { doTest(); }
public void testTypeAnnotations() { doTest(); }
public void testTextPsiMismatch() {
CommonCodeStyleSettings.IndentOptions options =
CodeStyleSettingsManager.getInstance(getProject()).getCurrentSettings().getIndentOptions(JavaFileType.INSTANCE);
int indent = options.INDENT_SIZE;
options.INDENT_SIZE *= 2;
try {
doTest("Bounds");
}
finally {
options.INDENT_SIZE = indent;
}
}
public void testJdk8Class() {
String testDir = JavaTestUtil.getJavaTestDataPath();
doTest(testDir + "/../../mockJDK-1.8/jre/lib/rt.jar!/java/lang/Class.class", testDir + "/psi/cls/mirror/Class.txt");
}
public void testStaticMethodInInterface() {
String testDir = JavaTestUtil.getJavaTestDataPath();
doTest(testDir + "/../../mockJDK-1.8/jre/lib/rt.jar!/java/util/function/Function.class", testDir + "/psi/cls/mirror/Function.txt");
}
public void testStrayInnersFiltering() throws IOException {
String path = JavaTestUtil.getJavaTestDataPath() + "/../../mockJDK-1.8/jre/lib/rt.jar!/java/lang/Class.class";
VirtualFile file = StandardFileSystems.jar().findFileByPath(path);
assertNotNull(path, file);
InnerClassSourceStrategy<VirtualFile> strategy = new InnerClassSourceStrategy<VirtualFile>() {
@Override
public VirtualFile findInnerClass(String innerName, VirtualFile outerClass) {
String baseName = outerClass.getNameWithoutExtension();
VirtualFile child = outerClass.getParent().findChild(baseName + "$" + innerName + ".class");
// stray inner classes should be filtered out
assert child != null : innerName + " is not an inner class of " + outerClass;
return child;
}
@Override
public void accept(VirtualFile innerClass, StubBuildingVisitor<VirtualFile> visitor) {
try {
byte[] bytes = innerClass.contentsToByteArray();
new ClassReader(bytes).accept(visitor, ClassReader.SKIP_FRAMES);
}
catch (IOException ignored) { }
}
};
PsiJavaFileStubImpl stub = new PsiJavaFileStubImpl("java.lang", true);
StubBuildingVisitor<VirtualFile> visitor = new StubBuildingVisitor<>(file, strategy, stub, 0, null);
new ClassReader(file.contentsToByteArray()).accept(visitor, ClassReader.SKIP_FRAMES);
}
public void testDocumentReuse() throws IOException {
File classFile = new File(FileUtil.getTempDirectory(), "ReuseTest.class");
FileUtil.writeToFile(classFile, "");
VirtualFile vFile = StandardFileSystems.local().refreshAndFindFileByPath(classFile.getPath());
assertNotNull(classFile.getPath(), vFile);
PsiFile psiFile = PsiManager.getInstance(getProject()).findFile(vFile);
assertNotNull(psiFile);
String testDir = getTestDataDir();
FileUtil.copy(new File(testDir, "pkg/ReuseTestV1.class"), classFile);
vFile.refresh(false, false);
PsiDocumentManager.getInstance(getProject()).commitAllDocuments();
String text1 = psiFile.getText();
assertTrue(text1, text1.contains("private int f1"));
assertFalse(text1, text1.contains("private int f2"));
Document doc1 = FileDocumentManager.getInstance().getCachedDocument(vFile);
assertNotNull(doc1);
assertSame(doc1, PsiDocumentManager.getInstance(getProject()).getDocument(psiFile));
FileUtil.copy(new File(testDir, "pkg/ReuseTestV2.class"), classFile);
vFile.refresh(false, false);
PsiDocumentManager.getInstance(getProject()).commitAllDocuments();
String text2 = psiFile.getText();
assertTrue(text2, text2.contains("private int f1"));
assertTrue(text2, text2.contains("private int f2"));
Document doc2 = FileDocumentManager.getInstance().getCachedDocument(vFile);
assertNotNull(doc2);
assertSame(doc2, PsiDocumentManager.getInstance(getProject()).getDocument(psiFile));
}
public void testElementAt() {
String path = getTestDataDir() + "/pkg/SimpleEnum.class";
VirtualFile vFile = StandardFileSystems.local().findFileByPath(path);
assertNotNull(path, vFile);
PsiFile psiFile = PsiManager.getInstance(getProject()).findFile(vFile);
assertNotNull(path, psiFile);
for (int i = 0; i < psiFile.getTextLength(); i++) {
PsiElement element = psiFile.findElementAt(i);
assertTrue(i + ":" + element, element != null && !(element instanceof ClsElementImpl));
}
}
public void testInnerClassDetection() throws IOException {
assertTrue(isInner("pkg/Nested$Inner1"));
assertTrue(isInner("pkg/Nested$Inner1$Inner2"));
assertTrue(isInner("pkg/NormalClass$1"));
assertTrue(isInner("pkg/LocalClass$1MyRunnable"));
assertTrue(isInner("pkg/KindaInner$RealInner$"));
assertTrue(isInner("pkg/KindaInner$Real$Inner"));
assertTrue(isInner("pkg/Groovy$Inner"));
assertTrue(isInner("pkg/Groovy$_closure1"));
assertTrue(isInner("weird/ToStringStyle$1"));
assertFalse(isInner("pkg/KindaInner$Class"));
assertFalse(isInner("weird/ToStringStyle"));
}
public void testModuleInfo() {
String testDir = getTestDataDir();
doTest(testDir + "../stubBuilder/module-info.class", testDir + "module-info.txt");
}
private static String getTestDataDir() {
return JavaTestUtil.getJavaTestDataPath() + "/psi/cls/mirror/";
}
private void doTest() {
doTest(getTestName(false));
}
private static void doTest(String name) {
String testDir = getTestDataDir();
doTest(testDir + "pkg/" + name + ".class", testDir + name + ".txt");
}
private static void doTest(String clsPath, String txtPath) {
VirtualFile file = (clsPath.contains("!/") ? StandardFileSystems.jar() : StandardFileSystems.local()).refreshAndFindFileByPath(clsPath);
assertNotNull(clsPath, file);
String expected;
try {
expected = StringUtil.trimTrailing(PlatformTestUtil.loadFileText(txtPath));
}
catch (IOException e) {
throw new RuntimeException(e);
}
assertEquals(expected, ClsFileImpl.decompile(file).toString());
((ClsFileImpl)getPsiManager().findFile(file)).getMirror();
}
private static boolean isInner(String name) throws IOException {
VirtualFile file = StandardFileSystems.local().findFileByPath(getTestDataDir() + name + ".class");
assertNotNull(file);
return ClassFileViewProvider.isInnerClass(file, file.contentsToByteArray(false));
}
}