diff options
Diffstat (limited to 'src/main/java/org/mockito/internal/creation/SuspendMethod.java')
-rw-r--r-- | src/main/java/org/mockito/internal/creation/SuspendMethod.java | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/src/main/java/org/mockito/internal/creation/SuspendMethod.java b/src/main/java/org/mockito/internal/creation/SuspendMethod.java index 018bc50..42ceac6 100644 --- a/src/main/java/org/mockito/internal/creation/SuspendMethod.java +++ b/src/main/java/org/mockito/internal/creation/SuspendMethod.java @@ -11,12 +11,18 @@ import java.util.Arrays; * See <a href="https://github.com/Kotlin/kotlin-coroutines/blob/master/kotlin-coroutines-informal.md#continuation-passing-style">Design docs for details</a>. */ public class SuspendMethod { - private static final String KOTLIN_CONTINUATION = "kotlin.coroutines.experimental.Continuation"; + private static final String KOTLIN_EXPERIMENTAL_CONTINUATION = "kotlin.coroutines.experimental.Continuation"; + private static final String KOTLIN_CONTINUATION = "kotlin.coroutines.Continuation"; public static Class<?>[] trimSuspendParameterTypes(Class<?>[] parameterTypes) { int n = parameterTypes.length; - if (n > 0 && parameterTypes[n - 1].getName().equals(KOTLIN_CONTINUATION)) + if (n > 0 && isContinuationType(parameterTypes[n - 1])) return Arrays.copyOf(parameterTypes, n - 1); return parameterTypes; } + + private static boolean isContinuationType(Class<?> parameterType) { + String name = parameterType.getName(); + return name.equals(KOTLIN_CONTINUATION) || name.equals(KOTLIN_EXPERIMENTAL_CONTINUATION); + } } |