diff --git a/testable-core/src/main/java/com/alibaba/testable/core/annotation/MockMethod.java b/testable-core/src/main/java/com/alibaba/testable/core/annotation/MockMethod.java index 108848a..7733ace 100644 --- a/testable-core/src/main/java/com/alibaba/testable/core/annotation/MockMethod.java +++ b/testable-core/src/main/java/com/alibaba/testable/core/annotation/MockMethod.java @@ -1,5 +1,6 @@ package com.alibaba.testable.core.annotation; +import javax.lang.model.type.NullType; import java.lang.annotation.*; /** @@ -18,4 +19,10 @@ public @interface MockMethod { */ String targetMethod() default ""; + /** + * explicit set target class instead of adding to parameter list + * @return target class type + */ + Class targetClass() default NullType.class; + }