List of usage examples for org.springframework.mock.web MockHttpServletRequest setQueryString
public void setQueryString(@Nullable String queryString)
From source file:com.amashchenko.struts2.pdfstream.PdfStreamResultTest.java
@Test public void testFindBaseUrl() throws Exception { Assert.assertNotNull(pdfStreamResult); MockHttpServletRequest request = new MockHttpServletRequest(); request.setLocalName("localhost"); request.setRequestURI("/contextPath/requestURI"); request.setQueryString("queryString"); request.setContextPath("/contextPath"); final String baseUrl = pdfStreamResult.findBaseUrl(request); Assert.assertEquals("http://localhost/contextPath/", baseUrl); }
From source file:com.tasktop.c2c.server.common.service.tests.ajp.AjpProxyWebTest.java
@Test public void testProxyHandlesCookies() throws Exception { final String ajpBaseUri = String.format("ajp://localhost:%s", container.getAjpPort()); Payload payload = new Payload(HttpServletResponse.SC_OK, "some content\none two three\n\nfour"); payload.getResponseHeaders().put("foo", "bar"); payload.getSessionVariables().put("s1", "v1"); TestServlet.setResponsePayload(payload); MockHttpServletRequest request = new MockHttpServletRequest(); request.setMethod("GET"); request.setRequestURI("/test"); request.setQueryString("a=b"); request.setParameter("a", new String[] { "b" }); request.addHeader("c", "d ef"); MockHttpServletResponse response = new MockHttpServletResponse(); proxy.proxyRequest(ajpBaseUri + "/foo", request, response); Request firstRequest = null;/*from w w w.j av a 2 s .c o m*/ for (int i = 0; i < 100; i++) { firstRequest = TestServlet.getLastRequest(); // If our request is not yet there, then pause and retry shortly - proxying is an async process, and this // request was sometimes coming back as null which was causing test failures on the first assert below. if (firstRequest == null) { Thread.sleep(10); } else { // Got our request, so break now. break; } } Assert.assertTrue(firstRequest.isNewSession()); Assert.assertEquals("v1", firstRequest.getSessionAttributes().get("s1")); List<org.apache.commons.httpclient.Cookie> cookies = new ArrayList<org.apache.commons.httpclient.Cookie>(); for (String headerName : response.getHeaderNames()) { if (headerName.equalsIgnoreCase("set-cookie") || headerName.equalsIgnoreCase("set-cookie2")) { cookies.addAll(Arrays.asList(new RFC2965Spec().parse("localhost", container.getPort(), "/", false, response.getHeader(headerName).toString()))); } } Assert.assertEquals(1, cookies.size()); Cookie cookie = cookies.get(0); Assert.assertEquals("almp.JSESSIONID", cookie.getName()); MockHttpServletRequest request2 = new MockHttpServletRequest(); request2.setMethod("GET"); request2.setRequestURI("/test"); request2.addHeader("Cookie", cookie.toExternalForm()); MockHttpServletResponse response2 = new MockHttpServletResponse(); payload = new Payload(HttpServletResponse.SC_OK, "test"); TestServlet.setResponsePayload(payload); proxy.proxyRequest(ajpBaseUri + "/foo", request2, response2); Request secondRequest = TestServlet.getLastRequest(); Assert.assertFalse(secondRequest.isNewSession()); Assert.assertEquals(firstRequest.getSessionId(), secondRequest.getSessionId()); Assert.assertEquals("v1", secondRequest.getSessionAttributes().get("s1")); }
From source file:nl.surfnet.coin.api.AccessConfirmationControllerTest.java
@Test public void test() { MockHttpServletRequest req = new MockHttpServletRequest("GET", "/foobar123"); String url = AccessConfirmationController.getUrlWithLanguageParam(req, "tr"); assertThat(url, IsEqual.equalTo("?lang=tr")); // see that a current lang-parameter is replaced req.setQueryString("lang=abc&def=123"); url = AccessConfirmationController.getUrlWithLanguageParam(req, "tr"); assertThat(url, IsEqual.equalTo("?def=123&lang=tr")); // see that a not-lang-parameter is not touched req.setQueryString("notlang=abc"); url = AccessConfirmationController.getUrlWithLanguageParam(req, "tr"); assertThat(url, IsEqual.equalTo("?notlang=abc&lang=tr")); }
From source file:fi.okm.mpass.shibboleth.authn.impl.ValidateWilmaResponseTest.java
/** * Test with invalid checksum format./*from w w w . ja v a 2 s.c o m*/ * @throws Exception */ @Test protected void testInvalidChecksumFormat() throws Exception { MockHttpServletRequest servletRequest = initializeServletRequest(true, true, true); servletRequest.setQueryString(servletRequest.getQueryString() + "invalid"); action.setHttpServletRequest(servletRequest); action.initialize(); prc.getSubcontext(AuthenticationContext.class, false).setAttemptedFlow(authenticationFlows.get(0)); final WilmaAuthenticationContext wilmaContext = prc.getSubcontext(AuthenticationContext.class, false) .getSubcontext(WilmaAuthenticationContext.class, true); wilmaContext.setNonce(nonce); final Event event = action.execute(src); ActionTestingSupport.assertEvent(event, AuthnEventIds.NO_CREDENTIALS); }
From source file:org.wallride.job.UpdatePostViewsItemWriter.java
@Override public void write(List<? extends List> items) throws Exception { WebApplicationContext context = WebApplicationContextUtils.getWebApplicationContext(servletContext, "org.springframework.web.servlet.FrameworkServlet.CONTEXT.guestServlet"); if (context == null) { throw new ServiceException("GuestServlet is not ready yet"); }/*from w w w . j a v a 2s. c om*/ final RequestMappingHandlerMapping mapping = context.getBean(RequestMappingHandlerMapping.class); for (List item : items) { UriComponents uriComponents = UriComponentsBuilder.fromUriString((String) item.get(0)).build(); logger.info("Processing [{}]", uriComponents.toString()); MockHttpServletRequest request = new MockHttpServletRequest(servletContext); request.setMethod("GET"); request.setRequestURI(uriComponents.getPath()); request.setQueryString(uriComponents.getQuery()); MockHttpServletResponse response = new MockHttpServletResponse(); BlogLanguageRewriteRule rewriteRule = new BlogLanguageRewriteRule(blogService); BlogLanguageRewriteMatch rewriteMatch = (BlogLanguageRewriteMatch) rewriteRule.matches(request, response); try { rewriteMatch.execute(request, response); } catch (ServletException e) { throw new ServiceException(e); } catch (IOException e) { throw new ServiceException(e); } request.setRequestURI(rewriteMatch.getMatchingUrl()); HandlerExecutionChain handler; try { handler = mapping.getHandler(request); } catch (Exception e) { throw new ServiceException(e); } if (!(handler.getHandler() instanceof HandlerMethod)) { continue; } HandlerMethod method = (HandlerMethod) handler.getHandler(); if (!method.getBeanType().equals(ArticleDescribeController.class) && !method.getBeanType().equals(PageDescribeController.class)) { continue; } // Last path mean code of post String code = uriComponents.getPathSegments().get(uriComponents.getPathSegments().size() - 1); Post post = postRepository.findOneByCodeAndLanguage(code, rewriteMatch.getBlogLanguage().getLanguage()); if (post == null) { logger.debug("Post not found [{}]", code); continue; } logger.info("Update the PageView. Post ID [{}]: {} -> {}", post.getId(), post.getViews(), item.get(1)); post.setViews(Long.parseLong((String) item.get(1))); postRepository.saveAndFlush(post); } }
From source file:gov.nyc.doitt.gis.geoclient.service.web.RestControllerTest.java
@Test public void testHandleMissingRequestParameter() { MockHttpServletRequest req = new MockHttpServletRequest(); String requestUri = "/foo"; String queryString = "bar=1"; req.setRequestURI(requestUri);/*w w w . j a v a 2 s. c o m*/ req.setQueryString(queryString); MissingAnyOfOptionalServletRequestParametersException e = new MissingAnyOfOptionalServletRequestParametersException( "dog", "cat"); ResponseEntity<BadRequest> result = this.restController.handleMissingRequestParameter(e, req); assertEquals(HttpStatus.BAD_REQUEST, result.getStatusCode()); assertEquals(String.format("%s?%s", requestUri, queryString), result.getBody().getRequestUri()); assertEquals(e.getMessage(), result.getBody().getMessage()); }
From source file:com.epam.ta.reportportal.ws.PagedResourcesAssemblerTest.java
private void prepareRequestContext(int currentPage, int currentPageSize) { MockHttpServletRequest request = new MockHttpServletRequest(); request.setRequestURI(REQUEST_URI); request.setParameter(CURRENT_PAGE_PARAMETER, String.valueOf(currentPage)); request.setParameter(PAGE_SIZE_PARAMETER, String.valueOf(currentPageSize)); request.setQueryString(String.format(QUERY_STRING_PATTERN, currentPage, currentPageSize)); ServletRequestAttributes attributes = new ServletRequestAttributes(request); RequestContextHolder.setRequestAttributes(attributes); }
From source file:fi.okm.mpass.shibboleth.authn.impl.ValidateWilmaResponseTest.java
/** * Test with invalid checksum./* w w w .j a v a 2 s . c o m*/ * @throws Exception */ @Test protected void testInvalidChecksum() throws Exception { MockHttpServletRequest servletRequest = initializeServletRequest(true, true, true); final String query = servletRequest.getQueryString(); servletRequest.setQueryString(query.substring(0, query.length() - 2) + "11"); action.setHttpServletRequest(servletRequest); action.initialize(); prc.getSubcontext(AuthenticationContext.class, false).setAttemptedFlow(authenticationFlows.get(0)); final WilmaAuthenticationContext wilmaContext = prc.getSubcontext(AuthenticationContext.class, false) .getSubcontext(WilmaAuthenticationContext.class, true); wilmaContext.setNonce(nonce); final Event event = action.execute(src); ActionTestingSupport.assertEvent(event, AuthnEventIds.NO_CREDENTIALS); }
From source file:org.ambraproject.admin.filter.RedirectToAmbraFilterTest.java
@Test(dataProvider = "journalContext") public void testFilter(VirtualJournalContext journalContext, String expectedRedirectUrl) throws ServletException, IOException { MockHttpServletRequest request = new MockHttpServletRequest(); MockHttpServletResponse response = new MockHttpServletResponse(); MockFilterChain filterChain = new MockFilterChain(); request.setAttribute(VirtualJournalContext.PUB_VIRTUALJOURNAL_CONTEXT, journalContext); request.setRequestURI("/article"); request.setQueryString("id=foo:bar"); filter.doFilter(request, response, filterChain); assertEquals(response.getRedirectedUrl(), expectedRedirectUrl + "/article?id=foo:bar", "response didn't get redirected correctly"); }
From source file:fi.okm.mpass.idp.authn.impl.OAuth2IdentityTest.java
/** * Initializes a servlet request.// w w w . ja va 2 s . c o m * @return */ protected MockHttpServletRequest initHttpServletRequest() { final MockHttpServletRequest httpRequest = new MockHttpServletRequest(); httpRequest.setQueryString("code=mockCode&state=mockState"); httpRequest.getSession(true).setAttribute(AbstractOAuth2Identity.SESSION_ATTR_STATE, new State("mockState")); return httpRequest; }