Springboot教程（五）——单元测试

idea中一般使用JUnit进行单元测试 

基本使用

我们可以在idea的test文件夹下的XXXXApplicationTests内进行单元测试：

可以在@Test标注的方法上写测试代码： 

@SpringBootTest
class C0101ApplicationTests {@Testfun contextLoads() {println("Hello World")}}

我们也可以写多个测试方法：

@SpringBootTest
class C0101ApplicationTests {@Testfun test1() {println("test1")}@Testfun test2() {println("test2")}}

我们也可以在测试类内使用@Autowired注解，如我们可以自动注入写好的服务：

@Autowired
lateinit var testService: TestService

---

我们来举个例子，先创建一个服务：

package com.example.c0101.serviceimport org.springframework.stereotype.Service@Service
class TestService {fun check(username: String, password: String): Boolean{return username == "admin" && password == "123456"}
}

然后在测试类内使用Autowired自动注入服务，并进行测试：

package com.example.c0101import com.example.c0101.service.TestService
import org.junit.jupiter.api.Test
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.boot.test.context.SpringBootTest@SpringBootTest
class C0101ApplicationTests {@Autowiredlateinit var testService: TestService@Testfun test() {println(testService.check("111", "123"))println(testService.check("admin", "123456"))}}

控制台输出：

...
false
true
...

---

测前准备和测后收尾

我们可以用以下注解实现测前准备和测后收尾：

• @BeforeEach：在每一个测试方法执行前执行，其标注的方法可以传入一个TestInfo类型的参数，为当前测试信息的对象
• @AfterEach：在每一个测试方法执行后执行，其标注的方法可以传入一个TestInfo类型的参数，为当前测试信息的对象
• @BeforeAll：在所有测试方法执行前只执行一次
• @AfterAll：在所有测试方法执行后只执行一次

另外，@BeforeAll和@AfterAll标注的方法需要为静态，在kotlin中需要放在companion object的代码块下，并用@JvmStatic注解标注

---

以下代码展示了这些注解的用法：

package com.example.c0101import com.example.c0101.service.TestService
import org.junit.jupiter.api.AfterAll
import org.junit.jupiter.api.AfterEach
import org.junit.jupiter.api.BeforeAll
import org.junit.jupiter.api.BeforeEach
import org.junit.jupiter.api.Test
import org.junit.jupiter.api.TestInfo
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.boot.test.context.SpringBootTest@SpringBootTest
class C0101ApplicationTests {@BeforeEachfun beforeEach(info: TestInfo){println("即将进入测试方法：${info.testMethod.get()}")}@AfterEachfun afterEach(info: TestInfo){println("已经离开测试方法：${info.testMethod.get()}")}companion object {@JvmStatic@BeforeAllfun beforeAll(){println("即将进入测试")}@JvmStatic@AfterAllfun afterAll(){println("测试已完成")}}@Testfun test() {println("Hello World")}}

控制台输出：

...
即将进入测试
...
即将进入测试方法：public void com.example.c0101.C0101ApplicationTests.test()
Hello World
已经离开测试方法：public void com.example.c0101.C0101ApplicationTests.test()
测试已完成
...

---

设置测试用例

要想设置测试用例，需要使用@ParameterizedTest注解，该注解可以传入name参数，可以为测试方法起别名。另外，可以用@ValueSource注解设置参数源：

@ParameterizedTest
@ValueSource(ints = [1, 2, 3, 4, 5])
fun test(num: Int) {println("$num")
}

注意，被@ParameterizedTest注解标注的测试方法就不需要用@Test注解标注了

JUnit会将所有的测试用例都测试一遍，因此这个测试方法会被执行5次：

...
1
2
3
4
5
...

我们也可以用@MethodSource注解设置测试用例，它将会把一个静态方法的返回值作为测试用例：

companion object{@JvmStaticfun getInt(): Stream<Int>{return Stream.of(1, 2, 3, 4, 5)}
}@ParameterizedTest
@MethodSource("getInt")
fun test(num: Int) {println("$num")
}

注意：这里面的Stream是java.util.stream下的Stream类

使用这种方法，我们就可以传入多个参数了：

companion object{@JvmStaticfun getProducts(): Stream<Arguments>{return Stream.of(Arguments.of("鼠标", 49.9),Arguments.of("键盘", 59.9))}
}@ParameterizedTest
@MethodSource("getProducts")
fun test(name: String, price: Double) {println("$name 卖 $price 元")
}

输出：

...
鼠标 卖 49.9 元
键盘 卖 59.9 元
...

断言

测试人员可以断言一件事是真的，如果这件事不是真的，则测试失败

JUnit提供了Assertions类，用于进行断言：

@Test
fun test() {Assertions.assertTrue(1 > 2)
}

这段代码断言了1>2是真的，如果不是真的（当然不是真的），则测试失败：

标题

断言的应用

还记得之前的测试服务的代码吗，这个服务在传入用户名为admin且密码为123456后应该返回true，如果返回的不是true，说明这个服务写错了；同理，如果传入的用户名不是admin或密码不是123456，则应该返回false，如果返回的不是false，同样说明这个服务写错了。我们可以用断言来测试这个功能：

@Autowired
lateinit var testService: TestService@Test
fun test() {Assertions.assertTrue(testService.check("admin", "123456"))Assertions.assertTrue(!testService.check("aaa", "123"))
}

可以看到，测试通过，说明check没有写错

模拟Servlet对象

如果要测试controller等需要使用Servlet对象（例如HttpServletRequest）的方法，就需要模拟Servlet对象，我们可以在测试类自动注入以下对象模拟：

@Autowired
lateinit var mockHttpServletRequest: MockHttpServletRequest
@Autowired
lateinit var mockHttpServletResponse: MockHttpServletResponse
@Autowired
lateinit var mockHttpSession: MockHttpSession

这些代码在idea里可能会报错，不过没有关系

另外，在测试contoller时，同样需要@Autowired：

@Autowired
lateinit var controller: TestController

---

我们来举个模拟Servlet的例子：

先创建一个controller：

package com.example.c0101.controllerimport jakarta.servlet.http.HttpServletRequest
import jakarta.servlet.http.HttpServletResponse
import jakarta.servlet.http.HttpSession
import org.springframework.web.bind.annotation.RequestMapping
import org.springframework.web.bind.annotation.RestController@RestController
class TestController {@RequestMappingfun index(request: HttpServletRequest, response: HttpServletResponse, session: HttpSession): String{response.status = 404return "Hello World"}}

当访问主页时，会设置状态码为404，并返回Hello World

接下来编写测试类：

package com.example.c0101import com.example.c0101.controller.TestController
import com.example.c0101.service.TestService
import org.junit.jupiter.api.Assertions
import org.junit.jupiter.api.Test
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.boot.test.context.SpringBootTest
import org.springframework.mock.web.MockHttpServletRequest
import org.springframework.mock.web.MockHttpServletResponse
import org.springframework.mock.web.MockHttpSession@SpringBootTest
class C0101ApplicationTests {@Autowiredlateinit var controller: TestController@Autowiredlateinit var mockHttpServletRequest: MockHttpServletRequest@Autowiredlateinit var mockHttpServletResponse: MockHttpServletResponse@Autowiredlateinit var mockHttpSession: MockHttpSession@Testfun test() {val res = controller.index(mockHttpServletRequest, mockHttpServletResponse, mockHttpSession)Assertions.assertTrue(mockHttpServletResponse.status == 404)println(res)}}

测试类中，我们断言了状态码一定是404，并输出了返回结果

控制台输出如下：

...
Hello World
...