@Service
public class UserService {
    @Autowired
    private UserRepository userRepository;

    public String getUserName(Long id) {
        User user = userRepository.findById(id).orElseThrow();
        return user.getName();
    }
}

@SpringBootTest
public class UserServiceTest {

    @Autowired
    private UserService userService;

    @SpyBean  // 對 UserRepository 進(jìn)行 Spy，未 Mock 的方法仍然走真實(shí)邏輯
    private UserRepository userRepository;

    @Test
    public void testGetUserName() {
        // 準(zhǔn)備測試數(shù)據(jù)
        User mockUser = new User(1L, "Alice");
        userRepository.save(mockUser);  // 真實(shí)調(diào)用 save()

        // 只 Mock findById()，讓它返回自定義數(shù)據(jù)
        doReturn(Optional.of(new User(1L, "Mocked Alice")))
            .when(userRepository).findById(1L);

        // 調(diào)用測試方法
        String userName = userService.getUserName(1L);

        // 驗(yàn)證
        assertEquals("Mocked Alice", userName);

        // 其他方法仍然走真實(shí)邏輯
        User savedUser = userRepository.findById(2L).orElse(null);
        assertNull(savedUser);  // 因?yàn)闆]有存 ID=2 的用戶
    }
}

// 讓方法返回固定值
doReturn("Mocked Result").when(mySpyBean).someMethod();

// 帶參數(shù)匹配
doReturn("Hello").when(mySpyBean).greet(anyString());

doAnswer(invocation -> {
    String arg = invocation.getArgument(0);
    return "Processed: " + arg;
}).when(mySpyBean).process(anyString());

doThrow(new RuntimeException("DB Error"))
    .when(mySpyBean).save(any());

// 默認(rèn)會走真實(shí)方法，但也可以顯式聲明
doCallRealMethod().when(mySpyBean).someMethod();

// 驗(yàn)證 someMethod 被調(diào)用 1 次
verify(mySpyBean, times(1)).someMethod();