Spring MVC : Unit Test using MockMVC

ชนใดเขียนโปรแกรมไม่มี Unit Test ในสันดานเป็นคนชอบกลนัก ฮ่าๆๆๆ อึ้มๆๆๆๆๆ ไม่ขอเอ่ย สภาพพพพพพพ

ไม่พูดพร่ำทำเพลง เราจะใช้ Project ก่อนหน้านี้ที่

Spring Boot | Spring Data JPA | Pagination and Sorting

อย่าลืม Add Junit เข้าไปใน pom นะครับ

<dependency>
   <groupId>junit</groupId>
   <artifactId>junit</artifactId>
   <version>4.13.2</version>
   <scope>test</scope>
</dependency>

เราจะมาสร้าง Unit Test ให้กับ CustomerController กันครับ

package com.zencode.jpa.jpaexample.controller;

import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.zencode.jpa.jpaexample.entity.Customer;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;

import java.util.List;

import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;

@RunWith(SpringRunner.class)
@SpringBootTest
public class CustomerControllerTest {

    private MockMvc mockMvc;

    @Autowired
    private WebApplicationContext context;

    private static ObjectMapper objectMapper = new ObjectMapper();

    @Before
    public void setUp() {
        mockMvc = MockMvcBuilders.webAppContextSetup(context).build();
    }

    @Test
    public void getAllCustomerTest() throws Exception {
        MvcResult mvcResult = mockMvc.perform(get("/customers"))
                .andExpect(status().isOk()).andReturn();
        String contentAsString = mvcResult.getResponse().getContentAsString();
        Assert.assertNotNull(contentAsString);

        List<Customer> customers = objectMapper.readValue(contentAsString, new TypeReference<List<Customer>>() {});

        Assert.assertEquals(299, customers.size());
        /**
         * อาจจะตรวจสอบช้อมุลที่เราต้องการตรวจสอบ
         */
    }

}

รัน Unit Test

นอกจากนี้เราก็ยัง POST, PUT, DELETE ได้

เช่น POST เราอาจจะมี Body ไป ด้วยเช่น

@Test
public void addNewCustomerTest() throws Exception {
    Customer customer = new ("Chiwa Kantawong", 1000000);
    String json  objectMapper.writeValueAsString(customer);
    MvcResult mvcResult = mockMvc.perform(
            post("/customers").content(json).contentType(MediaType.APPLICATION_JSON_VALUE)
    ).andExpect(status().isOk()).andReturn();
}

จัดตัวอย่างง่ายๆ กันไปก่อนนะครับ