org.springframework.security.htmlunit.server.WebTestClientHtmlUnitDriverBuilderTests.java Source code

Introduction

Here is the source code for org.springframework.security.htmlunit.server.WebTestClientHtmlUnitDriverBuilderTests.java
Source

/*
 * Copyright 2002-2017 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.security.htmlunit.server;

import org.junit.Test;
import org.openqa.selenium.WebDriver;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseCookie;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.security.web.util.TextEscapeUtils;
import org.springframework.stereotype.Controller;
import org.springframework.test.web.reactive.server.WebTestClient;
import org.springframework.web.bind.annotation.CookieValue;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import reactor.core.publisher.Mono;

import java.net.URI;
import java.time.Duration;

import static org.assertj.core.api.Assertions.assertThat;

/**
 * @author Rob Winch
 * @since 5.0
 */
public class WebTestClientHtmlUnitDriverBuilderTests {

    @Test
    public void helloWorld() {
        WebTestClient webTestClient = WebTestClient.bindToController(new HelloWorldController()).build();
        WebDriver driver = WebTestClientHtmlUnitDriverBuilder.webTestClientSetup(webTestClient).build();

        driver.get("http://localhost/");

        assertThat(driver.getPageSource()).contains("Hello World");
    }

    /**
     * @author Rob Winch
     * @since 5.0
     */
    @Controller
    class HelloWorldController {
        @ResponseBody
        @GetMapping(produces = MediaType.TEXT_HTML_VALUE)
        public String index() {
            return "<html>\n" + "<head>\n" + "<title>Hello World</title>\n" + "</head>\n" + "<body>\n"
                    + "<h1>Hello World</h1>\n" + "</body>\n" + "</html>";
        }
    }

    @Test
    public void cookies() {
        WebTestClient webTestClient = WebTestClient.bindToController(new CookieController()).build();
        WebDriver driver = WebTestClientHtmlUnitDriverBuilder.webTestClientSetup(webTestClient).build();

        driver.get("http://localhost/cookie");

        assertThat(driver.getPageSource()).contains("theCookie");

        driver.get("http://localhost/cookie/delete");

        assertThat(driver.getPageSource()).contains("null");
    }

    @Controller
    @ResponseBody
    class CookieController {
        @GetMapping(path = "/", produces = MediaType.TEXT_HTML_VALUE)
        public String view(@CookieValue(required = false) String cookieName) {
            return "<html>\n" + "<head>\n" + "<title>Hello World</title>\n" + "</head>\n" + "<body>\n" + "<h1>"
                    + TextEscapeUtils.escapeEntities(cookieName) + "</h1>\n" + "</body>\n" + "</html>";
        }

        @GetMapping("/cookie")
        public Mono<Void> setCookie(ServerHttpResponse response) {
            response.addCookie(ResponseCookie.from("cookieName", "theCookie").build());
            return redirect(response);
        }

        private Mono<Void> redirect(ServerHttpResponse response) {
            response.setStatusCode(HttpStatus.MOVED_PERMANENTLY);
            response.getHeaders().setLocation(URI.create("/"));
            return response.setComplete();
        }

        @GetMapping("/cookie/delete")
        public Mono<Void> deleteCookie(ServerHttpResponse response) {
            response.addCookie(ResponseCookie.from("cookieName", "").maxAge(Duration.ofSeconds(0)).build());
            return redirect(response);
        }
    }
}