diff options
Diffstat (limited to 'test/org/jetbrains')
-rw-r--r-- | test/org/jetbrains/java/decompiler/SingleClassesTest.java | 124 |
1 files changed, 124 insertions, 0 deletions
diff --git a/test/org/jetbrains/java/decompiler/SingleClassesTest.java b/test/org/jetbrains/java/decompiler/SingleClassesTest.java new file mode 100644 index 0000000..1e37d97 --- /dev/null +++ b/test/org/jetbrains/java/decompiler/SingleClassesTest.java @@ -0,0 +1,124 @@ +/* + * 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 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 tempDir; + private ConsoleDecompiler decompiler; + + @Before + public void setUp() throws IOException { + tempDir = File.createTempFile("decompiler_test_", "_dir"); + assertTrue(tempDir.delete()); + assertTrue(tempDir.mkdirs()); + decompiler = new ConsoleDecompiler(new HashMap<String, Object>() {{ + boolean win = System.getProperty("os.name", "").startsWith("Windows"); + put(IFernflowerPreferences.LOG_LEVEL, "warn"); + put(IFernflowerPreferences.RENAME_ENTITIES, "1"); + put(IFernflowerPreferences.HIDE_DEFAULT_CONSTRUCTOR, "1"); + put(IFernflowerPreferences.DECOMPILE_GENERIC_SIGNATURES, "0"); + put(IFernflowerPreferences.IDEA_NOT_NULL_ANNOTATION, "1"); + put(IFernflowerPreferences.LAMBDA_TO_ANONYMOUS_CLASS, "0"); + put(IFernflowerPreferences.USE_DEBUG_VARNAMES, "0"); + put(IFernflowerPreferences.NEW_LINE_SEPARATOR, (win ? "0" : "1")); + }}); + } + + @After + public void tearDown() { + decompiler = null; + delete(tempDir); + tempDir = null; + } + + @Test public void testClassFields() { doTest("TestClassFields"); } + @Test public void testClassLambda() { doTest("TestClassLambda"); } + @Test public void testClassLoop() { doTest("TestClassLoop"); } + @Test public void testClassSwitch() { doTest("TestClassSwitch"); } + @Test public void testClassTypes() { doTest("TestClassTypes"); } + @Test public void testClassVar() { doTest("TestClassVar"); } + + private void doTest(String testName) { + try { + File 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)); + + File classFile = new File(testDataDir, "/classes/pkg/" + testName + ".class"); + assertTrue(classFile.isFile()); + decompiler.addSpace(classFile, true); + decompiler.decompileContext(tempDir); + + File decompiledFile = new File(tempDir, testName + ".java"); + assertTrue(decompiledFile.isFile()); + + File referenceFile = new File(testDataDir, "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 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 { + 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(); + } + } + + 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()); + } + } +} |