在Spring Boot中,我們可以使用Spring Boot Test模塊和JUnit來編寫單元測試。下面是一個簡單的示例,展示了如何為一個簡單的服務類編寫單元測試:
<!-- ...其他依賴... -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
<version>4.12</version>
<scope>test</scope>
</dependency>
</dependencies>
CalculatorService
:package com.example.demo.service;
import org.springframework.stereotype.Service;
@Service
public class CalculatorService {
public int add(int a, int b) {
return a + b;
}
}
CalculatorServiceTests
:package com.example.demo.service;
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 static org.junit.Assert.assertEquals;
@RunWith(SpringRunner.class)
@SpringBootTest
public class CalculatorServiceTests {
@Autowired
private CalculatorService calculatorService;
@Test
public void testAdd() {
int result = calculatorService.add(5, 3);
assertEquals("Addition of 5 and 3 failed", 8, result);
}
}
在這個例子中,我們使用了@RunWith(SpringRunner.class)
注解來運行測試,這樣我們就可以使用Spring Boot的自動配置特性。@SpringBootTest
注解表示這是一個Spring Boot測試,它會啟動一個內嵌的應用程序上下文,并將其與測試類關聯。
我們使用@Autowired
注解將CalculatorService
實例注入到測試類中。然后,我們編寫了一個名為testAdd
的測試方法,該方法使用@Test
注解標記。在這個方法中,我們調用calculatorService.add()
方法,并使用assertEquals()
方法驗證結果是否正確。
要運行此測試,只需在IDE中右鍵單擊測試類或方法,然后選擇“運行測試”或“運行所有測試”。測試應該成功通過,表明我們的CalculatorService
類按預期工作。