Java Examples for org.springframework.ws.transport.WebServiceMessageSender

protected WebServiceMessage setupWebServiceMessageSender() throws IOException {
    final WebServiceMessageSender webServiceMessageSender = this.getWebServiceMessageSender();
    final WebServiceConnection webServiceConnection = mock(WebServiceConnection.class);
    when(webServiceMessageSender.supports(any(URI.class))).thenReturn(true);
    when(webServiceMessageSender.createConnection(any(URI.class))).thenReturn(webServiceConnection);
    final WebServiceMessage webServiceMessage = mock(WebServiceMessage.class);
    when(webServiceConnection.receive(any(WebServiceMessageFactory.class))).thenReturn(webServiceMessage);
    return webServiceMessage;
}

/**
	 * Create mock {@link WebServiceMessageSender}. If behavior not defined,
	 * throws {@link IllegalStateException}.
	 * 
	 * @return
	 */
public WebServiceMessageSender createMock() throws IllegalStateException {
    if (requestProcessors.isEmpty()) {
        throw new IllegalStateException("No request processor defined. Please call at least returnResponse() method.");
    }
    MockWebServiceMessageSender messageSender = new MockWebServiceMessageSender();
    messageSender.setRequestProcessors(requestProcessors);
    messageSender.setInterceptors(interceptors);
    return messageSender;
}

@Override
protected Endpoint createEndpoint(String uri, String remaining, Map<String, Object> parameters) throws Exception {
    URI webServiceEndpointUri = new URI(remaining);
    // Get a WebServiceTemplate from the registry if specified by an option on the component, else create a new template with Spring-WS defaults
    WebServiceTemplate webServiceTemplate = resolveAndRemoveReferenceParameter(parameters, "webServiceTemplate", WebServiceTemplate.class, new WebServiceTemplate());
    WebServiceMessageSender messageSender = resolveAndRemoveReferenceParameter(parameters, "messageSender", WebServiceMessageSender.class, null);
    WebServiceMessageFactory messageFactory = resolveAndRemoveReferenceParameter(parameters, "messageFactory", WebServiceMessageFactory.class, null);
    if (webServiceTemplate.getDefaultUri() == null) {
        webServiceTemplate.setDefaultUri(webServiceEndpointUri.toString());
    }
    if (messageSender != null) {
        webServiceTemplate.setMessageSender(messageSender);
    }
    if (messageFactory != null) {
        webServiceTemplate.setMessageFactory(messageFactory);
    }
    SpringWebserviceConfiguration configuration = new SpringWebserviceConfiguration();
    configuration.setWebServiceTemplate(webServiceTemplate);
    EndpointHelper.setProperties(getCamelContext(), configuration, parameters);
    return new SpringWebserviceEndpoint(this, configuration);
}

/**
	 * Creates a {@link SoapEndpointInvocationChain} based on the given endpoint and {@link
	 * org.springframework.ws.soap.addressing.version.AddressingVersion}.
	 */
private EndpointInvocationChain getEndpointInvocationChain(Object endpoint, AddressingVersion version, MessageAddressingProperties requestMap, MessageContext messageContext) {
    URI responseAction = getResponseAction(endpoint, requestMap);
    URI faultAction = getFaultAction(endpoint, requestMap);
    WebServiceMessageSender[] messageSenders = getMessageSenders(endpoint);
    MessageIdStrategy messageIdStrategy = getMessageIdStrategy(endpoint);
    List<EndpointInterceptor> interceptors = new ArrayList<EndpointInterceptor>();
    interceptors.addAll(Arrays.asList(preInterceptors));
    AddressingEndpointInterceptor addressingInterceptor = new AddressingEndpointInterceptor(version, messageIdStrategy, messageSenders, responseAction, faultAction);
    interceptors.add(addressingInterceptor);
    interceptors.addAll(Arrays.asList(postInterceptors));
    if (this.smartInterceptors != null) {
        for (SmartEndpointInterceptor smartInterceptor : smartInterceptors) {
            if (smartInterceptor.shouldIntercept(messageContext, endpoint)) {
                interceptors.add(smartInterceptor);
            }
        }
    }
    return new SoapEndpointInvocationChain(endpoint, interceptors.toArray(new EndpointInterceptor[interceptors.size()]), actorsOrRoles, isUltimateReceiver);
}

private void prepareMessageSenders(SpringWebserviceConfiguration configuration) {
    // Skip this whole thing if none of the relevant config options are set.
    if (!(configuration.getTimeout() > -1) && configuration.getSslContextParameters() == null) {
        return;
    }
    WebServiceTemplate webServiceTemplate = configuration.getWebServiceTemplate();
    WebServiceMessageSender[] messageSenders = webServiceTemplate.getMessageSenders();
    for (int i = 0; i < messageSenders.length; i++) {
        WebServiceMessageSender messageSender = messageSenders[i];
        if (messageSender instanceof HttpComponentsMessageSender) {
            if (configuration.getSslContextParameters() != null) {
                LOG.warn("Not applying SSLContextParameters based configuration to HttpComponentsMessageSender.  " + "If you are using this MessageSender, which you are not by default, you will need " + "to configure SSL using the Commons HTTP 3.x Protocol registry.");
            }
            if (configuration.getTimeout() > -1) {
                if (messageSender.getClass().equals(HttpComponentsMessageSender.class)) {
                    ((HttpComponentsMessageSender) messageSender).setReadTimeout(configuration.getTimeout());
                } else {
                    LOG.warn("Not applying timeout configuration to HttpComponentsMessageSender based implementation.  " + "You are using what appears to be a custom MessageSender, which you are not doing by default. " + "You will need configure timeout on your own.");
                }
            }
        } else if (messageSender.getClass().equals(HttpUrlConnectionMessageSender.class)) {
            // Only if exact match denoting likely use of default configuration.  We don't want to get
            // sub-classes that might have been otherwise injected.
            messageSenders[i] = new AbstractHttpWebServiceMessageSenderDecorator((HttpUrlConnectionMessageSender) messageSender, configuration, getEndpoint().getCamelContext());
        } else {
            // For example this will be the case during unit-testing with the net.javacrumbs.spring-ws-test API
            LOG.warn("Ignoring the timeout and SSLContextParameters options for {}.  You will need to configure " + "these options directly on your custom configured WebServiceMessageSender", messageSender);
        }
    }
}

@Test
public void simpleGatewayWithCustomMessageSender() {
    AbstractEndpoint endpoint = this.context.getBean("gatewayWithCustomMessageSender", AbstractEndpoint.class);
    assertEquals(EventDrivenConsumer.class, endpoint.getClass());
    Object gateway = new DirectFieldAccessor(endpoint).getPropertyValue("handler");
    assertEquals(SimpleWebServiceOutboundGateway.class, gateway.getClass());
    DirectFieldAccessor accessor = new DirectFieldAccessor(gateway);
    accessor = new DirectFieldAccessor(accessor.getPropertyValue("webServiceTemplate"));
    WebServiceMessageSender messageSender = (WebServiceMessageSender) context.getBean("messageSender");
    assertEquals(messageSender, ((WebServiceMessageSender[]) accessor.getPropertyValue("messageSenders"))[0]);
}

/* (non-Javadoc)
	 * @see org.springframework.ws.client.core.support.WebServiceGatewaySupport#initGateway()
	 */
@Override
protected void initGateway() throws Exception {
    super.initGateway();
    WebServiceMessageSender[] senders = getWebServiceTemplate().getMessageSenders();
    for (WebServiceMessageSender sender : senders) {
        if (sender instanceof HttpComponentsMessageSender) {
            HttpComponentsMessageSender hSender = (HttpComponentsMessageSender) sender;
            ClientConnectionManager connectionManager = hSender.getHttpClient().getConnectionManager();
            SchemeRegistry schemeRegistry = connectionManager.getSchemeRegistry();
            SSLSocketFactory sf = new SSLSocketFactory(keyStore, safeToString(keyStorePassword), trustStore);
            Scheme https = new Scheme("https", 443, sf);
            schemeRegistry.register(https);
            log.info("initGateway connection manager with https scheme");
        }
    }
}