模拟静态方法链

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);