INT-3109: make UriVariableTests work offline

JIRA: https://jira.springsource.org/browse/INT-3109

spring-integration-ws/src/test/java/org/springframework/integration/ws/config/UriVariableTests-context.xml

@@ -10,9 +10,8 @@
 	<bean id="interceptor" class="org.springframework.integration.ws.config.UriVariableTests$TestClientInterceptor"/>
 
 	<!--HTTP Transport-->
-	<ws:outbound-gateway request-channel="inputHttp"
-						 uri="http://test.org/{foo}-{bar}?param={param}"
-						 interceptor="interceptor">
+	<ws:outbound-gateway id="httpOutboundGateway" request-channel="inputHttp"
+						 uri="http://localhost/{foo}-{bar}?param={param}">
 		<ws:uri-variable name="foo" expression="payload.substring(1,7)"/>
 		<ws:uri-variable name="bar" expression="headers.x"/>
 		<ws:uri-variable name="param" expression="headers.param"/>

spring-integration-ws/src/test/java/org/springframework/integration/ws/config/UriVariableTests.java

@@ -17,12 +17,13 @@
 package org.springframework.integration.ws.config;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertThat;
 import static org.junit.Assert.assertTrue;
 
 import java.io.IOException;
 import java.net.URI;
 import java.net.URISyntaxException;
-
+import java.util.concurrent.atomic.AtomicReference;
 import javax.jms.BytesMessage;
 import javax.jms.Connection;
 import javax.jms.ConnectionFactory;
@@ -32,26 +33,34 @@ import javax.jms.MessageProducer;
 import javax.jms.Queue;
 import javax.jms.Session;
 
+import org.hamcrest.Matchers;
 import org.jivesoftware.smack.XMPPConnection;
 import org.jivesoftware.smack.packet.Packet;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.ArgumentCaptor;
 import org.mockito.Mockito;
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
+
+import org.springframework.beans.DirectFieldAccessor;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.integration.Message;
 import org.springframework.integration.MessageChannel;
 import org.springframework.integration.MessageHandlingException;
+import org.springframework.integration.core.MessageHandler;
 import org.springframework.integration.support.MessageBuilder;
 import org.springframework.integration.test.util.TestUtils;
 import org.springframework.test.context.ContextConfiguration;
 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
 import org.springframework.ws.client.WebServiceClientException;
 import org.springframework.ws.client.WebServiceIOException;
-import org.springframework.ws.client.WebServiceTransportException;
+import org.springframework.ws.client.core.WebServiceMessageCallback;
+import org.springframework.ws.client.core.WebServiceMessageExtractor;
+import org.springframework.ws.client.core.WebServiceTemplate;
 import org.springframework.ws.client.support.interceptor.ClientInterceptor;
 import org.springframework.ws.context.MessageContext;
-import org.springframework.ws.soap.SoapMessageCreationException;
 import org.springframework.ws.transport.WebServiceConnection;
 import org.springframework.ws.transport.WebServiceMessageSender;
 import org.springframework.ws.transport.context.TransportContext;
@@ -67,6 +76,10 @@ import org.springframework.ws.transport.mail.MailSenderConnection;
 @ContextConfiguration
 public class UriVariableTests {
 
+	@Autowired
+	@Qualifier("httpOutboundGateway.handler")
+	private MessageHandler httpOutboundGateway;
+
 	@Autowired
 	private TestClientInterceptor interceptor;
 
@@ -95,7 +108,24 @@ public class UriVariableTests {
 	private Int2720EmailTestClientInterceptor emailInterceptor;
 
 	@Test
+	@SuppressWarnings("unchecked")
 	public void testHttpUriVariables() {
+		WebServiceTemplate webServiceTemplate = TestUtils.getPropertyValue(this.httpOutboundGateway, "webServiceTemplate", WebServiceTemplate.class);
+		webServiceTemplate = Mockito.spy(webServiceTemplate);
+		final AtomicReference<String> uri = new AtomicReference<String>();
+		Mockito.doAnswer(new Answer<Object>() {
+			@Override
+			public Object answer(InvocationOnMock invocation) throws Throwable {
+				uri.set((String) invocation.getArguments()[0]);
+				throw new WebServiceIOException("intentional");
+			}
+		}).when(webServiceTemplate)
+				.sendAndReceive(Mockito.anyString(),
+						Mockito.any(WebServiceMessageCallback.class),
+						(WebServiceMessageExtractor<Object>) Mockito.any(WebServiceMessageExtractor.class));
+
+		new DirectFieldAccessor(this.httpOutboundGateway).setPropertyValue("webServiceTemplate", webServiceTemplate);
+
 		Message<?> message = MessageBuilder.withPayload("<spring/>")
 				.setHeader("x", "integration")
 				.setHeader("param", "test1 & test2")
@@ -105,11 +135,9 @@ public class UriVariableTests {
 		}
 		catch (MessageHandlingException e) {
 			// expected
-			Class<?> causeType = e.getCause().getClass();
-			assertTrue(WebServiceIOException.class.equals(causeType) // offline
-					|| SoapMessageCreationException.class.equals(causeType));
+			assertThat(e.getCause(), Matchers.is(Matchers.instanceOf(WebServiceIOException.class))); // offline
 		}
-		assertEquals("http://test.org/spring-integration?param=test1%20%26%20test2", this.interceptor.getLastUri().toString());
+		assertEquals("http://localhost/spring-integration?param=test1%20%26%20test2", uri.get());
 	}
 
 	@Test
@@ -170,7 +198,7 @@ public class UriVariableTests {
 		catch (MessageHandlingException e) {
 			// expected
 			Class<?> causeType = e.getCause().getClass();
-			assertTrue(WebServiceTransportException.class.equals(causeType)); // offline
+			assertTrue(WebServiceIOException.class.equals(causeType)); // offline
 		}
 		WebServiceConnection webServiceConnection = this.emailInterceptor.getLastWebServiceConnection();
 		assertEquals(testEmailTo, TestUtils.getPropertyValue(webServiceConnection, "to").toString());
@@ -252,7 +280,8 @@ public class UriVariableTests {
 				throw new IllegalStateException("expected MailSenderConnection in the TransportContext");
 			}
 
-			return super.handleRequest(messageContext);
+			super.handleRequest(messageContext);
+			throw new WebServiceIOException("intentional");
 		}
 	}