UT单元测试的时候,遇到一种情况就是service/dao为空,不能调用方法,一般情况下使用@Autowired,在项目启动后使用时就可以自动注入service/dao,但是使用junit单元测试时,不能成功进行注入。
测试类:
此时可以通过java反射的方式,以下是要测试的类,此时直接通过bookSoldOutService.queryOffShelfTaskList();调用方法是跑不通的,bookSoldOutService未进行注入。
public class OffTheShelfTask {
@Autowired
private BookSoldOutService bookSoldOutService;
public void task() {
logger.info("---begin OffTheShelfTask task---");
// 查询任务列表
List<OfflineShelfTaskPojo> offlineList = bookSoldOutService.queryOffShelfTaskList();
// 任务分周期list
List<OfflineShelfTaskPojo> periodList = new ArrayList<OfflineShelfTaskPojo>();
int count = 0;
if (offlineList != null && !offlineList.isEmpty()) {
for (OfflineShelfTaskPojo offlinePojo : offlineList) {
periodList.add(offlinePojo);
count++;
if (count % 50 == 0) {
// 每50调记录发送一次
periodSyncOffShelf(periodList);
periodList = new ArrayList<OfflineShelfTaskPojo>();
}
}
// 发送%50剩余的记录
periodSyncOffShelf(periodList);
}
logger.info("---end OffTheShelfTask task ---");
}
}
测试类:
public class OffTheShelfTaskTest extends TransactionRollbackTestCase{
private BookSoldOutService bookSoldOutService;
private OffTheShelfTask offTheShelfTask = new OffTheShelfTask();
@Test
public void testTask() throws NoSuchFieldException, SecurityException, IllegalArgumentException, IllegalAccessException {
//1. 通过反射找到该类里的某个字段
Field f = OffTheShelfTask.class.getDeclaredField("bookSoldOutService");
//2. 设置是否启用安全访问检查
f.setAccessible(true);
//3.将service/dao实例化
bookSoldOutService = new BookSoldOutService();
//3.为该对象设置service/dao
f.set(offTheShelfTask, bookSoldOutService);
offTheShelfTask.task();//最后调用方法的时候,service就注入了
}
}
最后通过调用对象.方法,就可以进行service/dao注入了。