java - 测试方法中的返回语句

Question

我应该如何在测试方法中正确返回元素。我的测试有问题。

when(teamService.createTeam(teamDto)).thenReturn();

在这一行中，我不知道如何正确编写 return 语句。thenReturn 之后的括号中应该包含什么。我的方法创建团队如下所示:

@Transactional
    public Team createTeam(TeamDto teamDto) {
        Assert.notNull(teamDto, "Object can't be null!");
        try {
            Assert.notNull(teamDto.getName());
            return teamRepository.save(modelMapper.map(teamDto, Team.class));
        } catch (Exception e) {
            throw new CreateEntityException(e);
        }
    }

在这个方法中，我返回 Team 对象，但是当我添加 Team 时，我有预期的表达式。

Answer 1

根据您想要做什么，您可以执行以下操作:

when(teamService.createTeam(teamDto)).thenReturn(new Team());

或

Team team = mock(Team.class);
when(teamService.createTeam(teamDto)).thenReturn(team);

PS:我假设您正在测试的类不是包含您发布的方法的类。我假设您正在 mock 该服务，因此在之前的某个地方，您应该编写如下内容:

TeamService teamService = mock(TeamService.class);