@Test public void testOverwriteContentTypeHeader() { HttpMessageSender messageSender = new HttpMessageSender(); String requestUrl = "http://localhost:8088/test"; final String responseBody = "<TestResponse><Message>Hello World!</Message></TestResponse>"; messageSender.setReplyMessageHandler(replyMessageHandler); messageSender.setRequestMethod(HttpMethod.POST); messageSender.setRequestUrl(requestUrl); messageSender.setContentType("text/xml"); messageSender.setCharset("ISO-8859-1"); Message<?> requestMessage = MessageBuilder.withPayload("<TestRequest><Message>Hello World!</Message></TestRequest>") .setHeader("Content-Type", "application/xml;charset=UTF-8") .setHeader("Accept", "application/xml") .build(); messageSender.setRestTemplate(restTemplate); reset(restTemplate, replyMessageHandler); restTemplate.setErrorHandler(anyObject(ResponseErrorHandler.class)); expectLastCall().once(); expect( restTemplate.exchange( eq(requestUrl), eq(HttpMethod.POST), anyObject(HttpEntity.class), eq(String.class))) .andAnswer( new IAnswer<ResponseEntity<String>>() { public ResponseEntity<String> answer() throws Throwable { HttpEntity<?> httpRequest = (HttpEntity<?>) getCurrentArguments()[2]; Assert.assertEquals( httpRequest.getBody().toString(), "<TestRequest><Message>Hello World!</Message></TestRequest>"); Assert.assertEquals(httpRequest.getHeaders().size(), 2); Assert.assertEquals( httpRequest.getHeaders().getContentType().toString(), "application/xml;charset=UTF-8"); Assert.assertEquals( httpRequest.getHeaders().getAccept().get(0).toString(), "application/xml"); return new ResponseEntity<String>(responseBody, HttpStatus.OK); } }) .once(); replyMessageHandler.onReplyMessage(anyObject(Message.class)); expectLastCall().once(); replay(restTemplate, replyMessageHandler); messageSender.send(requestMessage); verify(restTemplate, replyMessageHandler); }
@Test public void testErrorResponsePropagateStrategy() { HttpMessageSender messageSender = new HttpMessageSender(); String requestUrl = "http://localhost:8088/test"; final String responseBody = "<TestResponse><Message>Hello World!</Message></TestResponse>"; messageSender.setReplyMessageHandler(replyMessageHandler); messageSender.setRequestMethod(HttpMethod.POST); messageSender.setRequestUrl(requestUrl); messageSender.setErrorHandlingStrategy(ErrorHandlingStrategy.PROPAGATE); Message<?> requestMessage = MessageBuilder.withPayload("<TestRequest><Message>Hello World!</Message></TestRequest>") .build(); messageSender.setRestTemplate(restTemplate); reset(restTemplate, replyMessageHandler); restTemplate.setErrorHandler(anyObject(ResponseErrorHandler.class)); expectLastCall().once(); expect( restTemplate.exchange( eq(requestUrl), eq(HttpMethod.POST), anyObject(HttpEntity.class), eq(String.class))) .andReturn(new ResponseEntity<String>(responseBody, HttpStatus.FORBIDDEN)) .once(); replyMessageHandler.onReplyMessage(anyObject(Message.class)); expectLastCall() .andAnswer( new IAnswer<Object>() { public Object answer() throws Throwable { Message<?> responseMessage = (Message<?>) getCurrentArguments()[0]; Assert.assertEquals( responseMessage.getHeaders().get(CitrusHttpMessageHeaders.HTTP_STATUS_CODE), HttpStatus.FORBIDDEN); Assert.assertEquals( responseMessage.getHeaders().get(CitrusHttpMessageHeaders.HTTP_REASON_PHRASE), "FORBIDDEN"); return null; } }) .once(); replay(restTemplate, replyMessageHandler); messageSender.send(requestMessage); verify(restTemplate, replyMessageHandler); }
@Test public void testEndpointUriResolver() { HttpMessageSender messageSender = new HttpMessageSender(); String requestUrl = "http://localhost:8088/test"; String responseBody = "<TestResponse><Message>Hello World!</Message></TestResponse>"; messageSender.setReplyMessageHandler(replyMessageHandler); messageSender.setRequestMethod(HttpMethod.GET); messageSender.setRequestUrl(requestUrl); Message<?> requestMessage = MessageBuilder.withPayload("<TestRequest><Message>Hello World!</Message></TestRequest>") .build(); EndpointUriResolver endpointUriResolver = EasyMock.createMock(EndpointUriResolver.class); messageSender.setEndpointUriResolver(endpointUriResolver); messageSender.setRestTemplate(restTemplate); reset(restTemplate, replyMessageHandler, endpointUriResolver); restTemplate.setErrorHandler(anyObject(ResponseErrorHandler.class)); expectLastCall().once(); expect(endpointUriResolver.resolveEndpointUri(requestMessage, "http://localhost:8088/test")) .andReturn("http://localhost:8081/new") .once(); expect( restTemplate.exchange( eq("http://localhost:8081/new"), eq(HttpMethod.GET), anyObject(HttpEntity.class), eq(String.class))) .andReturn(new ResponseEntity<String>(responseBody, HttpStatus.OK)) .once(); replyMessageHandler.onReplyMessage(anyObject(Message.class)); expectLastCall().once(); replay(restTemplate, replyMessageHandler, endpointUriResolver); messageSender.send(requestMessage); verify(restTemplate, replyMessageHandler, endpointUriResolver); }
@Test public void testOverwriteRequestMethod() { HttpMessageSender messageSender = new HttpMessageSender(); String requestUrl = "http://localhost:8088/test"; final String responseBody = "<TestResponse><Message>Hello World!</Message></TestResponse>"; messageSender.setReplyMessageHandler(replyMessageHandler); messageSender.setRequestMethod(HttpMethod.GET); messageSender.setRequestUrl(requestUrl); Message<?> requestMessage = MessageBuilder.withPayload("<TestRequest><Message>Hello World!</Message></TestRequest>") .setHeader(CitrusHttpMessageHeaders.HTTP_REQUEST_METHOD, "GET") .build(); messageSender.setRestTemplate(restTemplate); reset(restTemplate, replyMessageHandler); restTemplate.setErrorHandler(anyObject(ResponseErrorHandler.class)); expectLastCall().once(); expect( restTemplate.exchange( eq(requestUrl), eq(HttpMethod.GET), anyObject(HttpEntity.class), eq(String.class))) .andAnswer( new IAnswer<ResponseEntity<String>>() { public ResponseEntity<String> answer() throws Throwable { HttpEntity<?> httpRequest = (HttpEntity<?>) getCurrentArguments()[2]; Assert.assertNull(httpRequest.getBody()); // null because of GET Assert.assertEquals(httpRequest.getHeaders().size(), 1); Assert.assertEquals( httpRequest.getHeaders().getContentType().toString(), "text/plain;charset=UTF-8"); return new ResponseEntity<String>(responseBody, HttpStatus.OK); } }) .once(); replyMessageHandler.onReplyMessage(anyObject(Message.class)); expectLastCall().once(); replay(restTemplate, replyMessageHandler); messageSender.send(requestMessage); verify(restTemplate, replyMessageHandler); }
@Test public void testErrorResponseExceptionStrategy() { HttpMessageSender messageSender = new HttpMessageSender(); String requestUrl = "http://localhost:8088/test"; messageSender.setReplyMessageHandler(replyMessageHandler); messageSender.setRequestMethod(HttpMethod.POST); messageSender.setRequestUrl(requestUrl); messageSender.setErrorHandlingStrategy(ErrorHandlingStrategy.THROWS_EXCEPTION); Message<?> requestMessage = MessageBuilder.withPayload("<TestRequest><Message>Hello World!</Message></TestRequest>") .build(); messageSender.setRestTemplate(restTemplate); reset(restTemplate, replyMessageHandler); restTemplate.setErrorHandler(anyObject(ResponseErrorHandler.class)); expectLastCall().once(); expect( restTemplate.exchange( eq(requestUrl), eq(HttpMethod.POST), anyObject(HttpEntity.class), eq(String.class))) .andThrow(new HttpClientErrorException(HttpStatus.FORBIDDEN)) .once(); replay(restTemplate, replyMessageHandler); try { messageSender.send(requestMessage); Assert.fail("Missing exception due to http error status code"); } catch (HttpClientErrorException e) { Assert.assertEquals(e.getMessage(), "403 FORBIDDEN"); verify(restTemplate, replyMessageHandler); } }