在Spring Boot环境中使用Mockito进行单元测试

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.http.ResponseEntity;
 
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.when;
 
@ExtendWith(MockitoExtension.class)
public class MyServiceTest {
 
    @Mock
    private MyRepository myRepository;
 
    @InjectMocks
    private MyService myService;
 
    @Test
    public void testFindById() {
        // 设置模拟行为
        when(myRepository.findById(1L)).thenReturn(ResponseEntity.ok("Mocked Data"));
 
        // 调用服务层方法
        ResponseEntity<String> result = myService.findById(1L);
 
        // 验证结果
        assertEquals("Mocked Data", result.getBody());
    }
}

这个代码示例展示了如何在Spring Boot项目中使用Mockito框架进行单元测试。我们模拟了MyRepository的findById方法，使得MyService的findById方法在测试时不会实际调用数据库，而是返回预设的模拟数据。这有助于提高测试速度和独立性，同时确保我们的服务层在数据源可用时也能正常工作。