IR to CFG test generator, sequential function tests

compiler/tests/org/jetbrains/kotlin/ir/AbstractIrCfgTestCase.kt

@@ -0,0 +1,63 @@
+/*
+ * Copyright 2010-2016 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.kotlin.ir
+
+import org.jetbrains.kotlin.ir.declarations.IrFile
+import org.jetbrains.kotlin.ir.declarations.IrFunction
+import org.jetbrains.kotlin.ir.declarations.IrModuleFragment
+import org.jetbrains.kotlin.ir.declarations.name
+import org.jetbrains.kotlin.ir2cfg.generators.FunctionGenerator
+import org.jetbrains.kotlin.ir2cfg.util.dump
+import org.jetbrains.kotlin.test.KotlinTestUtils
+import java.io.File
+
+abstract class AbstractIrCfgTestCase : AbstractIrGeneratorTestCase() {
+
+    private val IrFunction.name: String get() = this.descriptor.name.asString()
+
+    private fun IrFile.cfgDump(): String {
+        val builder = StringBuilder()
+        for (declaration in this.declarations) {
+            if (declaration is IrFunction) {
+                builder.appendln("// FUN: ${declaration.name}")
+                val cfg = FunctionGenerator(declaration).generate()
+                builder.appendln(cfg.dump())
+                builder.appendln("// END FUN: ${declaration.name}")
+            }
+        }
+        return builder.toString()
+    }
+
+    private fun IrModuleFragment.cfgDump(): String {
+        val builder = StringBuilder()
+        for (file in this.files) {
+            builder.appendln("// FILE: ${file.name}")
+            builder.appendln(file.cfgDump())
+            builder.appendln("// END FILE: ${file.name}")
+            builder.appendln()
+        }
+        return builder.toString()
+    }
+
+    override fun doTest(wholeFile: File, testFiles: List<TestFile>) {
+        val irModule = generateIrModule(false)
+        val irModuleDump = irModule.cfgDump()
+        val expectedPath = wholeFile.canonicalPath.replace(".kt", ".txt")
+        val expectedFile = File(expectedPath)
+        KotlinTestUtils.assertEqualsToFile(expectedFile, irModuleDump)
+    }
+}

compiler/tests/org/jetbrains/kotlin/ir/IrCfgTestCaseGenerated.java

@@ -0,0 +1,73 @@
+/*
+ * Copyright 2010-2016 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.kotlin.ir;
+
+import com.intellij.testFramework.TestDataPath;
+import org.jetbrains.kotlin.test.JUnit3RunnerWithInners;
+import org.jetbrains.kotlin.test.KotlinTestUtils;
+import org.jetbrains.kotlin.test.TestMetadata;
+import org.junit.runner.RunWith;
+
+import java.io.File;
+import java.util.regex.Pattern;
+
+/** This class is generated by {@link org.jetbrains.kotlin.generators.tests.TestsPackage}. DO NOT MODIFY MANUALLY */
+@SuppressWarnings("all")
+@TestMetadata("compiler/testData/ir/irCfg")
+@TestDataPath("$PROJECT_ROOT")
+@RunWith(JUnit3RunnerWithInners.class)
+public class IrCfgTestCaseGenerated extends AbstractIrCfgTestCase {
+    public void testAllFilesPresentInIrCfg() throws Exception {
+        KotlinTestUtils.assertAllTestsPresentByMetadata(this.getClass(), new File("compiler/testData/ir/irCfg"), Pattern.compile("^(.+)\\.kt$"), true);
+    }
+
+    @TestMetadata("expressionFun.kt")
+    public void testExpressionFun() throws Exception {
+        String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/ir/irCfg/expressionFun.kt");
+        doTest(fileName);
+    }
+
+    @TestMetadata("expressionUnit.kt")
+    public void testExpressionUnit() throws Exception {
+        String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/ir/irCfg/expressionUnit.kt");
+        doTest(fileName);
+    }
+
+    @TestMetadata("returnUnit.kt")
+    public void testReturnUnit() throws Exception {
+        String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/ir/irCfg/returnUnit.kt");
+        doTest(fileName);
+    }
+
+    @TestMetadata("sequentialFun.kt")
+    public void testSequentialFun() throws Exception {
+        String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/ir/irCfg/sequentialFun.kt");
+        doTest(fileName);
+    }
+
+    @TestMetadata("simpleFun.kt")
+    public void testSimpleFun() throws Exception {
+        String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/ir/irCfg/simpleFun.kt");
+        doTest(fileName);
+    }
+
+    @TestMetadata("simpleReturn.kt")
+    public void testSimpleReturn() throws Exception {
+        String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/ir/irCfg/simpleReturn.kt");
+        doTest(fileName);
+    }
+}