溫馨提示×

Spring Boot中的單元測試如何編寫

小樊
85
2024-08-30 04:19:00
欄目: 編程語言

在Spring Boot中，我們可以使用Spring Boot Test模塊和JUnit來編寫單元測試。下面是一個簡單的示例，展示了如何為一個簡單的服務類編寫單元測試：

  1. 首先，添加Spring Boot Test和JUnit依賴到項目的pom.xml文件中：
    <!-- ...其他依賴... -->
   <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>
  1. 創(chuàng)建一個簡單的服務類，例如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;
    }
}
  1. 編寫單元測試類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測試，它會啟動一個內嵌的應用程序上下文，并將其與測試類關聯(lián)。

我們使用@Autowired注解將CalculatorService實例注入到測試類中。然后，我們編寫了一個名為testAdd的測試方法，該方法使用@Test注解標記。在這個方法中，我們調用calculatorService.add()方法，并使用assertEquals()方法驗證結果是否正確。

要運行此測試，只需在IDE中右鍵單擊測試類或方法，然后選擇“運行測試”或“運行所有測試”。測試應該成功通過，表明我們的CalculatorService類按預期工作。

0 踩