Search in sources :

Example 11 with View

use of com.tvd12.ezyhttp.server.core.view.View in project ezyhttp by youngmonkeys.

the class ThymeleafViewContextTest method renderWithDialectButNotViewDialect.

@Test
public void renderWithDialectButNotViewDialect() throws Exception {
    // given
    TemplateResolver resolver = TemplateResolver.builder().build();
    ViewDialect viewDialect = mock(ViewDialect.class);
    ViewContext viewContext = new ThymeleafViewContextBuilder().templateResolver(resolver).viewDialects(Collections.singletonList(viewDialect)).build();
    ServletContext servletContext = mock(ServletContext.class);
    HttpServletRequest request = mock(HttpServletRequest.class);
    HttpServletResponse response = mock(HttpServletResponse.class);
    PrintWriter writer = mock(PrintWriter.class);
    when(response.getWriter()).thenReturn(writer);
    View view = View.builder().template("index.html").build();
    // when
    viewContext.render(servletContext, request, response, view);
    // then
    Asserts.assertNotNull(viewContext);
}
Also used : HttpServletRequest(javax.servlet.http.HttpServletRequest) TemplateResolver(com.tvd12.ezyhttp.server.core.view.TemplateResolver) ViewContext(com.tvd12.ezyhttp.server.core.view.ViewContext) ThymeleafViewContextBuilder(com.tvd12.ezyhttp.server.thymeleaf.ThymeleafViewContextBuilder) ServletContext(javax.servlet.ServletContext) HttpServletResponse(javax.servlet.http.HttpServletResponse) ViewDialect(com.tvd12.ezyhttp.server.core.view.ViewDialect) View(com.tvd12.ezyhttp.server.core.view.View) PrintWriter(java.io.PrintWriter) Test(org.testng.annotations.Test)

Example 12 with View

use of com.tvd12.ezyhttp.server.core.view.View in project ezyhttp by youngmonkeys.

the class ThymeleafViewContextTest method renderWithViewDialect.

@Test
public void renderWithViewDialect() throws Exception {
    // given
    TemplateResolver resolver = TemplateResolver.builder().build();
    ViewContext viewContext = new ThymeleafViewContextBuilder().templateResolver(resolver).viewDialects(Collections.singletonList(new VewHelloDialect())).build();
    ServletContext servletContext = mock(ServletContext.class);
    HttpServletRequest request = mock(HttpServletRequest.class);
    HttpServletResponse response = mock(HttpServletResponse.class);
    PrintWriter writer = mock(PrintWriter.class);
    when(response.getWriter()).thenReturn(writer);
    View view = View.builder().template("index.html").build();
    // when
    viewContext.render(servletContext, request, response, view);
    // then
    Asserts.assertNotNull(viewContext);
}
Also used : HttpServletRequest(javax.servlet.http.HttpServletRequest) TemplateResolver(com.tvd12.ezyhttp.server.core.view.TemplateResolver) ViewContext(com.tvd12.ezyhttp.server.core.view.ViewContext) ThymeleafViewContextBuilder(com.tvd12.ezyhttp.server.thymeleaf.ThymeleafViewContextBuilder) ServletContext(javax.servlet.ServletContext) HttpServletResponse(javax.servlet.http.HttpServletResponse) View(com.tvd12.ezyhttp.server.core.view.View) PrintWriter(java.io.PrintWriter) Test(org.testng.annotations.Test)

Example 13 with View

use of com.tvd12.ezyhttp.server.core.view.View in project ezyhttp by youngmonkeys.

the class BlockingServlet method handleResponseData.

protected void handleResponseData(HttpServletRequest request, HttpServletResponse response, Object data) throws Exception {
    Object body = data;
    if (data instanceof ResponseEntity) {
        ResponseEntity entity = (ResponseEntity) body;
        body = entity.getBody();
        response.setStatus(entity.getStatus());
        MultiValueMap headers = entity.getHeaders();
        if (headers != null) {
            Map<String, String> encodedHeaders = headers.toMap();
            for (Entry<String, String> entry : encodedHeaders.entrySet()) {
                response.addHeader(entry.getKey(), entry.getValue());
            }
        }
    } else if (data instanceof Redirect) {
        Redirect redirect = (Redirect) data;
        for (Cookie cookie : redirect.getCookies()) {
            response.addCookie(cookie);
        }
        for (Entry<String, String> e : redirect.getHeaders().entrySet()) {
            response.addHeader(e.getKey(), e.getValue());
        }
        Map<String, Object> attributes = redirect.getAttributes();
        if (attributes != null) {
            String attributesValue = objectMapper.writeValueAsString(attributes);
            Cookie attributesCookie = new Cookie(CoreConstants.COOKIE_REDIRECT_ATTRIBUTES_NAME, EzyBase64.encodeUtf(attributesValue));
            attributesCookie.setMaxAge(CoreConstants.COOKIE_REDIRECT_ATTRIBUTES_MAX_AGE);
            response.addCookie(attributesCookie);
        }
        response.sendRedirect(redirect.getUri() + redirect.getQueryString());
        return;
    } else if (data instanceof View) {
        if (viewContext == null) {
            throw new IllegalStateException("viewContext is null, " + "you must add ezyhttp-server-thymeleaf to your dependencies" + " or create viewContext by yourself");
        }
        View view = (View) data;
        for (Cookie cookie : view.getCookies()) {
            response.addCookie(cookie);
        }
        for (Entry<String, String> e : view.getHeaders().entrySet()) {
            response.addHeader(e.getKey(), e.getValue());
        }
        response.setContentType(view.getContentType());
        viewContext.render(getServletContext(), request, response, view);
        return;
    } else {
        response.setStatus(HttpServletResponse.SC_OK);
    }
    if (body != null) {
        responseBody(response, body);
    }
}
Also used : Cookie(javax.servlet.http.Cookie) ResponseEntity(com.tvd12.ezyhttp.core.response.ResponseEntity) Entry(java.util.Map.Entry) Redirect(com.tvd12.ezyhttp.server.core.view.Redirect) HashMap(java.util.HashMap) Map(java.util.Map) MultiValueMap(com.tvd12.ezyhttp.core.data.MultiValueMap) View(com.tvd12.ezyhttp.server.core.view.View) MultiValueMap(com.tvd12.ezyhttp.core.data.MultiValueMap)

Aggregations

View (com.tvd12.ezyhttp.server.core.view.View)11 Test (org.testng.annotations.Test)10 HttpServletRequest (javax.servlet.http.HttpServletRequest)5 HttpServletResponse (javax.servlet.http.HttpServletResponse)5 TemplateResolver (com.tvd12.ezyhttp.server.core.view.TemplateResolver)4 ViewContext (com.tvd12.ezyhttp.server.core.view.ViewContext)4 ThymeleafViewContextBuilder (com.tvd12.ezyhttp.server.thymeleaf.ThymeleafViewContextBuilder)4 PrintWriter (java.io.PrintWriter)4 ServletContext (javax.servlet.ServletContext)4 ViewDecorator (com.tvd12.ezyhttp.server.core.view.ViewDecorator)2 Cookie (javax.servlet.http.Cookie)2 HttpMethod (com.tvd12.ezyhttp.core.constant.HttpMethod)1 MultiValueMap (com.tvd12.ezyhttp.core.data.MultiValueMap)1 ResponseEntity (com.tvd12.ezyhttp.core.response.ResponseEntity)1 DoGet (com.tvd12.ezyhttp.server.core.annotation.DoGet)1 UnhandledErrorHandler (com.tvd12.ezyhttp.server.core.handler.UnhandledErrorHandler)1 Redirect (com.tvd12.ezyhttp.server.core.view.Redirect)1 ViewDialect (com.tvd12.ezyhttp.server.core.view.ViewDialect)1 HashMap (java.util.HashMap)1 Locale (java.util.Locale)1