diff --git a/test/src/test/java/org/springframework/ws/mock/client/MockSenderConnectionTest.java b/test/src/test/java/org/springframework/ws/mock/client/MockSenderConnectionTest.java index 9ae7dd14..0a441a2b 100644 --- a/test/src/test/java/org/springframework/ws/mock/client/MockSenderConnectionTest.java +++ b/test/src/test/java/org/springframework/ws/mock/client/MockSenderConnectionTest.java @@ -36,7 +36,7 @@ public class MockSenderConnectionTest { public void setUp() throws Exception { connection = new MockSenderConnection(); } - + @Test public void uri() throws Exception { assertTrue("connection does not have any URI", connection.hasAnyUri()); @@ -60,7 +60,6 @@ public class MockSenderConnectionTest { connection.send(request); - verify(requestMatcher1, requestMatcher2, request); } @@ -80,7 +79,6 @@ public class MockSenderConnectionTest { connection.send(request); } - @Test public void receive() throws Exception { ResponseCallback responseCallback = createMock(ResponseCallback.class); @@ -98,4 +96,28 @@ public class MockSenderConnectionTest { verify(responseCallback, messageFactory, response); } + + @Test + public void sendAndReceive() throws Exception { + WebServiceMessageFactory messageFactory = createMock(WebServiceMessageFactory.class); + WebServiceMessage request = createMock("request", WebServiceMessage.class); + WebServiceMessage response = createMock("response", WebServiceMessage.class); + + RequestMatcher requestMatcher = createMock("requestMatcher1", RequestMatcher.class); + ResponseCallback responseCallback = createMock(ResponseCallback.class); + connection.addRequestMatcher(requestMatcher).setResponseCallback(responseCallback); + + requestMatcher.match(request); + + expect(messageFactory.createWebServiceMessage()).andReturn(response); + responseCallback.doWithResponse(request, response); + + replay(responseCallback, messageFactory, response, request, requestMatcher); + + connection.send(request); + WebServiceMessage result = connection.receive(messageFactory); + assertSame(response, result); + + verify(responseCallback, messageFactory, response, request, requestMatcher); + } }