summaryrefslogtreecommitdiff
path: root/src/main/java/org/mockito/internal/creation/SuspendMethod.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/org/mockito/internal/creation/SuspendMethod.java')
-rw-r--r--src/main/java/org/mockito/internal/creation/SuspendMethod.java10
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);
+ }
}