如何使用Mockito测试SOAP消息

我对Mockito相当陌生，我想我会尝试使用它来测试SOAP处理程序。然而，事实证明，这比我预期/期望的要痛苦得多。

我希望验证我的处理程序是否能够提取SOAPMessage标头中的messageID。但是，从处理程序中，获取标头的唯一方法是通过context/message/part/envelope/header。使用Mockito，我的解决方案是模拟我的SOAPMessage，这意味着创建每个单独的对象，并存根该方法。

我只能想象有一种更简单/更干净的方法来实现这一点：

@RunWith(MockitoJUnitRunner.class)
public class UUIDHandlerTest {
    @Mock private SOAPMessage message;
    @Mock private SOAPEnvelope envelope;
    @Mock private SOAPHeader header;
    @Mock private SOAPPart part;
    @Mock
    private SOAPMessageContext context;
    @Before
    public void setup() throws SOAPException{
        when( context.getMessage()).thenReturn(message);
        when( message.getSOAPPart()).thenReturn(part);
        when( part.getEnvelope()).thenReturn(envelope);
        when( envelope.getHeader()).thenReturn(header);
    }

    @Test
    public void testHandleInboundMessage() {
        when( context.get(SOAPMessageContext.MESSAGE_OUTBOUND_PROPERTY)).thenReturn(false);
        when(header.getElementsByTagName(anyString())).thenAnswer(new Answer<NodeList>() {
            /* (non-Javadoc)
             * @see org.mockito.stubbing.Answer#answer(org.mockito.invocation.InvocationOnMock)
             */
            @Override
            public NodeList answer(InvocationOnMock invocation) throws Throwable {
                DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
                DocumentBuilder db = dbf.newDocumentBuilder();
                Document doc = db.parse(new ByteArrayInputStream(new String("<wsa:MessageID>messageId</wsa:MessageID>").getBytes()));
                // TODO Auto-generated method stub
                return doc.getElementsByTagName("wsa:MessageID");
            }
        });

        // call the test class
        new UUIDHandler().handleMessage(context);
        // check the MDC value
        assertEquals("messageId", MDC.get(LoggerConstants.DC_PROPERTY_MESSAGE_ID));
    }
}

正如我所说，它是有效的，但它看起来像是一个非常丑陋/沉重的解决方案。

有没有什么办法让这件事更简单/更干净？

谢谢！

Eric

    SOAPMessageContext context = 
            mock(SOAPMessageContext.class, RETURNS_DEEP_STUBS);
    when(context.getMessage().getSOAPPart().getEnvelope().
            getHeader().getElementsByTagName(anyString())).
            then(...);

    @Mock(answer = Answers.RETURNS_DEEP_STUBS) SOAPMessageContext context;