Spring Boot中JUnit4测试含数据库依赖代码的问题咨询

你当前测试失败的核心原因是：虽然用Mockito创建了CountryDao的Mock对象，但没有为getAllCountries()方法设置预期返回值。Mockito默认会为mock对象的方法返回空值（比如列表返回空列表），所以getAllCountries()返回空列表，后续流过滤找不到匹配的Country，最终getCountryByName()返回null。

完全不需要实际数据库（包括内存数据库），只需用Mockito的when(...).thenReturn(...)方法，指定countryDao.getAllCountries()返回包含目标Country的列表即可。

修改后的测试代码：

import org.junit.Test;
import static org.mockito.Mockito.*;
import static org.junit.Assert.*;
import java.util.Arrays;

@Test
public void test_returns_country_by_name() {
    // 创建Mock的CountryDao
    CountryDao mockCountryDao = Mockito.mock(CountryDao.class);

    // 准备测试数据：构造包含"Canada"的Country列表
    Country canada = new Country();
    canada.setName("Canada");
    // 按需设置Country的其他属性
    when(mockCountryDao.getAllCountries()).thenReturn(Arrays.asList(canada));

    // 初始化服务类实例
    CountryServiceImpl countryService = new CountryServiceImpl(mockCountryDao);
    String targetName = "Canada";

    // 执行测试方法
    Country result = countryService.getCountryByName(targetName);

    // 断言验证结果
    assertNotNull(result);
    assertEquals(targetName, result.getName());
    // 可选：验证Dao的方法确实被调用了一次
    verify(mockCountryDao, times(1)).getAllCountries();
}

1. 测试边界与异常场景 #

除了正常情况，还需要覆盖边界场景，确保逻辑健壮：

@Test
public void test_returns_null_for_non_existent_country() {
    CountryDao mockCountryDao = Mockito.mock(CountryDao.class);
    // 返回仅包含USA的列表
    when(mockCountryDao.getAllCountries()).thenReturn(Arrays.asList(new Country("USA")));

    CountryServiceImpl countryService = new CountryServiceImpl(mockCountryDao);
    Country result = countryService.getCountryByName("Canada");

    assertNull(result);
}

@Test
public void test_ignores_case_when_matching_country_name() {
    CountryDao mockCountryDao = Mockito.mock(CountryDao.class);
    Country canada = new Country();
    canada.setName("Canada");
    when(mockCountryDao.getAllCountries()).thenReturn(Arrays.asList(canada));

    CountryServiceImpl countryService = new CountryServiceImpl(mockCountryDao);
    Country result = countryService.getCountryByName("canada");

    assertNotNull(result);
    assertEquals("Canada", result.getName());
}

2. 参数化测试 #

用JUnit4的@RunWith(Parameterized.class)批量测试多组输入，减少重复代码，提升测试覆盖率：

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import static org.mockito.Mockito.*;
import static org.junit.Assert.*;
import java.util.Arrays;
import java.util.Collection;

@RunWith(Parameterized.class)
public class CountryServiceParameterizedTest {
    private final String inputName;
    private final Country expectedResult;
    private CountryDao mockCountryDao;
    private CountryServiceImpl countryService;

    // 构造方法接收测试参数
    public CountryServiceParameterizedTest(String inputName, Country expectedResult) {
        this.inputName = inputName;
        this.expectedResult = expectedResult;
    }

    // 定义测试数据集
    @Parameterized.Parameters
    public static Collection<Object[]> testData() {
        Country canada = new Country();
        canada.setName("Canada");
        Country usa = new Country();
        usa.setName("USA");

        return Arrays.asList(new Object[][]{
                {"Canada", canada},
                {"canada", canada}, // 大小写忽略场景
                {"USA", usa},
                {"Mexico", null}    // 不存在的国家场景
        });
    }

    @Before
    public void setUp() {
        mockCountryDao = Mockito.mock(CountryDao.class);
        when(mockCountryDao.getAllCountries()).thenReturn(Arrays.asList(canada, usa));
        countryService = new CountryServiceImpl(mockCountryDao);
    }

    @Test
    public void test_get_country_by_name() {
        Country result = countryService.getCountryByName(inputName);
        if (expectedResult != null) {
            assertEquals(expectedResult.getName(), result.getName());
        } else {
            assertNull(result);
        }
    }
}

当前测试的错误在于未配置Mock对象的方法返回值，只要用Mockito指定getAllCountries()的返回数据，就能在完全不依赖数据库的情况下完成测试。通过覆盖边界场景、使用参数化测试，可以更全面地验证服务类的业务逻辑。

内容的提问来源于stack exchange，提问作者Abhinav