使用 Mockito 在 Spring Boot 中测试服务层

Question

我正在尝试使用通过存储库层连接到数据库的 mockito 对我的服务层进行单元测试。

我的测试用例：

@InjectMocks
private EmployeeService employeeService;

@Mock
private EmployeeRepository employeeRepository;     

@Test
public void getActionFromEmployeeIdTest() throws ActionException {
 //setup     
 when(employeeRepository.getActionsByEmployeeId(anyLong()).thenReturn(EmployeeEntity);

 // exercise
 List<Employee> result = employeeService.getActionsByEmployeeIdService(101);

 //verify
 assertEqual(EmployeeEntity, result);
}

服务层：

@Service
public class EmployeeService {
    @Override
    public List<EmployeeUser> getActionsByEmployeeIdService(long employeeId)  {
        Employee employee = employeeRepository.findByEmployeeIdId(employeeId);
        List<EmployeeUser> actions = employeeUserRepository.getActionsByEmployeeId(employeeId);
        return actions;
}

存储层：

@Repository
public interface EmployeeUserRepository extends JpaRepository<EmployeeUser,Long> {
    @Transactional
    @Query(value = "Select e from EmployeeUser e where e.employeeId = :employeeId" )
    List<EmployeeUser> getActionsByEmployeeId(@Param("employeeId") long employeeId);
}

我将@InjectMocks 用于employeeService 和@Mock 用于employeeRepository。

result 似乎返回 null。无论如何我可以在我的测试中返回一个非空值吗？

Answer 1

您正在使用 employeeRepository 的引用来调用 getActionssByEmployeeId(anyLong()) 方法。

when(employeeRepository.getActionssByEmployeeId(anyLong()).thenReturn(EmployeeEntity);

但是在服务类中，这个方法被 employeeUserRepository

调用

List<EmployeeUser> actions = employeeUserRepository.getActionsByEmployeeIdId(employeeId);

请检查一下。