List of usage examples for org.springframework.mock.web MockHttpServletResponse getStatus
@Override public int getStatus()
From source file:org.jasig.cas.support.oauth.web.OAuth20TokenAuthorizationCodeControllerTests.java
@Test public void verifyRedirectUriDoesNotStartWithServiceId() throws Exception { final AuthorizationCode authorizationCode = mock(AuthorizationCode.class); final OAuthRegisteredService service = getRegisteredService(REDIRECT_URI, CLIENT_SECRET); final CentralOAuthService centralOAuthService = mock(CentralOAuthService.class); when(centralOAuthService.getToken(CODE, AuthorizationCode.class)).thenReturn(authorizationCode); when(centralOAuthService.getRegisteredService(CLIENT_ID)).thenReturn(service); final MockHttpServletRequest mockRequest = new MockHttpServletRequest("POST", CONTEXT + OAuthConstants.TOKEN_URL); mockRequest.setParameter(OAuthConstants.GRANT_TYPE, OAuthConstants.AUTHORIZATION_CODE); mockRequest.setParameter(OAuthConstants.CODE, CODE); mockRequest.setParameter(OAuthConstants.CLIENT_ID, CLIENT_ID); mockRequest.setParameter(OAuthConstants.CLIENT_SECRET, CLIENT_SECRET); mockRequest.setParameter(OAuthConstants.REDIRECT_URI, OTHER_REDIRECT_URI); final MockHttpServletResponse mockResponse = new MockHttpServletResponse(); final OAuth20WrapperController oauth20WrapperController = new OAuth20WrapperController(); oauth20WrapperController.setCentralOAuthService(centralOAuthService); oauth20WrapperController.afterPropertiesSet(); final ModelAndView modelAndView = oauth20WrapperController.handleRequest(mockRequest, mockResponse); assertNull(modelAndView);/*w w w . j av a 2 s. c o m*/ assertEquals(HttpStatus.SC_BAD_REQUEST, mockResponse.getStatus()); assertEquals("application/json", mockResponse.getContentType()); final ObjectMapper mapper = new ObjectMapper(); final String expected = "{\"error\":\"" + OAuthConstants.INVALID_REQUEST + "\",\"error_description\":\"" + OAuthConstants.INVALID_REDIRECT_URI_DESCRIPTION + "\"}"; final JsonNode expectedObj = mapper.readTree(expected); final JsonNode receivedObj = mapper.readTree(mockResponse.getContentAsString()); assertEquals(expectedObj.get("error").asText(), receivedObj.get("error").asText()); assertEquals(expectedObj.get("error_description").asText(), receivedObj.get("error_description").asText()); }
From source file:org.jasig.cas.support.oauth.web.OAuth20TokenAuthorizationCodeControllerTests.java
@Test public void verifyInvalidGrantType() throws Exception { final AuthorizationCode authorizationCode = mock(AuthorizationCode.class); when(authorizationCode.getType()).thenReturn(TokenType.PERSONAL); final OAuthRegisteredService service = getRegisteredService(REDIRECT_URI, CLIENT_SECRET); final CentralOAuthService centralOAuthService = mock(CentralOAuthService.class); when(centralOAuthService.getToken(CODE, AuthorizationCode.class)).thenReturn(authorizationCode); when(centralOAuthService.getRegisteredService(CLIENT_ID)).thenReturn(service); final MockHttpServletRequest mockRequest = new MockHttpServletRequest("POST", CONTEXT + OAuthConstants.TOKEN_URL); mockRequest.setParameter(OAuthConstants.GRANT_TYPE, OAuthConstants.AUTHORIZATION_CODE); mockRequest.setParameter(OAuthConstants.CODE, CODE); mockRequest.setParameter(OAuthConstants.CLIENT_ID, CLIENT_ID); mockRequest.setParameter(OAuthConstants.CLIENT_SECRET, CLIENT_SECRET); mockRequest.setParameter(OAuthConstants.REDIRECT_URI, REDIRECT_URI); final MockHttpServletResponse mockResponse = new MockHttpServletResponse(); final OAuth20WrapperController oauth20WrapperController = new OAuth20WrapperController(); oauth20WrapperController.setCentralOAuthService(centralOAuthService); oauth20WrapperController.afterPropertiesSet(); final ModelAndView modelAndView = oauth20WrapperController.handleRequest(mockRequest, mockResponse); assertNull(modelAndView);/*from w ww . ja v a 2s . c o m*/ assertEquals(HttpStatus.SC_BAD_REQUEST, mockResponse.getStatus()); assertEquals("application/json", mockResponse.getContentType()); final ObjectMapper mapper = new ObjectMapper(); final String expected = "{\"error\":\"" + OAuthConstants.INVALID_GRANT + "\",\"error_description\":\"" + OAuthConstants.INVALID_GRANT_TYPE_DESCRIPTION + "\"}"; final JsonNode expectedObj = mapper.readTree(expected); final JsonNode receivedObj = mapper.readTree(mockResponse.getContentAsString()); assertEquals(expectedObj.get("error").asText(), receivedObj.get("error").asText()); assertEquals(expectedObj.get("error_description").asText(), receivedObj.get("error_description").asText()); }
From source file:io.restassured.module.mockmvc.internal.MockMvcRequestSenderImpl.java
private String assembleStatusLine(MockHttpServletResponse response, Exception resolvedException) { StringBuilder builder = new StringBuilder(); builder.append(response.getStatus()); if (isNotBlank(response.getErrorMessage())) { builder.append(" ").append(response.getErrorMessage()); } else if (resolvedException != null) { builder.append(" ").append(resolvedException.getMessage()); }//from w w w . j av a2s.com return builder.toString(); }
From source file:org.wrml.server.WrmlServletTest.java
@Test public void requestWithBadHostHeader2() throws ServletException, IOException { MockHttpServletRequest request = new MockHttpServletRequest(); initMockHttpRequest(request, CAPRICA_SIX_ENDPOINT); request.setMethod(Method.Get.getProtocolGivenName()); //request.addHeader(HttpHeaders.ACCEPT, JSON_MEDIA_TYPE); request.addHeader(WrmlServlet.WRML_HOST_HEADER_NAME, BAD_HOST_2); initMockWrmlRequest(request, Method.Get, CAPRICA_SIX_ENDPOINT, CAPRICA_SCHEMA_URI); MockHttpServletResponse response = new MockHttpServletResponse(); _Servlet.service(request, response); // Verify Model Write Assert.assertTrue(response.getContentType().contains("text/plain")); Assert.assertEquals(HttpServletResponse.SC_BAD_REQUEST, response.getStatus()); }
From source file:org.wrml.server.WrmlServletTest.java
@Test public void requestRootWithSlash() throws ServletException, IOException { MockHttpServletRequest request = new MockHttpServletRequest(); initMockHttpRequest(request, DOCROOT_SLASH_ENDPOINT); request.setMethod(Method.Get.getProtocolGivenName()); MockHttpServletResponse response = new MockHttpServletResponse(); initMockWrmlRequest(request, Method.Get, DOCROOT_ENDPOINT, CAPRICA_SCHEMA_URI); _Servlet.service(request, response); // Verify Model Write Assert.assertEquals(HttpServletResponse.SC_OK, response.getStatus()); Assert.assertEquals(DEFAULT_CONTENT_TYPE, response.getContentType()); }
From source file:org.wrml.server.WrmlServletTest.java
@Test public void requestWithHostHeader() throws ServletException, IOException { MockHttpServletRequest request = new MockHttpServletRequest(); initMockHttpRequest(request, CAPRICA_SIX_SPOOF_1_ENDPOINT); request.setMethod(Method.Get.getProtocolGivenName()); request.addHeader(WrmlServlet.WRML_HOST_HEADER_NAME, LOCALHOST); MockHttpServletResponse response = new MockHttpServletResponse(); initMockWrmlRequest(request, Method.Get, CAPRICA_SIX_SPOOF_1_ENDPOINT, CAPRICA_SCHEMA_URI); _Servlet.service(request, response); // Verify Model Write Assert.assertEquals(HttpServletResponse.SC_OK, response.getStatus()); Assert.assertEquals(DEFAULT_CONTENT_TYPE, response.getContentType()); Assert.assertEquals(response.getContentAsByteArray().length, response.getContentLength()); }
From source file:org.wrml.server.WrmlServletTest.java
@Test public void requestWithHostAndPortHeaders() throws ServletException, IOException { MockHttpServletRequest request = new MockHttpServletRequest(); initMockHttpRequest(request, CAPRICA_SIX_ENDPOINT); request.setMethod(Method.Get.getProtocolGivenName()); request.addHeader(WrmlServlet.WRML_HOST_HEADER_NAME, LOCALHOST); request.addHeader(WrmlServlet.WRML_PORT_HEADER_NAME, PORT_1); MockHttpServletResponse response = new MockHttpServletResponse(); initMockWrmlRequest(request, Method.Get, CAPRICA_SIX_SPOOF_2_ENDPOINT, CAPRICA_SCHEMA_URI); _Servlet.service(request, response); // Verify Model Write Assert.assertEquals(HttpServletResponse.SC_OK, response.getStatus()); Assert.assertEquals(DEFAULT_CONTENT_TYPE, response.getContentType()); Assert.assertEquals(response.getContentAsByteArray().length, response.getContentLength()); }
From source file:org.jasig.cas.support.oauth.web.OAuth20ProfileControllerTests.java
@Test public void verifyOKWithPersonalToken() throws Exception { final Map<String, Object> map = new HashMap<>(); map.put(NAME, VALUE);//from w w w . j av a 2 s . co m final List<String> list = Arrays.asList(VALUE, VALUE); map.put(NAME2, list); final Principal principal = mock(Principal.class); when(principal.getId()).thenReturn(ID); when(principal.getAttributes()).thenReturn(map); final Authentication authentication = mock(Authentication.class); when(authentication.getPrincipal()).thenReturn(principal); final TicketGrantingTicket ticketGrantingTicket = mock(TicketGrantingTicket.class); when(ticketGrantingTicket.isExpired()).thenReturn(false); when(ticketGrantingTicket.getAuthentication()).thenReturn(authentication); final Service service = new SimpleWebApplicationServiceImpl("id"); final AccessToken accessToken = mock(AccessToken.class); when(accessToken.getId()).thenReturn(AT_ID); when(accessToken.getType()).thenReturn(TokenType.PERSONAL); when(accessToken.getService()).thenReturn(service); when(accessToken.getTicketGrantingTicket()).thenReturn(ticketGrantingTicket); final CentralOAuthService centralOAuthService = mock(CentralOAuthService.class); when(centralOAuthService.getToken(AT_ID, AccessToken.class)).thenReturn(accessToken); final MockHttpServletRequest mockRequest = new MockHttpServletRequest("GET", CONTEXT + OAuthConstants.PROFILE_URL); mockRequest.setParameter(OAuthConstants.ACCESS_TOKEN, AT_ID); final MockHttpServletResponse mockResponse = new MockHttpServletResponse(); final OAuth20WrapperController oauth20WrapperController = new OAuth20WrapperController(); oauth20WrapperController.setCentralOAuthService(centralOAuthService); oauth20WrapperController.afterPropertiesSet(); final ModelAndView modelAndView = oauth20WrapperController.handleRequest(mockRequest, mockResponse); assertNull(modelAndView); assertEquals(HttpStatus.SC_OK, mockResponse.getStatus()); assertEquals(CONTENT_TYPE, mockResponse.getContentType()); final ObjectMapper mapper = new ObjectMapper(); final String expected = "{\"id\":\"" + ID + "\"}"; final JsonNode expectedObj = mapper.readTree(expected); final JsonNode receivedObj = mapper.readTree(mockResponse.getContentAsString()); assertEquals(expectedObj.get("id").asText(), receivedObj.get("id").asText()); }
From source file:org.jasig.cas.support.oauth.web.OAuth20TokenAuthorizationCodeControllerTests.java
@Test public void verifyOnlineOK() throws Exception { final ServiceTicket serviceTicket = mock(ServiceTicket.class); when(serviceTicket.getCreationTime()).thenReturn(new Date().getTime()); final AuthorizationCode authorizationCode = mock(AuthorizationCode.class); when(authorizationCode.getTicket()).thenReturn(serviceTicket); when(authorizationCode.getType()).thenReturn(TokenType.ONLINE); final OAuthRegisteredService service = getRegisteredService(REDIRECT_URI, CLIENT_SECRET); final AccessToken accessToken = mock(AccessToken.class); when(accessToken.getId()).thenReturn(AT_ID); when(accessToken.getTicket()).thenReturn(serviceTicket); final CentralOAuthService centralOAuthService = mock(CentralOAuthService.class); when(centralOAuthService.getToken(CODE, AuthorizationCode.class)).thenReturn(authorizationCode); when(centralOAuthService.getRegisteredService(CLIENT_ID)).thenReturn(service); when(centralOAuthService.grantOnlineAccessToken(authorizationCode)).thenReturn(accessToken); final MockHttpServletRequest mockRequest = new MockHttpServletRequest("POST", CONTEXT + OAuthConstants.TOKEN_URL); mockRequest.setParameter(OAuthConstants.GRANT_TYPE, OAuthConstants.AUTHORIZATION_CODE); mockRequest.setParameter(OAuthConstants.CODE, CODE); mockRequest.setParameter(OAuthConstants.CLIENT_ID, CLIENT_ID); mockRequest.setParameter(OAuthConstants.CLIENT_SECRET, CLIENT_SECRET); mockRequest.setParameter(OAuthConstants.REDIRECT_URI, REDIRECT_URI); final MockHttpServletResponse mockResponse = new MockHttpServletResponse(); final OAuth20WrapperController oauth20WrapperController = new OAuth20WrapperController(); oauth20WrapperController.setCentralOAuthService(centralOAuthService); oauth20WrapperController.setTimeout(TIMEOUT); oauth20WrapperController.afterPropertiesSet(); final ModelAndView modelAndView = oauth20WrapperController.handleRequest(mockRequest, mockResponse); assertNull(modelAndView);//w w w . j a v a2 s . c o m assertEquals(HttpStatus.SC_OK, mockResponse.getStatus()); assertEquals("application/json", mockResponse.getContentType()); final ObjectMapper mapper = new ObjectMapper(); final String expected = "{\"token_type\":\"" + OAuthConstants.BEARER_TOKEN + "\",\"expires_in\":\"" + TIMEOUT + "\",\"access_token\":\"" + AT_ID + "\"}"; final JsonNode expectedObj = mapper.readTree(expected); final JsonNode receivedObj = mapper.readTree(mockResponse.getContentAsString()); assertEquals(expectedObj.get("token_type").asText(), receivedObj.get("token_type").asText()); assertTrue("received expires_at greater or equal to expected", expectedObj.get("expires_in").asInt() >= receivedObj.get("expires_in").asInt()); assertEquals(expectedObj.get("access_token").asText(), receivedObj.get("access_token").asText()); }
From source file:org.jasig.cas.support.oauth.web.OAuth20ServiceValidateControllerTests.java
@Test public void verifyGetTicketException() throws Exception { final MockHttpServletRequest mockRequest = new MockHttpServletRequest("GET", URL); final WebApplicationService webApplicationService = mock(WebApplicationService.class); when(webApplicationService.getArtifactId()).thenReturn(SERVICE_TICKET_ID); final ArgumentExtractor argumentExtractor = mock(ArgumentExtractor.class); when(argumentExtractor.extractService(mockRequest)).thenReturn(webApplicationService); final TicketGrantingTicket ticketGrantingTicket = mock(TicketGrantingTicket.class); final ServiceTicket serviceTicket = mock(ServiceTicket.class); when(serviceTicket.getGrantingTicket()).thenReturn(ticketGrantingTicket); when(serviceTicket.getService()).thenReturn(webApplicationService); final Assertion assertion = mock(Assertion.class); final CentralAuthenticationService centralAuthenticationService = mock(CentralAuthenticationService.class); when(centralAuthenticationService.getTicket(SERVICE_TICKET_ID, Ticket.class)) .thenThrow(new InvalidTicketException("weak")); when(centralAuthenticationService.validateServiceTicket(SERVICE_TICKET_ID, webApplicationService)) .thenReturn(assertion);//ww w . ja v a 2s . c o m final Set<String> scopes = new HashSet<>(); scopes.add(SCOPE1); scopes.add(SCOPE2); final AccessToken accessToken = mock(AccessToken.class); when(accessToken.getId()).thenReturn(AT_ID); when(accessToken.getScopes()).thenReturn(scopes); final CentralOAuthService centralOAuthService = mock(CentralOAuthService.class); when(centralOAuthService.grantCASAccessToken(ticketGrantingTicket, webApplicationService)) .thenReturn(accessToken); final MockHttpServletResponse mockResponse = new MockHttpServletResponse(); final OAuth20ServiceValidateController oauth20ServiceValidateController = new OAuth20ServiceValidateController(); oauth20ServiceValidateController.setArgumentExtractor(argumentExtractor); oauth20ServiceValidateController.setCentralAuthenticationService(centralAuthenticationService); oauth20ServiceValidateController.setCentralOAuthService(centralOAuthService); oauth20ServiceValidateController.setSuccessView(SUCCESS_VIEW); final ModelAndView modelAndView = oauth20ServiceValidateController.handleRequest(mockRequest, mockResponse); assertEquals(HttpStatus.SC_OK, mockResponse.getStatus()); assertEquals("", mockResponse.getContentAsString()); final Map<String, Object> model = modelAndView.getModel(); assertTrue(!model.containsKey(OAuthConstants.CAS_PROTOCOL_ACCESS_TOKEN)); assertTrue(!model.containsKey(OAuthConstants.CAS_PROTOCOL_ACCESS_TOKEN_SCOPE)); }