java-tutorials/spring-5-reactive/src/test/java/com/baeldung/functional/FunctionalWebApplicationIntegrationTest.java

package com.baeldung.functional;

import static org.springframework.web.reactive.function.BodyInserters.fromObject;
import static org.springframework.web.reactive.function.BodyInserters.fromResource;

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Ignore;
import org.junit.Test;
import org.springframework.boot.web.server.WebServer;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import org.springframework.http.MediaType;
import org.springframework.security.test.context.support.WithMockUser;
import org.springframework.test.web.reactive.server.WebTestClient;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.reactive.function.BodyInserters;

public class FunctionalWebApplicationIntegrationTest {

    private static WebTestClient client;
    private static WebServer server;

    @BeforeClass
    public static void setup() throws Exception {
        server = new FunctionalWebApplication().start();
        client = WebTestClient.bindToServer()
            .baseUrl("http://localhost:" + server.getPort())
            .build();
    }

    @AfterClass
    public static void destroy() {
        server.stop();
    }

    @Test
    public void givenRouter_whenGetTest_thenGotHelloWorld() throws Exception {
        client.get()
            .uri("/test")
            .exchange()
            .expectStatus()
            .isOk()
            .expectBody(String.class)
            .isEqualTo("helloworld");
    }

    @Test
    public void givenIndexFilter_whenRequestRoot_thenRewrittenToTest() throws Exception {
        client.get()
            .uri("/")
            .exchange()
            .expectStatus()
            .isOk()
            .expectBody(String.class)
            .isEqualTo("helloworld");
    }

    @Test
    public void givenLoginForm_whenPostValidToken_thenSuccess() throws Exception {
        MultiValueMap<String, String> formData = new LinkedMultiValueMap<>(1);
        formData.add("user", "baeldung");
        formData.add("token", "you_know_what_to_do");

        client.post()
            .uri("/login")
            .contentType(MediaType.APPLICATION_FORM_URLENCODED)
            .body(BodyInserters.fromFormData(formData))
            .exchange()
            .expectStatus()
            .isOk()
            .expectBody(String.class)
            .isEqualTo("welcome back!");
    }

    @Test
    public void givenLoginForm_whenRequestWithInvalidToken_thenFail() throws Exception {
        MultiValueMap<String, String> formData = new LinkedMultiValueMap<>(2);
        formData.add("user", "baeldung");
        formData.add("token", "try_again");

        client.post()
            .uri("/login")
            .contentType(MediaType.APPLICATION_FORM_URLENCODED)
            .body(BodyInserters.fromFormData(formData))
            .exchange()
            .expectStatus()
            .isBadRequest();
    }

    @Test
    public void givenUploadForm_whenRequestWithMultipartData_thenSuccess() throws Exception {
        Resource resource = new ClassPathResource("/baeldung-weekly.png");
        client.post()
            .uri("/upload")
            .contentType(MediaType.MULTIPART_FORM_DATA)
            .body(fromResource(resource))
            .exchange()
            .expectStatus()
            .isOk()
            .expectBody(String.class)
            .isEqualTo(String.valueOf(resource.contentLength()));
    }

    @Ignore("We get 404 after Spring Boot 2.4 upgrade. We need to solve it in a new task.")
    @Test
    public void givenActors_whenAddActor_thenAdded() throws Exception {
        client.get()
            .uri("/actor")
            .exchange()
            .expectStatus()
            .isOk()
            .expectBodyList(Actor.class)
            .hasSize(2);

        client.post()
            .uri("/actor")
            .body(fromObject(new Actor("Clint", "Eastwood")))
            .exchange()
            .expectStatus()
            .isOk();

        client.get()
            .uri("/actor")
            .exchange()
            .expectStatus()
            .isOk()
            .expectBodyList(Actor.class)
            .hasSize(3);
    }

    @Test
    public void givenResources_whenAccess_thenGot() throws Exception {
        client.get()
            .uri("/files/hello.txt")
            .exchange()
            .expectStatus()
            .isOk()
            .expectBody(String.class)
            .isEqualTo("hello");
    }

}