diff options
Diffstat (limited to 'plugins/java-decompiler/engine/test/org/jetbrains/java')
3 files changed, 323 insertions, 0 deletions
diff --git a/plugins/java-decompiler/engine/test/org/jetbrains/java/decompiler/BulkDecompilationTest.java b/plugins/java-decompiler/engine/test/org/jetbrains/java/decompiler/BulkDecompilationTest.java new file mode 100644 index 000000000000..1b5353b37a49 --- /dev/null +++ b/plugins/java-decompiler/engine/test/org/jetbrains/java/decompiler/BulkDecompilationTest.java @@ -0,0 +1,110 @@ +/* + * Copyright 2000-2014 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.java.decompiler; + +import org.hamcrest.Matchers; +import org.jetbrains.java.decompiler.main.decompiler.ConsoleDecompiler; +import org.jetbrains.java.decompiler.util.InterpreterUtil; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; + +import java.io.*; +import java.util.Enumeration; +import java.util.zip.ZipEntry; +import java.util.zip.ZipFile; + +import static org.junit.Assert.assertThat; +import static org.junit.Assert.assertTrue; + +public class BulkDecompilationTest { + private DecompilerTestFixture fixture; + + @Before + public void setUp() throws IOException { + fixture = new DecompilerTestFixture(); + fixture.setUp(); + } + + @After + public void tearDown() { + fixture.tearDown(); + fixture = null; + } + + @Test + public void testDirectory() { + File classes = new File(fixture.getTempDir(), "classes"); + unpack(new File(fixture.getTestDataDir(), "bulk.jar"), classes); + + ConsoleDecompiler decompiler = fixture.getDecompiler(); + decompiler.addSpace(classes, true); + decompiler.decompileContext(); + + compareDirectories(new File(fixture.getTestDataDir(), "bulk"), fixture.getTargetDir()); + } + + @Test + public void testJar() { + ConsoleDecompiler decompiler = fixture.getDecompiler(); + decompiler.addSpace(new File(fixture.getTestDataDir(), "bulk.jar"), true); + decompiler.decompileContext(); + + File unpacked = new File(fixture.getTempDir(), "unpacked"); + unpack(new File(fixture.getTargetDir(), "bulk.jar"), unpacked); + + compareDirectories(new File(fixture.getTestDataDir(), "bulk"), unpacked); + } + + private static void unpack(File archive, File targetDir) { + try { + ZipFile zip = new ZipFile(archive); + try { + Enumeration<? extends ZipEntry> entries = zip.entries(); + while (entries.hasMoreElements()) { + ZipEntry entry = entries.nextElement(); + if (!entry.isDirectory()) { + File file = new File(targetDir, entry.getName()); + assertTrue(file.getParentFile().mkdirs() || file.getParentFile().isDirectory()); + InputStream in = zip.getInputStream(entry); + OutputStream out = new FileOutputStream(file); + InterpreterUtil.copyStream(in, out); + out.close(); + in.close(); + } + } + } + finally { + zip.close(); + } + } + catch (IOException e) { + throw new RuntimeException(e); + } + } + + private static void compareDirectories(File expected, File actual) { + String[] expectedList = expected.list(); + String[] actualList = actual.list(); + assertThat(actualList, Matchers.arrayContainingInAnyOrder(expectedList)); + for (String name : expectedList) { + File child = new File(expected, name); + if (child.isDirectory()) { + compareDirectories(child, new File(actual, name)); + } + } + } +} diff --git a/plugins/java-decompiler/engine/test/org/jetbrains/java/decompiler/DecompilerTestFixture.java b/plugins/java-decompiler/engine/test/org/jetbrains/java/decompiler/DecompilerTestFixture.java new file mode 100644 index 000000000000..b82b65e2155b --- /dev/null +++ b/plugins/java-decompiler/engine/test/org/jetbrains/java/decompiler/DecompilerTestFixture.java @@ -0,0 +1,88 @@ +/* + * Copyright 2000-2014 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.java.decompiler; + +import org.jetbrains.java.decompiler.main.decompiler.ConsoleDecompiler; +import org.jetbrains.java.decompiler.main.extern.IFernflowerPreferences; + +import java.io.*; +import java.util.HashMap; + +import static org.junit.Assert.assertTrue; + +public class DecompilerTestFixture { + private File testDataDir; + private File tempDir; + private File targetDir; + private ConsoleDecompiler decompiler; + + public void setUp() throws IOException { + testDataDir = new File("testData"); + if (!isTestDataDir(testDataDir)) testDataDir = new File("community/plugins/java-decompiler/engine/testData"); + if (!isTestDataDir(testDataDir)) testDataDir = new File("plugins/java-decompiler/engine/testData"); + assertTrue("current dir: " + new File("").getAbsolutePath(), isTestDataDir(testDataDir)); + + //noinspection SSBasedInspection + tempDir = File.createTempFile("decompiler_test_", "_dir"); + assertTrue(tempDir.delete()); + + targetDir = new File(tempDir, "decompiled"); + assertTrue(targetDir.mkdirs()); + decompiler = new ConsoleDecompiler(this.targetDir, new HashMap<String, Object>() {{ + put(IFernflowerPreferences.LOG_LEVEL, "warn"); + put(IFernflowerPreferences.DECOMPILE_GENERIC_SIGNATURES, "1"); + put(IFernflowerPreferences.REMOVE_SYNTHETIC, "1"); + put(IFernflowerPreferences.REMOVE_BRIDGE, "1"); + put(IFernflowerPreferences.LITERALS_AS_IS, "1"); + }}); + } + + public void tearDown() { + if (tempDir != null) { + delete(tempDir); + } + } + + public File getTestDataDir() { + return testDataDir; + } + + public File getTempDir() { + return tempDir; + } + + public File getTargetDir() { + return targetDir; + } + + public ConsoleDecompiler getDecompiler() { + return decompiler; + } + + private static boolean isTestDataDir(File dir) { + return dir.isDirectory() && new File(dir, "classes").isDirectory() && new File(dir, "results").isDirectory(); + } + + private static void delete(File file) { + if (file.isDirectory()) { + File[] files = file.listFiles(); + if (files != null) { + for (File f : files) delete(f); + } + } + assertTrue(file.delete()); + } +} diff --git a/plugins/java-decompiler/engine/test/org/jetbrains/java/decompiler/SingleClassesTest.java b/plugins/java-decompiler/engine/test/org/jetbrains/java/decompiler/SingleClassesTest.java new file mode 100644 index 000000000000..70aa605e8630 --- /dev/null +++ b/plugins/java-decompiler/engine/test/org/jetbrains/java/decompiler/SingleClassesTest.java @@ -0,0 +1,125 @@ +/* + * Copyright 2000-2014 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.java.decompiler; + +import org.jetbrains.java.decompiler.main.decompiler.ConsoleDecompiler; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; + +import java.io.*; +import java.util.ArrayList; +import java.util.Collections; +import java.util.List; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertTrue; + +public class SingleClassesTest { + private DecompilerTestFixture fixture; + + @Before + public void setUp() throws IOException { + fixture = new DecompilerTestFixture(); + fixture.setUp(); + } + + @After + public void tearDown() { + fixture.tearDown(); + fixture = null; + } + + @Test public void testClassFields() { doTest("pkg/TestClassFields"); } + @Test public void testClassLambda() { doTest("pkg/TestClassLambda"); } + @Test public void testClassLoop() { doTest("pkg/TestClassLoop"); } + @Test public void testClassSwitch() { doTest("pkg/TestClassSwitch"); } + @Test public void testClassTypes() { doTest("pkg/TestClassTypes"); } + @Test public void testClassVar() { doTest("pkg/TestClassVar"); } + @Test public void testClassNestedInitializer() { doTest("pkg/TestClassNestedInitializer"); } + @Test public void testClassCast() { doTest("pkg/TestClassCast"); } + @Test public void testDeprecations() { doTest("pkg/TestDeprecations"); } + @Test public void testExtendsList() { doTest("pkg/TestExtendsList"); } + @Test public void testMethodParameters() { doTest("pkg/TestMethodParameters"); } + @Test public void testCodeConstructs() { doTest("pkg/TestCodeConstructs"); } + @Test public void testConstants() { doTest("pkg/TestConstants"); } + @Test public void testEnum() { doTest("pkg/TestEnum"); } + @Test public void testDebugSymbols() { doTest("pkg/TestDebugSymbols"); } + @Test public void testInvalidMethodSignature() { doTest("InvalidMethodSignature"); } + + private void doTest(String testFile) { + try { + File classFile = new File(fixture.getTestDataDir(), "/classes/" + testFile + ".class"); + assertTrue(classFile.isFile()); + String testName = classFile.getName().replace(".class", ""); + + ConsoleDecompiler decompiler = fixture.getDecompiler(); + for (File file : collectClasses(classFile)) { + decompiler.addSpace(file, true); + } + + decompiler.decompileContext(); + + File decompiledFile = new File(fixture.getTargetDir(), testName + ".java"); + assertTrue(decompiledFile.isFile()); + + File referenceFile = new File(fixture.getTestDataDir(), "results/" + testName + ".dec"); + assertTrue(referenceFile.isFile()); + + String decompiledContent = getContent(decompiledFile); + String referenceContent = getContent(referenceFile); + assertEquals(referenceContent, decompiledContent); + } + catch (Exception e) { + throw new RuntimeException(e); + } + } + + private static List<File> collectClasses(File classFile) { + List<File> files = new ArrayList<File>(); + files.add(classFile); + + File parent = classFile.getParentFile(); + if (parent != null) { + final String pattern = classFile.getName().replace(".class", "") + "\\$.+\\.class"; + File[] inner = parent.listFiles(new FilenameFilter() { + @Override + public boolean accept(File dir, String name) { + return name.matches(pattern); + } + }); + if (inner != null) Collections.addAll(files, inner); + } + + return files; + } + + private static String getContent(File file) throws IOException { + Reader reader = new InputStreamReader(new FileInputStream(file), "UTF-8"); + try { + char[] buff = new char[16 * 1024]; + StringBuilder content = new StringBuilder(); + int n; + while ((n = reader.read(buff)) > 0) { + content.append(buff, 0, n); + } + return content.toString(); + } + finally { + reader.close(); + } + } +} |