diff options
Diffstat (limited to 'test/org')
3 files changed, 205 insertions, 45 deletions
diff --git a/test/org/jetbrains/java/decompiler/BulkDecompilationTest.java b/test/org/jetbrains/java/decompiler/BulkDecompilationTest.java new file mode 100644 index 0000000..1b5353b --- /dev/null +++ b/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/test/org/jetbrains/java/decompiler/DecompilerTestFixture.java b/test/org/jetbrains/java/decompiler/DecompilerTestFixture.java new file mode 100644 index 0000000..f8fb9a5 --- /dev/null +++ b/test/org/jetbrains/java/decompiler/DecompilerTestFixture.java @@ -0,0 +1,86 @@ +/* + * 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(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() { + 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/test/org/jetbrains/java/decompiler/SingleClassesTest.java b/test/org/jetbrains/java/decompiler/SingleClassesTest.java index e251e91..9868011 100644 --- a/test/org/jetbrains/java/decompiler/SingleClassesTest.java +++ b/test/org/jetbrains/java/decompiler/SingleClassesTest.java @@ -16,49 +16,28 @@ package org.jetbrains.java.decompiler; import org.jetbrains.java.decompiler.main.decompiler.ConsoleDecompiler; -import org.jetbrains.java.decompiler.main.extern.IFernflowerPreferences; import org.junit.After; import org.junit.Before; import org.junit.Test; import java.io.*; -import java.util.HashMap; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertTrue; public class SingleClassesTest { - private File testDataDir; - private File tempDir; - private ConsoleDecompiler decompiler; + private DecompilerTestFixture fixture; @Before 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(isTestDataDir(testDataDir)); - - //noinspection SSBasedInspection - tempDir = File.createTempFile("decompiler_test_", "_dir"); - assertTrue(tempDir.delete()); - assertTrue(tempDir.mkdirs()); - - decompiler = new ConsoleDecompiler(tempDir, 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"); - }}); + fixture = new DecompilerTestFixture(); + fixture.setUp(); } @After public void tearDown() { - decompiler = null; - delete(tempDir); - tempDir = null; - testDataDir = null; + fixture.tearDown(); + fixture = null; } @Test public void testClassFields() { doTest("TestClassFields"); } @@ -77,9 +56,10 @@ public class SingleClassesTest { private void doTest(final String testName) { try { - File classFile = new File(testDataDir, "/classes/pkg/" + testName + ".class"); + File classFile = new File(fixture.getTestDataDir(), "/classes/pkg/" + testName + ".class"); assertTrue(classFile.isFile()); + ConsoleDecompiler decompiler = fixture.getDecompiler(); decompiler.addSpace(classFile, true); File[] innerClasses = classFile.getParentFile().listFiles(new FilenameFilter() { @Override @@ -93,10 +73,10 @@ public class SingleClassesTest { decompiler.decompileContext(); - File decompiledFile = new File(tempDir, testName + ".java"); + File decompiledFile = new File(fixture.getTargetDir(), testName + ".java"); assertTrue(decompiledFile.isFile()); - File referenceFile = new File(testDataDir, "results/" + testName + ".dec"); + File referenceFile = new File(fixture.getTestDataDir(), "results/" + testName + ".dec"); assertTrue(referenceFile.isFile()); String decompiledContent = getContent(decompiledFile); @@ -108,10 +88,6 @@ public class SingleClassesTest { } } - private static boolean isTestDataDir(File dir) { - return dir.isDirectory() && new File(dir, "classes").isDirectory() && new File(dir, "results").isDirectory(); - } - private static String getContent(File file) throws IOException { Reader reader = new InputStreamReader(new FileInputStream(file), "UTF-8"); try { @@ -127,16 +103,4 @@ public class SingleClassesTest { reader.close(); } } - - private static void delete(File file) { - if (file.isDirectory()) { - File[] files = file.listFiles(); - if (files != null) { - for (File f : files) delete(f); - } - } - else { - assertTrue(file.delete()); - } - } } |