根据这段代码帮我写一份单元测试: public ApiResult getProByEnterpriseIds(List<Long> enterpriseIds) { List<OrganizationVO> vos = new ArrayList<>(); List<OrganizationEntity> organizationEntityList = organizationMapper.selectList( new QueryWrapper<OrganizationEntity>().lambda() .in(OrganizationEntity::getEnterpriseId, enterpriseIds) .eq(OrganizationEntity::getTypeCode, "project")); for (OrganizationEntity entity : organizationEntityList) { OrganizationVO vo = new OrganizationVO(); BeanUtils.copyProperties(entity, vo); vo.setEnterpriseId(entity.getEnterpriseId()); vos.add(vo); } return ApiResult.success(vos); }
时间: 2024-02-14 08:10:23 浏览: 36
好的,根据你提供的代码,我为你编写了一个单元测试的示例代码,如下所示:
```java
@RunWith(MockitoJUnitRunner.class)
public class OrganizationServiceTest {
@Mock
private OrganizationMapper organizationMapper;
@InjectMocks
private OrganizationServiceImpl organizationService;
@Test
public void testGetProByEnterpriseIds() {
// 构造测试数据
List<Long> enterpriseIds = Arrays.asList(1L, 2L, 3L);
List<OrganizationEntity> organizationEntityList = Arrays.asList(
new OrganizationEntity(1L, "project", "Project A", 1L),
new OrganizationEntity(2L, "project", "Project B", 2L),
new OrganizationEntity(3L, "project", "Project C", 3L)
);
when(organizationMapper.selectList(any())).thenReturn(organizationEntityList);
// 执行被测试的方法
ApiResult result = organizationService.getProByEnterpriseIds(enterpriseIds);
// 验证结果是否正确
verify(organizationMapper).selectList(any());
assertEquals(3, result.getData().size());
assertEquals("Project A", result.getData().get(0).getName());
assertEquals("Project B", result.getData().get(1).getName());
assertEquals("Project C", result.getData().get(2).getName());
}
}
```
这个示例代码使用了Mockito框架对`OrganizationMapper`进行了mock,并且注入到了`OrganizationServiceImpl`中。在测试方法中,我们构造了测试数据,并且使用`when`方法为`organizationMapper.selectList`方法设置了有效的返回值。然后,我们调用被测试的`organizationService.getProByEnterpriseIds`方法,并且验证方法是否正确执行。最后,我们使用`assertEquals`方法来验证返回的结果是否正确。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)