diff --git a/spring-ws-test/src/main/java/org/springframework/ws/test/client/MockWebServiceMessageSender.java b/spring-ws-test/src/main/java/org/springframework/ws/test/client/MockWebServiceMessageSender.java index b380a46e..1bf29361 100644 --- a/spring-ws-test/src/main/java/org/springframework/ws/test/client/MockWebServiceMessageSender.java +++ b/spring-ws-test/src/main/java/org/springframework/ws/test/client/MockWebServiceMessageSender.java @@ -31,9 +31,10 @@ import org.springframework.ws.transport.WebServiceMessageSender; * * @author Arjen Poutsma * @author Lukas Krecan + * @author Greg Turnquist * @since 2.0 */ -class MockWebServiceMessageSender implements WebServiceMessageSender { +public class MockWebServiceMessageSender implements WebServiceMessageSender { private final List expectedConnections = new LinkedList(); @@ -78,4 +79,9 @@ class MockWebServiceMessageSender implements WebServiceMessageSender { } } + void reset() { + expectedConnections.clear(); + connectionIterator = null; + } + } diff --git a/spring-ws-test/src/main/java/org/springframework/ws/test/client/MockWebServiceServer.java b/spring-ws-test/src/main/java/org/springframework/ws/test/client/MockWebServiceServer.java index 0b2be7a6..5a1db499 100644 --- a/spring-ws-test/src/main/java/org/springframework/ws/test/client/MockWebServiceServer.java +++ b/spring-ws-test/src/main/java/org/springframework/ws/test/client/MockWebServiceServer.java @@ -88,6 +88,7 @@ import org.springframework.ws.test.support.MockStrategiesHelper; * * @author Arjen Poutsma * @author Lukas Krecan + * @author Greg Turnquist * @since 2.0 */ public class MockWebServiceServer { @@ -165,7 +166,7 @@ public class MockWebServiceServer { } /** - * Verifies that all expectations were met. + * Verifies that all of the {@link MockWebServiceMessageSender}'s expectations were met. * * @throws AssertionError in case of unmet expectations */ @@ -173,7 +174,12 @@ public class MockWebServiceServer { mockMessageSender.verifyConnections(); } - + /** + * Reset the {@link MockWebServiceMessageSender}'s expectations. + */ + public void reset() { + mockMessageSender.reset(); + } }