Examples with ExampleServlet cn.taketoday.test.web.servlet.ExampleServlet used on opensource projects

Example 21 with ExampleServlet

use of cn.taketoday.test.web.servlet.ExampleServlet in project today-infrastructure by TAKETODAY.

the class AbstractServletWebServerFactoryTests method serverHeaderIsDisabledByDefaultWhenUsingSsl.

@Test
void serverHeaderIsDisabledByDefaultWhenUsingSsl() throws Exception {
    AbstractServletWebServerFactory factory = getFactory();
    factory.setSsl(getSsl(null, "password", "src/test/resources/test.jks"));
    this.webServer = factory.getWebServer(new ServletRegistrationBean<>(new ExampleServlet(true, false), "/hello"));
    this.webServer.start();
    SSLConnectionSocketFactory socketFactory = new SSLConnectionSocketFactory(new SSLContextBuilder().loadTrustMaterial(null, new TrustSelfSignedStrategy()).build());
    HttpClient httpClient = this.httpClientBuilder.get().setSSLSocketFactory(socketFactory).build();
    ClientHttpResponse response = getClientResponse(getLocalUrl("https", "/hello"), HttpMethod.GET, new HttpComponentsClientHttpRequestFactory(httpClient));
    assertThat(response.getHeaders().get("Server")).isNullOrEmpty();
}

Example 22 with ExampleServlet

use of cn.taketoday.test.web.servlet.ExampleServlet in project today-infrastructure by TAKETODAY.

the class AbstractServletWebServerFactoryTests method sslDisabled.

@Test
void sslDisabled() throws Exception {
    AbstractServletWebServerFactory factory = getFactory();
    Ssl ssl = getSsl(null, "password", "classpath:test.jks");
    ssl.setEnabled(false);
    factory.setSsl(ssl);
    this.webServer = factory.getWebServer(new ServletRegistrationBean<>(new ExampleServlet(true, false), "/hello"));
    this.webServer.start();
    SSLConnectionSocketFactory socketFactory = new SSLConnectionSocketFactory(new SSLContextBuilder().loadTrustMaterial(null, new TrustSelfSignedStrategy()).build());
    HttpClient httpClient = HttpClients.custom().setSSLSocketFactory(socketFactory).build();
    HttpComponentsClientHttpRequestFactory requestFactory = new HttpComponentsClientHttpRequestFactory(httpClient);
    assertThatExceptionOfType(SSLException.class).isThrownBy(() -> getResponse(getLocalUrl("https", "/hello"), requestFactory));
}

Example 23 with ExampleServlet

use of cn.taketoday.test.web.servlet.ExampleServlet in project today-infrastructure by TAKETODAY.

the class AbstractServletWebServerFactoryTests method assertForwardHeaderIsUsed.

protected void assertForwardHeaderIsUsed(ServletWebServerFactory factory) throws IOException, URISyntaxException {
    this.webServer = factory.getWebServer(new ServletRegistrationBean<>(new ExampleServlet(true, false), "/hello"));
    this.webServer.start();
    assertThat(getResponse(getLocalUrl("/hello"), "X-Forwarded-For:140.211.11.130")).contains("remoteaddr=140.211.11.130");
}

Example 24 with ExampleServlet

use of cn.taketoday.test.web.servlet.ExampleServlet in project today-infrastructure by TAKETODAY.

the class AbstractServletWebServerFactoryTests method sslWithInvalidAliasFailsDuringStartup.

@Test
void sslWithInvalidAliasFailsDuringStartup() {
    AbstractServletWebServerFactory factory = getFactory();
    Ssl ssl = getSsl(null, "password", "test-alias-404", "src/test/resources/test.jks");
    factory.setSsl(ssl);
    ServletRegistrationBean<ExampleServlet> registration = new ServletRegistrationBean<>(new ExampleServlet(true, false), "/hello");
    ThrowingCallable call = () -> factory.getWebServer(registration).start();
    assertThatSslWithInvalidAliasCallFails(call);
}

Example 25 with ExampleServlet

use of cn.taketoday.test.web.servlet.ExampleServlet in project today-infrastructure by TAKETODAY.

the class AbstractServletWebServerFactoryTests method sessionServletRegistration.

protected final ServletContextInitializer sessionServletRegistration() {
    ServletRegistrationBean<ExampleServlet> bean = new ServletRegistrationBean<>(new ExampleServlet() {

        @Override
        public void service(ServletRequest request, ServletResponse response) throws IOException {
            HttpSession session = ((HttpServletRequest) request).getSession(true);
            long value = System.currentTimeMillis();
            Object existing = session.getAttribute("boot");
            session.setAttribute("boot", value);
            PrintWriter writer = response.getWriter();
            writer.append(String.valueOf(existing)).append(":").append(String.valueOf(value));
        }
    }, "/session");
    bean.setName("session");
    return bean;
}