模拟静态方法链

Question

我正在尝试模拟一个静态方法

StaticClass.staticMethod(id).setContentType("text/plain").build();

我能够使用 PowerMockito 模拟静态方法及其返回类型，如下所示：

PowerMockito.when(StaticClass.staticMethod(id)).thenReturn(returnValue);

但是如何将该值发送到链式方法 setContentType()？

Answer 1

您需要多个时间。您需要模拟更多对象staticMethod 和contentType。

PowerMockito.when(StaticClass.staticMethod(id)).thenReturn(staticMethod);
PowerMockito.when(staticMethod.setContentType("text/plain")).thenReturn(contentType);
PowerMockito.when(contentType.build()).thenReturn(returnValue);