在Spring-Boot中进行单元测试

要进行单元测试,需要引入依赖:

<dependency>
   <groupId>org.springframework.boot</groupId>
   <artifactId>spring-boot-starter-test</artifactId>
   <scope>test</scope>
</dependency>

设有一个Controller,其路径为src/main/java/com/template/controller/HelloWorldController:

package com.template.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class HelloWorldController {

    @RequestMapping("/HelloWorld")
    public JSONObject index() {
        JSONObject r = new JSONObject();
        ((JSONObject) r).put("name", "tesla");
        ((JSONObject) r).put("age", 88);
        return r;
    }
}

现在要对该Controller编写单元测试:

1.在路径src/test/java/com/template下,创建对应的controller文件夹,并在其下添加HelloWorldControllerTest.java。

2.编写测试代码:

package com.template.controller;

import com.alibaba.fastjson.JSON;
import com.template.TemplateApplication;
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.reactive.AutoConfigureWebTestClient;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

@RunWith(SpringRunner.class)
@SpringBootTest
public class HelloWorldControllerTest {
    @Autowired
    HelloWorldController hwController;

    // 初始化
    @Before
    public void setUp() throws Exception {
        System.out.println("执行初始化");
    }

    // 测试
    @Test
    public void test() throws Exception {
        JSONObject r = hwController.index();
        System.out.println(r.toJSONString());
    }
}

其中:

  • 类名为HelloWorldControllerTest,也就是HelloWorldController的类名后添加一个Test,这不是强制的,可以看作一种习惯。类名前要添加@RunWith(SpringRunner.class)@SpringBootTest
  • 使用@Autowired来引入HelloWorldController变量。这样就可以在成员函数中调用HelloWorldController的各个函数。
  • @Before注解用于初始化。因为测试之前很可能需要设定一些变量等,可以在该函数里执行。测试函数运行时,会先执行@Before注解函数。
  • 测试函数使用@Test注解。在该函数中调用HelloWorldController变量来执行某些操作即可。

3.在测试函数test()函数上点右键,选择Run或者Debug。

即可看到控制台打印的结果。

 

通常,使用System.out.println()将测试需要查看的值打印出来,或者直接加断点,就已经可以满足测试需求了。然而为了更高效的测试,还可以使用两种方式:

  1. Assert.assertThat()断言。
  2. Mock
 

Assert.assertThat()断言

格式:

Assert.assertThat(需测试对象, 测试条件)

即传入两个参数,一个是需要测试的对象,另一个是测试的条件。

例如:

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.template.TemplateApplication;
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.autoconfigure.web.reactive.AutoConfigureWebTestClient;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import static org.hamcrest.core.Is.is;

@RunWith(SpringRunner.class)
@SpringBootTest
public class HelloWorldControllerTest {
    @Autowired
    HelloWorldController hwController;

    // 初始化
    @Before
    public void setUp() throws Exception {
        System.out.println("执行初始化");
    }

    // 测试
    @Test
    public void test() throws Exception {
        JSONObject r = hwController.index();
        Assert.assertThat(r.getString("name"),is("tesla"));
    }
}

若测试通过,会看到提示:

若测试没有通过:

这里判断条件使用了is()函数。

判断条件所用的函数非常多:

1.   assertThat( testedNumber, allOf( greaterThan(8), lessThan(16) ) );

allOf传入多个条件,所有条件必须都成立测试才通过,相当于“与”(&&)。

2.   assertThat( testedNumber, anyOf( greaterThan(16), lessThan(8) ) );

anyOf传入多个条件,所有条件只要有一个成立则测试通过,相当于“或”(||)。
3.   assertThat( testedNumber, anything() );

anything传入一个String或者不传,无论传入的是什么,永远为true。
4.   assertThat( testedString, is( "developerWorks" ) );

is若前面待测的object等于后面给出的object,则测试通过。
5.   assertThat( testedString, not( "developerWorks" ) );

not与is匹配符相反,若前面待测的object不等于后面给出的object,则测试通过。

除此之外还有很多其他用于字符串,数值,iterator,map等的判断条件。

 

Mock

对一个Controller进行测试,通常需要:启动服务→打开浏览器/测试工具→发出请求→收到Controller的返回数据并打印。而且测试一个Controller,必须将整个工程全部加载。

Mock可以创建一个虚拟的对象来代替那些不易构造或不易获取的对象,通常就是request对象。并且可以只测试当前Controller,且不需要启动Tomcat等服务。相比于常规的调试,Mock测试更加便捷。

Mock为Spring自带,不需要额外引入依赖库。

例如:

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.http.MediaType;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.RequestBuilder;
import org.springframework.test.web.servlet.ResultActions;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.result.MockMvcResultHandlers;
import org.springframework.test.web.servlet.result.MockMvcResultMatchers;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;

@RunWith(SpringRunner.class)
@SpringBootTest
public class HelloWorldControllerTest {
    private MockMvc mockMvc; 
    @Autowired
    private HelloWorldController helloWorldController;

    // 初始化
    @Before
    public void setUp() throws Exception {
        System.out.println("执行初始化");
        mockMvc = MockMvcBuilders.standaloneSetup(helloWorldController
).build();
    }

    // 测试
    @Test
    public void test() throws Exception {
        RequestBuilder request = MockMvcRequestBuilders.get("/HelloWorld")
                .contentType(MediaType.APPLICATION_JSON)    //发送所用的数据格式
                .accept(MediaType.APPLICATION_JSON) //接收所使用的数据格式
                .param("id","201801");  //附加参数

        // 执行请求
        ResultActions result = mockMvc.perform(request);
        // 分析结果
        result.andExpect(MockMvcResultMatchers.status().isOk()) // 执行状态
                .andExpect(MockMvcResultMatchers.jsonPath("name").value("Tesla")) // 期望值
                .andDo(MockMvcResultHandlers.print()) // 打印
                .andReturn();   // 返回
    }
}

如上,Mock的使用过程为:

1.定义MockMvc变量和Controller类变量:

private MockMvc mockMvc;
@Autowired
private HelloWorldController helloWorldController;

2.在修饰的初始化函数中,创建MockMvc对象:

@Before
public void setUp() throws Exception {
    System.out.println("执行初始化");
    mockMvc = MockMvcBuilders.standaloneSetup(helloWorldController).build();
}

创建MockMvc对象时,需要传入1中定义的Controller对象。

注意该传入的Controller对象必须使用 1中的方式以@Autowired引入,而不可以如下:

mockMvc = MockMvcBuilders.standaloneSetup(new HelloWorldController()).build();

使用上面这种方式,虽然可以得到Controller对象,且其成员函数都可以正常访问,但Controller中以@Autowired引入的各种对象却无法初始化,导致这些对象的值为null。

最典型的就是Service。Controller通常并不会承担过多的逻辑,而是将其交给Service,于是在Controller中往往会以@Autowired的形式引入Service。然而,若使用上面new的方式来创建Mock对象,会导致Controller中引入的Service对象无法注入,从而为null。执行测试,控制台会报空指针错误。

3.创建请求

RequestBuilder request = MockMvcRequestBuilders.get("/HelloWorld")
        .contentType(MediaType.APPLICATION_JSON)    //发送所用的数据格式
        .accept(MediaType.APPLICATION_JSON) //接收所使用的数据格式
        .param("id","201801");  //附加参数

调用的请求为get,同时传入接口名。

  • 可以将get换为post,或者其他请求方式。
  • contentType和accept定义了发送和接收用的数据格式,这里设置为JSON。
  • param用于附加1个参数,每多1个参数,都可以多调用一次param来附加。无论是get还是post,都可以使用param来附加参数。附加后的参数会填充到请求的parameters字段中。

4.执行请求

ResultActions result = mockMvc.perform(request);

ResultActions一共有3个接口:

public interface ResultActions {
    ResultActions andExpect(ResultMatcher var1) throws Exception;

    ResultActions andDo(ResultHandler var1) throws Exception;

    MvcResult andReturn();
}
  • andExpect():用于测试是否为期望值。
  • andDo():用于执行某个指定的操作。
  • andReturn():用于返回执行结果,为MvcResul类型。

5.分析结果

MvcResult mvcResult = result.andExpect(MockMvcResultMatchers.status().isOk()) // 执行状态
                    .andExpect(MockMvcResultMatchers.jsonPath("name").value("Tesla")) // 期望值
                    .andDo(MockMvcResultHandlers.print()) // 打印
                    .andReturn();   // 返回

可以看到分析结果使用了4中ResultActions的函数。

其中使用MockMvcResultMatchers.jsonPath()可以获取到一个JsonPathResultMatchers,可以用于判断request请求的返回值是否符合期望。
若要自行解析返回结果,则可以:

String resultStr = mvcResult.getResponse().getContentAsString();

然后将resultStr转为JSON即可。
实际测试过程中,往往是Assert.seertThath()与Mock配合使用。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值