Test framework for instructions

2011-04-04 18:40:39 +04:00
parent 90b6e78f89
commit 1fc134072c
14 changed files with 389 additions and 149 deletions
@@ -0,0 +1,149 @@
+/*
+ * @author max
+ */
+package org.jetbrains.jet.cfg;
+
+import com.intellij.openapi.util.io.FileUtil;
+import junit.framework.AssertionFailedError;
+import junit.framework.Test;
+import junit.framework.TestSuite;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.jet.JetTestCaseBase;
+import org.jetbrains.jet.lang.ErrorHandler;
+import org.jetbrains.jet.lang.cfg.pseudocode.Instruction;
+import org.jetbrains.jet.lang.cfg.pseudocode.JetControlFlowDataTrace;
+import org.jetbrains.jet.lang.cfg.pseudocode.JetControlFlowDataTraceFactory;
+import org.jetbrains.jet.lang.cfg.pseudocode.Pseudocode;
+import org.jetbrains.jet.lang.psi.JetElement;
+import org.jetbrains.jet.lang.psi.JetFile;
+import org.jetbrains.jet.lang.psi.JetNamedDeclaration;
+import org.jetbrains.jet.lang.resolve.AnalyzingUtils;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.io.PrintStream;
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+public class JetControlFlowTest extends JetTestCaseBase {
+    static {
+        System.setProperty("idea.platform.prefix", "Idea");
+    }
+
+    public JetControlFlowTest(String dataPath, String name) {
+        super(dataPath, name);
+    }
+
+    @Override
+    protected void runTest() throws Throwable {
+        configureByFile(getTestFilePath());
+        JetFile file = (JetFile) getFile();
+
+        AnalyzingUtils.analyzeNamespace(file.getRootNamespace(), ErrorHandler.THROW_EXCEPTION, new JetControlFlowDataTraceFactory() {
+            @NotNull
+            @Override
+            public JetControlFlowDataTrace createTrace(JetElement element) {
+                return new JetControlFlowDataTrace() {
+                    private final Map<JetElement, Pseudocode> data = new LinkedHashMap<JetElement, Pseudocode>();
+
+                    @Override
+                    public void recordControlFlowData(@NotNull JetElement element, @NotNull Pseudocode pseudocode) {
+                        data.put(element, pseudocode);
+                    }
+
+                    @Override
+                    public void close() {
+                        try {
+                            try {
+                                processCFData(name);
+                            } catch (AssertionFailedError e) {
+                                dumpDot(name, data.values());
+                                throw e;
+                            }
+                        } catch (IOException e) {
+                            throw new RuntimeException(e);
+                        }
+                    }
+
+                    private void processCFData(String name) throws IOException {
+                        Collection<Pseudocode> pseudocodes = data.values();
+                        for (Pseudocode pseudocode : pseudocodes) {
+                            pseudocode.postProcess();
+                        }
+
+                        StringBuilder instructionDump = new StringBuilder();
+                        for (Pseudocode pseudocode : pseudocodes) {
+                            pseudocode.dumpInstructions(instructionDump);
+                            instructionDump.append("=====================\n");
+                        }
+
+                        String expectedInstructionsFileName = getTestDataPath() + "/" + getTestFilePath().replace(".jet", ".instructions");
+                        File expectedInstructionsFile = new File(expectedInstructionsFileName);
+                        if (!expectedInstructionsFile.exists()) {
+                            FileUtil.writeToFile(expectedInstructionsFile, instructionDump.toString());
+                            fail("No expected instructions for " + name + " generated result is written into " + expectedInstructionsFileName);
+                        }
+                        String expectedInstructions = FileUtil.loadFile(expectedInstructionsFile);
+
+                        assertEquals(expectedInstructions, instructionDump.toString());
+
+//                        StringBuilder graphDump = new StringBuilder();
+//                        for (Pseudocode pseudocode : pseudocodes) {
+//                            topOrderDump(pseudocode.)
+//                        }
+                    }
+                };
+            }
+        });
+    }
+
+    private void dumpDot(String name, Collection<Pseudocode> pseudocodes) throws FileNotFoundException {
+        String graphFileName = getTestDataPath() + "/" + getTestFilePath().replace(".jet", ".dot");
+        File target = new File(graphFileName);
+
+        PrintStream out = new PrintStream(target);
+
+        out.println("digraph " + name + " {");
+        int[] count = new int[1];
+        Map<Instruction, String> nodeToName = new HashMap<Instruction, String>();
+        for (Pseudocode pseudocode : pseudocodes) {
+            pseudocode.dumpNodes(out, count, nodeToName);
+        }
+        int i = 0;
+        for (Pseudocode pseudocode : pseudocodes) {
+            String label;
+            JetElement correspondingElement = pseudocode.getCorrespondingElement();
+            if (correspondingElement instanceof JetNamedDeclaration) {
+                JetNamedDeclaration namedDeclaration = (JetNamedDeclaration) correspondingElement;
+                label = namedDeclaration.getName();
+            }
+            else {
+                label = "anonymous_" + i;
+            }
+            out.println("subgraph cluster_" + i + " {\n" +
+                        "label=\"" + label + "\";\n" +
+                        "color=blue;\n");
+            pseudocode.dumpEdges(out, count, nodeToName);
+            out.println("}");
+            i++;
+        }
+        out.println("}");
+        out.close();
+    }
+
+    public static TestSuite suite() {
+        TestSuite suite = new TestSuite();
+        suite.addTest(JetTestCaseBase.suiteForDirectory(getTestDataPathBase(), "/cfg/", true, new JetTestCaseBase.NamedTestFactory() {
+            @NotNull
+            @Override
+            public Test createTest(@NotNull String dataPath, @NotNull String name) {
+                return new JetControlFlowTest(dataPath, name);
+            }
+        }));
+        return suite;
+    }
+
+}