TDD范文

Test Driven Development

Controller测试

package io.github.talelin.latticy.controller.cms;

import cn.hutool.core.date.DateUtil;
import io.github.talelin.latticy.mapper.LogMapper;
import io.github.talelin.latticy.model.LogDO;
import lombok.extern.slf4j.Slf4j;
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.autoconfigure.web.servlet.AutoConfigureMockMvc;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.http.MediaType;
import org.springframework.test.annotation.Rollback;
import org.springframework.test.context.ActiveProfiles;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.result.MockMvcResultMatchers;
import org.springframework.transaction.annotation.Transactional;

import java.util.Date;

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

@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@Transactional // 数据操作后回滚
@Rollback
@AutoConfigureMockMvc
@Slf4j
@ActiveProfiles("test")
public class LogControllerTest {

    @Autowired
    private MockMvc mvc;

    @Autowired
    private LogMapper logMapper;

    @Before
    public void setUp() throws Exception {
        Date time = new Date();
        String permission = "查看lin的信息";
        String message = "就是个瓜皮";
        String method = "GET";
        String path = "/";
        Integer statusCode = 200;
        Integer userId = 1;
        String username = "pedro大大";

        LogDO logDO = LogDO
                .builder()
                .permission(permission)
                .message(message)
                .method(method)
                .statusCode(statusCode)
                .path(path)
                .userId(userId)
                .username(username)
                .build();
        logMapper.insert(logDO);
    }

    @Test
    public void getLogs() throws Exception {
        mvc.perform(get("/cms/log/").param("name", "pedro大大")
                .contentType(MediaType.APPLICATION_JSON))
                .andDo(print())
                .andExpect(status().isOk())
                .andExpect(MockMvcResultMatchers.jsonPath("$.total").isNumber())
                .andExpect(MockMvcResultMatchers.jsonPath("$.page").isNumber());

    }

    @Test
    public void getLogs1() throws Exception {
        mvc.perform(get("/cms/log/").param("name", "pedro")
                .contentType(MediaType.APPLICATION_JSON))
                .andDo(print())
                .andExpect(status().isOk())
                .andExpect(MockMvcResultMatchers.jsonPath("$.total").isNumber())
                .andExpect(MockMvcResultMatchers.jsonPath("$.page").isNumber())
                .andExpect(MockMvcResultMatchers.jsonPath("$.items").isEmpty());

    }

    @Test
    public void getLogs2() throws Exception {
        String yesterday = DateUtil.yesterday().toString("yyyy-MM-dd HH:mm:ss");
        String tomorrow = DateUtil.tomorrow().toString("yyyy-MM-dd HH:mm:ss");
        mvc.perform(get("/cms/log/")
                .param("name", "pedro")
                .param("start", yesterday)
                .param("end", tomorrow)
                .contentType(MediaType.APPLICATION_JSON))
                .andDo(print())
                .andExpect(status().isOk())
                .andExpect(MockMvcResultMatchers.jsonPath("$.total").isNumber())
                .andExpect(MockMvcResultMatchers.jsonPath("$.page").isNumber())
                .andExpect(MockMvcResultMatchers.jsonPath("$.items").isEmpty());

    }

    @Test
    public void getLogs3() throws Exception {
        String yesterday = DateUtil.yesterday().toString("yyyy-M-d HH:mm:ss");
        String tomorrow = DateUtil.tomorrow().toString("yyyy-M-dd HH:mm:ss");
        mvc.perform(get("/cms/log/")
                .param("name", "pedro")
                .param("start", yesterday)
                .param("end", tomorrow)
                .contentType(MediaType.APPLICATION_JSON))
                .andDo(print())
                .andExpect(status().isOk())
                .andExpect(MockMvcResultMatchers.jsonPath("$.total").isNumber())
                .andExpect(MockMvcResultMatchers.jsonPath("$.page").isNumber())
                .andExpect(MockMvcResultMatchers.jsonPath("$.items").isEmpty());

    }

    @Test
    public void searchLogs() throws Exception {
        mvc.perform(get("/cms/log/search")
                .param("name", "pedro大大")
                .param("keyword", "瓜皮")
                .contentType(MediaType.APPLICATION_JSON))
                .andDo(print())
                .andExpect(status().isOk())
                .andExpect(MockMvcResultMatchers.jsonPath("$.total").isNumber())
                .andExpect(MockMvcResultMatchers.jsonPath("$.items").isArray());
    }

    @Test
    public void searchLogs1() throws Exception {
        // yyyy-MM-dd HH:mm:ss
        String yesterday = DateUtil.yesterday().toString("yyyy-MM-dd HH:mm:ss");
        String tomorrow = DateUtil.tomorrow().toString("yyyy-MM-dd HH:mm:ss");
        log.info("{}, {}", yesterday, tomorrow);
        mvc.perform(get("/cms/log/search")
                .param("start", yesterday)
                .param("end", tomorrow)
                .contentType(MediaType.APPLICATION_JSON))
                .andDo(print())
                .andExpect(status().isOk())
                .andExpect(MockMvcResultMatchers.jsonPath("$.total").isNumber())
                .andExpect(MockMvcResultMatchers.jsonPath("$.items").isArray());
    }

    @Test
    public void getUsers() throws Exception {
        mvc.perform(get("/cms/log/users")
                .contentType(MediaType.APPLICATION_JSON))
                .andDo(print())
                .andExpect(status().isOk())
                .andExpect(MockMvcResultMatchers.
                        jsonPath("$.items").isArray());
    }
}