我想捕获对模拟对象的调用
public interface Service {
public String stringify(Object o);
}
service = mockery.mock(Service.class);
mockery.allowing(service::stringify).with(42).will(() -> "42");
所以在allowing
里面我有一个Function<Object, String>
有没有什么反射魔法可以让我从方法引用创建的函数中找到服务?
public WithClause allowing(Function<T,R> f) {
Object myServiceBackAgain = findTargetOf(function);
....
}
我知道Function总是来自这些方法引用,所以我很乐意根据需要进行降频转换。
这与相关的问题不同。是否可以将方法引用转换为MethodHandle?因为,首先,这不是同一个问题,只是在一个相关的领域。即使我能得到一个MethodHandle,我也无法从中得到目标
使用这篇SO文章中的技巧,您可以找到目标。下面的重要方法是findTarget
。事实证明,Lambda确实捕获了它们的目标,并且您可以从SerializedLambda
访问它们。
然而,这是一个相当恶劣的反思黑客,它可能会在未来的版本中崩溃。我不宽恕它的使用。
import java.io.Serializable;
import java.lang.invoke.SerializedLambda;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Optional;
import java.util.function.Function;
public class FindMethodReferenceTarget {
public static void main(String[] args) {
String s = "123";
Optional<Object> target = findTarget(s::charAt);
System.out.println(target.get().equals(s));
Object o = new FindMethodReferenceTarget();
target = findTarget(o::equals);
System.out.println(target.get().equals(o));
}
private static <T, R> Optional<Object> findTarget(
DebuggableFunction<T, R> methodReference) {
return getLambda(methodReference).map(l -> l.getCapturedArg(0));
}
private static Optional<SerializedLambda> getLambda(Serializable lambda) {
for (Class<?> cl = lambda.getClass(); cl != null; cl = cl.getSuperclass()) {
try {
Method m = cl.getDeclaredMethod("writeReplace");
m.setAccessible(true);
Object replacement = m.invoke(lambda);
if (!(replacement instanceof SerializedLambda)) {
break; // custom interface implementation
}
SerializedLambda l = (SerializedLambda) replacement;
return Optional.of(l);
} catch (NoSuchMethodException e) {
// do nothing
} catch (IllegalAccessException | InvocationTargetException e) {
break;
}
}
return Optional.empty();
}
@FunctionalInterface
private static interface DebuggableFunction<T, R> extends
Serializable,
Function<T, R> {}
}
没有直接的方法可以找到目标,因为方法引用只是被翻译成lambdas(根据定义,lambdas是匿名的)。因此,您需要使用一种变通方法。
假设您熟悉Java7的代理,因为您已经成功地实现了mock
工厂方法。
解决方法是,当有人调用你的allowing
方法时,你设置某种全局标志来提醒所有mock你想记录下一次调用,然后你调用你得到的lambda。通过查看哪个mock记录了调用,您现在已经找到了方法引用的目标,您可以取消设置全局标志,继续执行mocking框架的其余部分。
我知道这很难看。