FIR: add transformer consistency test

2018-03-16 18:33:43 +03:00
parent cdc536b147
commit 95af0268b8
2 changed files with 52 additions and 1 deletions
@@ -46,7 +46,7 @@ class RawFirBuilderTotalKotlinTestCase : AbstractRawFirBuilderTestCase() {
        println("TIME PER FILE: ${(time / counter) * 1e-6} ms, COUNTER: $counter")
    }

-    fun testConsistency() {
+    fun testVisitConsistency() {
        val root = File(testDataPath)
        for (file in root.walkTopDown()) {
            if (file.isDirectory) continue
@@ -63,4 +63,20 @@ class RawFirBuilderTotalKotlinTestCase : AbstractRawFirBuilderTestCase() {
        }
    }

+    fun testTransformConsistency() {
+        val root = File(testDataPath)
+        for (file in root.walkTopDown()) {
+            if (file.isDirectory) continue
+            if (file.path.contains("testData") || file.path.contains("resources")) continue
+            if (file.extension != "kt") continue
+            val ktFile = createKtFile(file.toRelativeString(root))
+            val firFile = ktFile.toFirFile()
+            try {
+                firFile.checkTransformedChildren()
+            } catch (e: Throwable) {
+                println("EXCEPTION in: " + file.toRelativeString(root))
+                throw e
+            }
+        }
+    }
 }