示例代码:
public String test(){
....
....
String departmentManager;
Department dept = departMentService.queryById("1001");
while(true) {
if("C".equals(dept.getDeptType)){
departmentManager = userService.queryManagerByDeptId(dept.getDeptId());
break;
}
else {
dept = departMentService.queryById(dept.getParentId());
}
}
....
....
}
测试用例:
@Test
public void testDemo() {
List<Department> deptList = new ArrayList<>();
Department dept = new Department();
dept.setId(1001);
dept.setDeptType("A");
deptList.add(dept);
dept = new Department();
dept.setId(1002);
dept.setDeptType("B");
deptList.add(dept);
dept = new Department();
dept.setId(1003);
dept.setDeptType("C");
deptList.add(dept);
Mockito.when(userService.queryById(Mockito.anyString()))
.thenReturn(deptList.get(0))
.thenReturn(deptList.get(1))
.thenReturn(deptList.get(2));
String deptManager = "负责人1";
Mockito.when(userService.queryManagerById(Mockito.anyString()))
.thenReturn(deptManager);
}