Support vararg parameters (fixes #54)

kosi-libs · May 2, 2023 · 087f9a0 · 087f9a0
1 parent 00b4f62
commit 087f9a0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/mockmp-processor/src/main/kotlin/org/kodein/mock/ksp/MocKMPProcessor.kt b/mockmp-processor/src/main/kotlin/org/kodein/mock/ksp/MocKMPProcessor.kt
@@ -245,9 +245,14 @@ public class MocKMPProcessor(
                         gFun.addAnnotation(it.toAnnotationSpec())
                     }
                     vFun.parameters.forEach { vParam ->
-                        gFun.addParameter(vParam.name!!.asString(), vParam.type.toTypeName(typeParamResolver))
+                        gFun.addParameter(vParam.name!!.asString(), vParam.type.toTypeName(typeParamResolver), if (vParam.isVararg) listOf(KModifier.VARARG) else emptyList())
+                    }
+                    val paramsDescription = vFun.parameters.joinToString {
+                        (it.type.resolve().declaration as? KSClassDeclaration)?.qualifiedName?.asString()?.let { str ->
+                            if (it.isVararg) "vararg $str" else str
+                        }
+                            ?: "?"
                     }
-                    val paramsDescription = vFun.parameters.joinToString { (it.type.resolve().declaration as? KSClassDeclaration)?.qualifiedName?.asString() ?: "?" }
                     val paramsCall = if (vFun.parameters.isEmpty()) "" else vFun.parameters.joinToString { it.name!!.asString() }
                     val register = if (Modifier.SUSPEND in vFun.modifiers) "registerSuspend" else "register"
                     val default = if (vFun.isAbstract) "" else "default = { super.${vFun.simpleName.asString()}($paramsCall) }"

diff --git a/tests/tests-junit4/src/commonMain/kotlin/foo/Types.kt b/tests/tests-junit4/src/commonMain/kotlin/foo/Types.kt
@@ -47,7 +47,7 @@ typealias BarCB = (String) -> Int
 interface Bar : Foo<Bar> {
     fun doNothing() {}
     fun doSomething() { doNothing() }
-    fun newData(string: String, int: Int): Data
+    fun newData(string: String, vararg int: Int): Data
     fun doData(data: Data)
     fun doAll(string: String, int: Int, data: Data)
     suspend fun newData(): Data