Java中模拟链式调用的最佳解决方案

Question

我有以下代码：

handler = NodeHandler.getINodeHandler(localZone).getITspPlatformHandler().getITspProcessorManagementHandler();

我这样嘲笑：

mockStatic(NodeHandler.class);
INodeHandler iNodeHandler = mock(INodeHandler.class,Mockito.RETURNS_DEEP_STUBS);
when(NodeHandler.getINodeHandler(localZone)).thenReturn(iNodeHandler);
ITspProcessorManagementHandler iTspProcessorManagementHandler =mock(ITspProcessorManagementHandler.class,Mockito.RETURNS_DEEP_STUBS);
when(iNodeHandler.getITspPlatformHandler().getITspProcessorManagementHandler()).thenReturn(iTspProcessorManagementHandler);

在几行代码之后，会出现另一个链式方法调用：

ITspTrafficProcessor processor = NodeHandler.getINodeHandler(localZone, localUI).getITspPlatformHandler().getITspProcessorManagementHandler()
                .getITspProcessorHandler(procs[i]).getITspTrafficProcessorHandler(0).getAttributes();

我这样嘲笑：

when(NodeHandler.getINodeHandler(localZone,UI.CORBA)).thenReturn(iNodeHandler);
when(iNodeHandler.getITspPlatformHandler().getITspProcessorManagementHandler()(+1+).getITspProcessorHandler(anyString())
            .getITspTrafficProcessorHandler(anyInt()).getAttributes()).thenReturn(null);

所以我的问题是，我找不到比这更好的解决方案，因为问题是如果我告诉mockito将null返回给处理程序而不是iTspProcessorManagementHandler，那么我在（1）处得到一个nullpoer异常，但如果我对我的代码进行以下更改：

INodeHandler iNodeHandler = mock(INodeHandler.class,Mockito.RETURNS_MOCKS);

而不是mockito嘲笑每个方法调用，并且我的当-时返回语句不会返回我想要的内容，例如null。所以有什么建议可以做一个更好的解决方案吗？？？？

Answer 1

像这样凌乱的嘲弄表明您可以改进抽象。我会考虑将该特定逻辑封装在助手接口/类中，或者将“TrainCrank”返回的预期类型注入到方法/类中。