您好,我有一个将人员添加到团队的方法。我想为此方法编写一个测试,但我是junit/mockito测试的新手,所以我有很多问题:这是我的添加方法:
@Transactional
public void addPersonsToTeams(Long teamId, Long personId) {
Assert.notNull(personId, "Object can't be null!");
Assert.notNull(teamId, "Object can't be null!");
try {
Person person = personRepository.getOne(personId);
Team team = teamRepository.getOne(teamId);
person.getTeams().add(team);
personRepository.save(person);
} catch (Exception e) {
throw new CreateEntityException();
}
}
这两个实体(人员/团队)之间存在关系,这是我的测试代码,但它不起作用:
@Test
public void shouldAddPersonToTeam(){
Team team = new Team(1l, "TestCase1", "Description1", "Krakow", 12);
Person person = new Person(1L, "jan", "mucha", "krakow", "email1@onet.com", "Programing", "Developer");
teamService.createTeam(mapper.map(team, TeamDto.class));
personService.addPerson(mapper.map(person, PersonDto.class));
teamService.addPersonsToTeams(team.getId(), person.getId());
verify(teamRepository, times(1)).save(team);
verify(personRepository, times(1)).save(person);
}
模拟组合:
public class TeamServiceTest {
private TeamService teamService;
private ModelMapper mapper;
private PersonService personService;
@Mock
private TeamRepository teamRepository;
private PersonRepository personRepository; //-this is never assigned :/
@Before
public void setUp() {
MockitoAnnotations.initMocks(this);
this.mapper = new ModelMapper();
teamService = new TeamService(teamRepository, this.mapper);
personService = new PersonService(personRepository, this.mapper);
}
你的问题是台词
Person person = personRepository.getOne(personId);
Team team = teamRepository.getOne(teamId);
在您正在测试的方法中。在addPersonsToTeams的单元测试中,您不想测试两个getOne方法的行为。这就是使用Mockito的全部意义所在——您可以为单个方法编写单元测试,而不需要其他方法的行为影响测试。
这意味着您需要指定这两个调用将返回什么。这是“存根”调用,只有在调用方法的对象是spies或mock时,才能执行此操作。所以在Mockito中,你可以写
doReturn(myPerson).when(mockPersonRepository).getOne(personId);
这意味着每当调用mockPERReposity.getOne(man Id)
时,什么都不会发生。该方法本身不会运行,Mockito只是立即返回myPerson
。
这就是你想要的技巧。因此,当您在测试中添加存根时,可能会出现如下情况。
@Mock private TeamRepository mockTeamRepository;
@Mock private PersonRepository mockPersonRepository;
@Before
public void setUp() {
MockitoAnnotations.initMocks(this);
mapper = new ModelMapper();
teamService = new TeamService(mockTeamRepository, mapper);
personService = new PersonService(mockPersonRepository, mapper);
}
@Test
public void shouldAddPersonToTeam(){
Team team = new Team(1L, "TestCase1", "Description1", "Krakow", 12);
Person person = new Person(1L, "jan", "mucha", "krakow", "email1@onet.com", "Programing", "Developer");
doReturn(team).when(mockTeamRepository).getOne(1L);
doReturn(person).when(mockPersonRepository).getOne(1L);
teamService.createTeam(mapper.map(team, TeamDto.class));
personService.addPerson(mapper.map(person, PersonDto.class));
teamService.addPersonsToTeams(team.getId(), person.getId());
verify(mockTeamRepository).save(team);
verify(mockPersonRepository).save(person);
}
只需多加几分。
验证调用中写入次(1)
——默认验证模式为次(1)
,因此,如果不使用它,代码就不会太混乱