Alexander Udalov
74d6ef73d4
Add an assertion to prevent duplicate initialization of CodegenTestFiles instance in CodegenTestCase, which was happening in ScriptGenTest with multiple invocation of loadFile()
231 lines
10 KiB
Java
231 lines
10 KiB
Java
/*
|
|
* Copyright 2010-2015 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 org.jetbrains.kotlin.codegen;
|
|
|
|
import com.intellij.openapi.util.Ref;
|
|
import kotlin.io.FilesKt;
|
|
import org.intellij.lang.annotations.Language;
|
|
import org.jetbrains.annotations.NotNull;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import org.jetbrains.kotlin.backend.common.output.OutputFile;
|
|
import org.jetbrains.kotlin.backend.common.output.OutputFileCollection;
|
|
import org.jetbrains.kotlin.name.SpecialNames;
|
|
import org.jetbrains.kotlin.test.ConfigurationKind;
|
|
import org.jetbrains.kotlin.test.KotlinTestUtils;
|
|
import org.jetbrains.kotlin.utils.StringsKt;
|
|
import org.jetbrains.org.objectweb.asm.ClassReader;
|
|
import org.jetbrains.org.objectweb.asm.ClassVisitor;
|
|
import org.jetbrains.org.objectweb.asm.Opcodes;
|
|
|
|
import java.io.File;
|
|
import java.util.Collections;
|
|
|
|
public class OuterClassGenTest extends CodegenTestCase {
|
|
@NotNull
|
|
@Override
|
|
protected String getPrefix() {
|
|
return "outerClassInfo";
|
|
}
|
|
|
|
public void testClass() throws Exception {
|
|
doTest("foo.Foo", "outerClassInfo");
|
|
}
|
|
|
|
public void testClassObject() throws Exception {
|
|
doTest("foo.Foo$" + SpecialNames.DEFAULT_NAME_FOR_COMPANION_OBJECT.asString(), "outerClassInfo");
|
|
}
|
|
|
|
public void testInnerClass() throws Exception {
|
|
doTest("foo.Foo$InnerClass", "outerClassInfo");
|
|
}
|
|
|
|
public void testInnerObject() throws Exception {
|
|
doTest("foo.Foo$InnerObject", "outerClassInfo");
|
|
}
|
|
|
|
public void testLocalClassInFunction() throws Exception {
|
|
doTest("foo.Foo$foo$LocalClass", "foo.Foo$1LocalClass", "outerClassInfo");
|
|
}
|
|
|
|
public void testLocalObjectInFunction() throws Exception {
|
|
doTest("foo.Foo$foo$LocalObject", "foo.Foo$1LocalObject", "outerClassInfo");
|
|
}
|
|
|
|
public void testObjectInPackageClass() throws Exception {
|
|
doTest("foo.PackageInnerObject", "outerClassInfo");
|
|
}
|
|
|
|
public void testLambdaInNoInlineFun() throws Exception {
|
|
doTest("foo.Foo$foo$1", "foo.Foo$1Lambda", "outerClassInfo");
|
|
}
|
|
|
|
public void testLambdaInConstructor() throws Exception {
|
|
doTest("foo.Foo$s$1", "foo.Foo$1LambdaInConstructor", "outerClassInfo");
|
|
}
|
|
|
|
public void testObjectLiteralInPackageClass() throws Exception {
|
|
OuterClassInfo expectedInfo = new OuterClassInfo("foo/OuterClassInfo", null, null);
|
|
doCustomTest("foo/OuterClassInfoKt\\$packageObjectLiteral\\$1", expectedInfo, "outerClassInfo");
|
|
}
|
|
|
|
public void testLocalClassInTopLevelFunction() throws Exception {
|
|
OuterClassInfo expectedInfo = new OuterClassInfo("foo/OuterClassInfo", "packageMethod", "(Lfoo/Foo;)V");
|
|
doCustomTest("foo/OuterClassInfoKt\\$packageMethod\\$PackageLocalClass", expectedInfo, "outerClassInfo");
|
|
}
|
|
|
|
public void testLocalObjectInTopLevelFunction() throws Exception {
|
|
OuterClassInfo expectedInfo = new OuterClassInfo("foo/OuterClassInfo", "packageMethod", "(Lfoo/Foo;)V");
|
|
doCustomTest("foo/OuterClassInfoKt\\$packageMethod\\$PackageLocalObject", expectedInfo, "outerClassInfo");
|
|
}
|
|
|
|
public void testLocalObjectInInlineFunction() throws Exception {
|
|
OuterClassInfo expectedInfo = new OuterClassInfo("foo/Foo", "inlineFoo", "(Lkotlin/jvm/functions/Function0;)V");
|
|
doCustomTest("foo/Foo\\$inlineFoo\\$localObject\\$1", expectedInfo, "inlineObject");
|
|
}
|
|
|
|
public void testLocalObjectInlined() throws Exception {
|
|
OuterClassInfo expectedInfo = new OuterClassInfo("foo/Bar", "callToInline", "()V");
|
|
doCustomTest("foo/Bar\\$callToInline\\$\\$inlined\\$inlineFoo\\$1", expectedInfo, "inlineObject");
|
|
}
|
|
|
|
public void testLocalObjectInInlineLambda() throws Exception {
|
|
OuterClassInfo expectedInfo = new OuterClassInfo("foo/Bar", "objectInInlineLambda", "()V");
|
|
doCustomTest("foo/Bar\\$objectInInlineLambda\\$\\$inlined\\$simpleFoo\\$lambda\\$1", expectedInfo, "inlineObject");
|
|
}
|
|
|
|
public void testLocalObjectInLambdaInlinedIntoObject() throws Exception {
|
|
OuterClassInfo intoObjectInfo = new OuterClassInfo("foo/Bar", "objectInLambdaInlinedIntoObject", "()V");
|
|
doCustomTest("foo/Bar\\$objectInLambdaInlinedIntoObject\\$\\$inlined\\$inlineFoo\\$1", intoObjectInfo, "inlineObject");
|
|
}
|
|
|
|
public void testLocalObjectInLambdaInlinedIntoObject2() throws Exception {
|
|
OuterClassInfo objectInLambda = new OuterClassInfo("foo/Bar$objectInLambdaInlinedIntoObject$$inlined$inlineFoo$1", "run", "()V");
|
|
doCustomTest("foo/Bar\\$objectInLambdaInlinedIntoObject\\$\\$inlined\\$inlineFoo\\$1\\$lambda\\$1",
|
|
objectInLambda, "inlineObject");
|
|
}
|
|
|
|
public void testLambdaInInlineFunction() throws Exception {
|
|
OuterClassInfo expectedInfo = new OuterClassInfo("foo/Foo", "inlineFoo", "(Lkotlin/jvm/functions/Function0;)V");
|
|
doCustomTest("foo/Foo\\$inlineFoo\\$1", expectedInfo, "inlineLambda");
|
|
}
|
|
|
|
public void testLambdaInlined() throws Exception {
|
|
OuterClassInfo expectedInfo = new OuterClassInfo("foo/Bar", "callToInline", "()V");
|
|
doCustomTest("foo/Bar\\$callToInline\\$\\$inlined\\$inlineFoo\\$1", expectedInfo, "inlineLambda");
|
|
}
|
|
|
|
public void testLambdaInInlineLambda() throws Exception {
|
|
OuterClassInfo expectedInfo = new OuterClassInfo("foo/Bar", "objectInInlineLambda", "()V");
|
|
doCustomTest("foo/Bar\\$objectInInlineLambda\\$\\$inlined\\$simpleFoo\\$lambda\\$1", expectedInfo, "inlineLambda");
|
|
}
|
|
|
|
public void testLambdaInLambdaInlinedIntoObject() throws Exception {
|
|
OuterClassInfo intoObjectInfo = new OuterClassInfo("foo/Bar", "objectInLambdaInlinedIntoObject", "()V");
|
|
doCustomTest("foo/Bar\\$objectInLambdaInlinedIntoObject\\$\\$inlined\\$inlineFoo\\$1", intoObjectInfo, "inlineLambda");
|
|
}
|
|
|
|
public void testLambdaInLambdaInlinedIntoObject2() throws Exception {
|
|
OuterClassInfo objectInLambda = new OuterClassInfo("foo/Bar$objectInLambdaInlinedIntoObject$$inlined$inlineFoo$1", "invoke", "()V");
|
|
doCustomTest("foo/Bar\\$objectInLambdaInlinedIntoObject\\$\\$inlined\\$inlineFoo\\$1\\$lambda\\$1",
|
|
objectInLambda, "inlineLambda");
|
|
}
|
|
|
|
private void doTest(@NotNull String classFqName, @NotNull String testDataFile) throws Exception {
|
|
doTest(classFqName, classFqName, testDataFile);
|
|
}
|
|
|
|
@Override
|
|
protected void setUp() throws Exception {
|
|
super.setUp();
|
|
createEnvironmentWithMockJdkAndIdeaAnnotations(ConfigurationKind.JDK_ONLY);
|
|
}
|
|
|
|
private void doTest(@NotNull String classFqName, @NotNull String javaClassName, @NotNull String testDataFile) throws Exception {
|
|
File javaOut = CodegenTestUtil.compileJava(
|
|
Collections.singletonList(KotlinTestUtils.getTestDataPathBase() + "/codegen/" + getPrefix() + "/" + testDataFile + ".java"),
|
|
Collections.<String>emptyList(),
|
|
Collections.<String>emptyList()
|
|
);
|
|
|
|
String javaClassPath = javaClassName.replace('.', File.separatorChar) + ".class";
|
|
|
|
ClassReader javaReader = new ClassReader(FilesKt.readBytes(new File(javaOut, javaClassPath)));
|
|
ClassReader kotlinReader = getKotlinClassReader(classFqName.replace('.', '/').replace("$", "\\$"), testDataFile);
|
|
|
|
checkInfo(kotlinReader, javaReader);
|
|
}
|
|
|
|
private void doCustomTest(
|
|
@Language("RegExp") @NotNull String internalNameRegexp,
|
|
@NotNull OuterClassInfo expectedInfo,
|
|
@NotNull String testDataFile
|
|
) {
|
|
ClassReader kotlinReader = getKotlinClassReader(internalNameRegexp, testDataFile);
|
|
OuterClassInfo kotlinInfo = readOuterClassInfo(kotlinReader);
|
|
String message = "Error in enclosingMethodInfo info for class: " + kotlinReader.getClassName();
|
|
if (kotlinInfo == null) {
|
|
assertNull(expectedInfo.getOwner());
|
|
}
|
|
else {
|
|
assertTrue(message + "\n" + kotlinInfo.getOwner() + " doesn't start with " + expectedInfo.getOwner(),
|
|
kotlinInfo.getOwner().startsWith(expectedInfo.getOwner()));
|
|
}
|
|
assertEquals(message, expectedInfo.getMethodName(), kotlinInfo.getMethodName());
|
|
assertEquals(message, expectedInfo.getMethodDesc(), kotlinInfo.getMethodDesc());
|
|
}
|
|
|
|
@NotNull
|
|
private ClassReader getKotlinClassReader(@Language("RegExp") @NotNull String internalNameRegexp, @NotNull String testDataFile) {
|
|
loadFile(getPrefix() + "/" + testDataFile + ".kt");
|
|
OutputFileCollection outputFiles = generateClassesInFile();
|
|
for (OutputFile file : outputFiles.asList()) {
|
|
if (file.getRelativePath().matches(internalNameRegexp + "\\.class")) {
|
|
return new ClassReader(file.asByteArray());
|
|
}
|
|
}
|
|
throw new AssertionError(
|
|
"Couldn't find class by regexp: " + internalNameRegexp + " in:\n" + StringsKt.join(outputFiles.asList(), "\n")
|
|
);
|
|
}
|
|
|
|
private static void checkInfo(@NotNull ClassReader kotlinReader, @NotNull ClassReader javaReader) {
|
|
OuterClassInfo kotlinInfo = readOuterClassInfo(kotlinReader);
|
|
OuterClassInfo javaInfo = readOuterClassInfo(javaReader);
|
|
compareInfo(kotlinReader.getClassName(), kotlinInfo, javaInfo);
|
|
}
|
|
|
|
private static void compareInfo(
|
|
@NotNull String kotlinClassName,
|
|
@Nullable OuterClassInfo kotlinInfo,
|
|
@Nullable OuterClassInfo expectedJavaInfo
|
|
) {
|
|
assertEquals("Error in enclosingMethodInfo info for: " + kotlinClassName + " class", expectedJavaInfo, kotlinInfo);
|
|
}
|
|
|
|
@Nullable
|
|
private static OuterClassInfo readOuterClassInfo(@NotNull ClassReader reader) {
|
|
final Ref<OuterClassInfo> info = Ref.create();
|
|
reader.accept(new ClassVisitor(Opcodes.ASM5) {
|
|
@Override
|
|
public void visitOuterClass(@NotNull String owner, @Nullable String name, @Nullable String desc) {
|
|
info.set(new OuterClassInfo(owner, name, desc));
|
|
}
|
|
}, 0);
|
|
return info.get();
|
|
}
|
|
}
|