[Tests] Introduce transformers functions object

2021-12-06 20:27:22 +03:00
parent af73034235
commit 2874462e3a
12 changed files with 54 additions and 15 deletions
@@ -124,7 +124,12 @@ private class TestGeneratorImplInstance(
        p.println("import " + KtTestUtil::class.java.canonicalName + ";")

        for (clazz in testClassModels.flatMapTo(mutableSetOf()) { classModel -> classModel.imports }) {
-            p.println("import ${clazz.name};")
+            val realName = when (clazz) {
+                TransformingTestMethodModel.TransformerFunctionsClassPlaceHolder::class.java ->
+                    "org.jetbrains.kotlin.test.runners.TransformersFunctions"
+                else -> clazz.name
+            }
+            p.println("import $realName;")
        }

        if (suiteClassPackage != baseTestClassPackage) {