Spring Boot与JUnit中Optional的单元测试问题

针对你遇到的thenReturn无法正确返回Optional对象的问题，先给你明确正确的Mock写法，再讲完整的测试最佳实践：

一、正确的Mock返回Optional写法 #

假设你要测试的方法签名是这样的：

public interface UserService {
    Optional<User> findById(Long id);
}

用Mockito配合JUnit测试时，必须保证thenReturn传入的参数和方法返回类型完全匹配，正确示例：

// 模拟返回有值的Optional
when(userService.findById(1L)).thenReturn(Optional.of(new User("张三", 25)));

// 模拟返回空Optional
when(userService.findById(2L)).thenReturn(Optional.empty());

如果之前报错，大概率是这几个原因：

• Mock的方法签名和原方法不匹配（比如原方法返回Optional<User>，你却返回了User对象）
• 泛型类型不匹配（比如误传Optional<Object>替代Optional<User>）
• Mockito版本和Spring Boot版本不兼容（Spring Boot 3.x需搭配Mockito 4.x及以上版本）

二、测试Optional的最佳实践 #

1. 覆盖全场景测试 #

必须分别测试Optional有值和Optional为空两种分支，确保业务逻辑覆盖完整：

import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.springframework.boot.test.context.SpringBootTest;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.when;

@SpringBootTest
public class UserServiceTest {

    @Mock
    private UserService userService;

    @Test
    void findById_ShouldReturnUserWhenExists() {
        User expectedUser = new User("张三", 25);
        when(userService.findById(1L)).thenReturn(Optional.of(expectedUser));
        
        Optional<User> result = userService.findById(1L);
        
        assertTrue(result.isPresent());
        assertEquals(expectedUser, result.get());
    }

    @Test
    void findById_ShouldReturnEmptyWhenNotExists() {
        when(userService.findById(2L)).thenReturn(Optional.empty());
        
        Optional<User> result = userService.findById(2L);
        
        assertFalse(result.isPresent());
    }
}

2. 用更简洁的断言工具（推荐AssertJ） #

如果引入AssertJ依赖，断言代码会更流畅易读，避免直接用get()（空Optional调用get()会抛异常）：

import static org.assertj.core.api.Assertions.assertThat;

// ...

@Test
void findById_ShouldReturnUserWhenExists() {
    User expectedUser = new User("张三", 25);
    when(userService.findById(1L)).thenReturn(Optional.of(expectedUser));
    
    Optional<User> result = userService.findById(1L);
    
    assertThat(result).contains(expectedUser);
}

@Test
void findById_ShouldReturnEmptyWhenNotExists() {
    when(userService.findById(2L)).thenReturn(Optional.empty());
    
    Optional<User> result = userService.findById(2L);
    
    assertThat(result).isEmpty();
}

3. 避免不必要的get()调用 #

如果业务逻辑中需要处理Optional，测试时尽量模拟真实调用场景，比如用ifPresent验证：

@Test
void handleUser_ShouldProcessWhenUserExists() {
    User testUser = new User("张三", 25);
    when(userService.findById(1L)).thenReturn(Optional.of(testUser));
    
    // 假设业务方法是接收Optional后处理
    userService.handleUser(userService.findById(1L));
    
    // 验证处理逻辑被触发（比如用verify）
    verify(someProcessor).process(testUser);
}

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