Refactor CompileJavaAgainstKotlinTest.
Avoid duplication.
This commit is contained in:
@@ -16,47 +16,34 @@
|
|||||||
|
|
||||||
package org.jetbrains.jet.jvm.compiler;
|
package org.jetbrains.jet.jvm.compiler;
|
||||||
|
|
||||||
import com.intellij.openapi.util.Disposer;
|
|
||||||
import com.intellij.openapi.util.io.FileUtil;
|
|
||||||
import com.intellij.openapi.vfs.CharsetToolkit;
|
|
||||||
import com.intellij.psi.PsiFileFactory;
|
|
||||||
import com.intellij.psi.impl.PsiFileFactoryImpl;
|
|
||||||
import com.intellij.testFramework.LightVirtualFile;
|
|
||||||
import junit.framework.Test;
|
import junit.framework.Test;
|
||||||
import org.jetbrains.annotations.NotNull;
|
import org.jetbrains.annotations.NotNull;
|
||||||
import org.jetbrains.jet.JetTestCaseBuilder;
|
import org.jetbrains.jet.JetTestCaseBuilder;
|
||||||
import org.jetbrains.jet.JetTestUtils;
|
|
||||||
import org.jetbrains.jet.cli.jvm.compiler.CompileEnvironmentUtil;
|
|
||||||
import org.jetbrains.jet.cli.jvm.compiler.JetCoreEnvironment;
|
|
||||||
import org.jetbrains.jet.codegen.*;
|
|
||||||
import org.jetbrains.jet.lang.psi.JetFile;
|
|
||||||
import org.jetbrains.jet.plugin.JetLanguage;
|
|
||||||
import org.jetbrains.jet.test.TestCaseWithTmpdir;
|
import org.jetbrains.jet.test.TestCaseWithTmpdir;
|
||||||
import org.junit.Assert;
|
import org.junit.Assert;
|
||||||
|
|
||||||
import javax.tools.JavaCompiler;
|
|
||||||
import javax.tools.JavaFileObject;
|
|
||||||
import javax.tools.StandardJavaFileManager;
|
|
||||||
import javax.tools.ToolProvider;
|
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
import java.nio.charset.Charset;
|
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Locale;
|
|
||||||
|
import static org.jetbrains.jet.jvm.compiler.LoadDescriptorUtil.compileJavaToDir;
|
||||||
|
import static org.jetbrains.jet.jvm.compiler.LoadDescriptorUtil.compileKotlinToDirAndGetAnalyzeExhaust;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @author Stepan Koltsov
|
* @author Stepan Koltsov
|
||||||
*
|
*
|
||||||
* @see WriteSignatureTest
|
* @see WriteSignatureTest
|
||||||
*/
|
*/
|
||||||
|
@SuppressWarnings({"JUnitTestCaseWithNonTrivialConstructors", "JUnitTestCaseWithNoTests"})
|
||||||
public class CompileJavaAgainstKotlinTest extends TestCaseWithTmpdir {
|
public class CompileJavaAgainstKotlinTest extends TestCaseWithTmpdir {
|
||||||
|
|
||||||
|
@NotNull
|
||||||
private final File ktFile;
|
private final File ktFile;
|
||||||
|
@NotNull
|
||||||
private final File javaFile;
|
private final File javaFile;
|
||||||
private JetCoreEnvironment jetCoreEnvironment;
|
|
||||||
|
|
||||||
public CompileJavaAgainstKotlinTest(File ktFile) {
|
public CompileJavaAgainstKotlinTest(@NotNull File ktFile) {
|
||||||
this.ktFile = ktFile;
|
this.ktFile = ktFile;
|
||||||
Assert.assertTrue(ktFile.getName().endsWith(".kt"));
|
Assert.assertTrue(ktFile.getName().endsWith(".kt"));
|
||||||
this.javaFile = new File(ktFile.getPath().replaceFirst("\\.kt", ".java"));
|
this.javaFile = new File(ktFile.getPath().replaceFirst("\\.kt", ".java"));
|
||||||
@@ -69,33 +56,13 @@ public class CompileJavaAgainstKotlinTest extends TestCaseWithTmpdir {
|
|||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void runTest() throws Throwable {
|
protected void runTest() throws Throwable {
|
||||||
jetCoreEnvironment = JetTestUtils.createEnvironmentWithMockJdkAndIdeaAnnotations(myTestRootDisposable);
|
compileKotlinToDirAndGetAnalyzeExhaust(ktFile, tmpdir, getTestRootDisposable());
|
||||||
|
|
||||||
String text = FileUtil.loadFile(ktFile);
|
List<String> options = Arrays.asList(
|
||||||
|
"-classpath", tmpdir.getPath() + System.getProperty("path.separator") + "out/production/stdlib",
|
||||||
JetFile psiFile = JetTestUtils.createFile(ktFile.getName(), text, jetCoreEnvironment.getProject());
|
"-d", tmpdir.getPath()
|
||||||
|
);
|
||||||
ClassFileFactory classFileFactory = GenerationUtils.compileFileGetClassFileFactoryForTest(psiFile);
|
compileJavaToDir(Collections.singleton(javaFile), options);
|
||||||
|
|
||||||
CompileEnvironmentUtil.writeToOutputDirectory(classFileFactory, tmpdir);
|
|
||||||
|
|
||||||
Disposer.dispose(myTestRootDisposable);
|
|
||||||
|
|
||||||
JavaCompiler javaCompiler = ToolProvider.getSystemJavaCompiler();
|
|
||||||
|
|
||||||
StandardJavaFileManager fileManager = javaCompiler.getStandardFileManager(null, Locale.ENGLISH, Charset.forName("utf-8"));
|
|
||||||
try {
|
|
||||||
Iterable<? extends JavaFileObject> javaFileObjectsFromFiles = fileManager.getJavaFileObjectsFromFiles(Collections.singleton(javaFile));
|
|
||||||
List<String> options = Arrays.asList(
|
|
||||||
"-classpath", tmpdir.getPath() + System.getProperty("path.separator") + "out/production/stdlib",
|
|
||||||
"-d", tmpdir.getPath()
|
|
||||||
);
|
|
||||||
JavaCompiler.CompilationTask task = javaCompiler.getTask(null, fileManager, null, options, null, javaFileObjectsFromFiles);
|
|
||||||
|
|
||||||
Assert.assertTrue(task.call());
|
|
||||||
} finally {
|
|
||||||
fileManager.close();
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public static Test suite() {
|
public static Test suite() {
|
||||||
|
|||||||
@@ -114,20 +114,23 @@ public final class LoadDescriptorUtil {
|
|||||||
@NotNull Disposable disposable
|
@NotNull Disposable disposable
|
||||||
)
|
)
|
||||||
throws IOException {
|
throws IOException {
|
||||||
compileJavaToDir(javaFiles, outDir);
|
compileJavaWithAnnotationsJar(javaFiles, outDir);
|
||||||
return extractTestNamespaceFromBinaries(outDir, disposable);
|
return extractTestNamespaceFromBinaries(outDir, disposable);
|
||||||
}
|
}
|
||||||
|
|
||||||
private static void compileJavaToDir(@NotNull Collection<File> javaFiles, @NotNull File ourDir) throws IOException {
|
private static void compileJavaWithAnnotationsJar(@NotNull Collection<File> javaFiles, @NotNull File outDir) throws IOException {
|
||||||
|
compileJavaToDir(javaFiles, Arrays.asList(
|
||||||
|
"-classpath", "out/production/runtime" + File.pathSeparator + JetTestUtils.getAnnotationsJar().getPath(),
|
||||||
|
"-d", outDir.getPath()
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void compileJavaToDir(@NotNull Collection<File> javaFiles, @NotNull List<String> options) throws IOException {
|
||||||
JavaCompiler javaCompiler = ToolProvider.getSystemJavaCompiler();
|
JavaCompiler javaCompiler = ToolProvider.getSystemJavaCompiler();
|
||||||
|
|
||||||
StandardJavaFileManager fileManager = javaCompiler.getStandardFileManager(null, Locale.ENGLISH, Charset.forName("utf-8"));
|
StandardJavaFileManager fileManager = javaCompiler.getStandardFileManager(null, Locale.ENGLISH, Charset.forName("utf-8"));
|
||||||
try {
|
try {
|
||||||
Iterable<? extends JavaFileObject> javaFileObjectsFromFiles = fileManager.getJavaFileObjectsFromFiles(javaFiles);
|
Iterable<? extends JavaFileObject> javaFileObjectsFromFiles = fileManager.getJavaFileObjectsFromFiles(javaFiles);
|
||||||
List<String> options = Arrays.asList(
|
|
||||||
"-classpath", "out/production/runtime" + File.pathSeparator + JetTestUtils.getAnnotationsJar().getPath(),
|
|
||||||
"-d", ourDir.getPath()
|
|
||||||
);
|
|
||||||
JavaCompiler.CompilationTask task = javaCompiler.getTask(null, fileManager, null, options, null, javaFileObjectsFromFiles);
|
JavaCompiler.CompilationTask task = javaCompiler.getTask(null, fileManager, null, options, null, javaFileObjectsFromFiles);
|
||||||
|
|
||||||
Assert.assertTrue(task.call());
|
Assert.assertTrue(task.call());
|
||||||
|
|||||||
Reference in New Issue
Block a user