java反射 -- 通过反射注入service/dao,进行UT测试

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 offlineList = bookSoldOutService.queryOffShelfTaskList();
		// 任务分周期list
		List periodList = new ArrayList();
		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();
				}
			}
			// 发送%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注入了。


你可能感兴趣的:(java反射)