autoMock工具开发

开发背景

对于复杂的AO层或者是试用这边的FO层，如果使用easyMock来对其进行单元测试将会给测试人员带来无限的烦恼，因为该层的代码复合了很多的MO或者是外部的一些相关类，这使得easyMock的代码写起来会很麻烦，因为不仅需要mock要测试方法的参数，还要mock内部的一个自动注入属性所调用的方法，这会是一个令测试人员头疼的问题，automock工具就是为了解析这个问题而诞生的。

 

工具运行机制

         测试人员将要测试的类的路径（文件系统的绝对路径）输入并调用工具的生成easymock代法的方法，便能将要测试的类的基础代码全部生成好，并将代码输入至控制台及复制到粘贴板，测试人员直接粘贴代码到测试文件并引入必要的包，设置必要的参数就可以了。

 

工具实现原理

读入java代码，通过ASTParser对java代码进行解析，从中解析出自动注入的属性以及其被调用的方法，使用模板生成工具类进行代码生成

工具实现类图

 
<!--[endif]-->

工具时序图

 
<!--[endif]-->

 

工具运行示例

需要测试的示例代码

public class StudentAO { private StudentManager studentManager; private InfomationManager infomationManager; public void getSameStudent(StudentDO studentDO,Boolean a){ int age=studentDO.getAge(); String name=studentDO.getName(); if(!isYoungStudent(studentDO,3)){ return; } studentManager.changeStudentAge(10); if(infomationManager.isYoungStudent(10)){ return; } studentManager.isYoungStudent(10); if(this.isSameStudent()==false){ return; } infomationManager.changeStudentAge(20); studentDO=studentManager.getStudentDO("student"); } public boolean sameStudent(StudentDO studentDO,Boolean a){ int age=studentDO.getAge(); String name=studentDO.getName(); studentManager.isYoungStudent(10); if(this.isSameStudent()==false){ return true; } infomationManager.changeStudentAge(20); studentDO=studentManager.getStudentDO("student"); return false; } private boolean isYoungStudent(StudentDO studentDO,int a){ if(studentDO.getAge()>0){ return true; } return false; } private boolean isSameStudent(){ if(studentManager.isSameStudent(null)){ return true; } return isYoungStudent(null,5); } public void setStudentManager(StudentManager studentManager) { this.studentManager = studentManager; } public void setInfomationManager(InfomationManager infomationManager) { this.infomationManager = infomationManager; } }

 

调用工具生成代码

public static void main(String args[]){ MockUtil mockUtil=new MockUtil(); String codeUrl="D:\\study\\automock\\src\\test\\java\\com\\taobao\\matrix\\automock\\util\\StudentAO.java"; try { String code=mockUtil.mockCode(codeUrl); } catch (Exception e) { e.printStackTrace(); } } 生成的代码如下 @Test public void getSameStudent_test(){ //实例化需要测试的对象 StudentAO studentAO=new StudentAO(); //参数及属性相关设置start StudentDO studentDO=EasyMock.createMock(StudentDO.class); //参数设置start //参数设置end EasyMock.replay(studentDO); //参数及属性相关设置end //参数及属性相关设置start StudentManager studentManager=EasyMock.createMock(StudentManager.class); studentManager.changeStudentAge(0); EasyMock.expectLastCall(); EasyMock.expect(studentManager.isYoungStudent(0)).andReturn(false).anyTimes(); EasyMock.expect(studentManager.getStudentDO("mocktest")).andReturn(null).anyTimes(); EasyMock.expect(studentManager.isSameStudent((com.taobao.matrix.automock.util.StudentDO)EasyMock.anyObject())).andReturn(false).anyTimes(); EasyMock.replay(studentManager); //参数及属性相关设置end //参数及属性相关设置start InfomationManager infomationManager=EasyMock.createMock(InfomationManager.class); EasyMock.expect(infomationManager.isYoungStudent(0)).andReturn(false).anyTimes(); infomationManager.changeStudentAge(0); EasyMock.expectLastCall(); EasyMock.replay(infomationManager); //参数及属性相关设置end //将属性注入start studentAO.setStudentManager(studentManager); studentAO.setInfomationManager(infomationManager); //将属性注入end //调用测试方法 studentAO.getSameStudent(studentDO,false); //assert验证区块start //assert验证区块end } @Test public void sameStudent_test(){ //实例化需要测试的对象 StudentAO studentAO=new StudentAO(); //参数及属性相关设置start StudentDO studentDO=EasyMock.createMock(StudentDO.class); //参数设置start //参数设置end EasyMock.replay(studentDO); //参数及属性相关设置end //参数及属性相关设置start StudentManager studentManager=EasyMock.createMock(StudentManager.class); EasyMock.expect(studentManager.isYoungStudent(0)).andReturn(false).anyTimes(); EasyMock.expect(studentManager.getStudentDO("mocktest")).andReturn(null).anyTimes(); EasyMock.expect(studentManager.isSameStudent((com.taobao.matrix.automock.util.StudentDO)EasyMock.anyObject())).andReturn(false).anyTimes(); EasyMock.replay(studentManager); //参数及属性相关设置end //参数及属性相关设置start InfomationManager infomationManager=EasyMock.createMock(InfomationManager.class); infomationManager.changeStudentAge(0); EasyMock.expectLastCall(); EasyMock.replay(infomationManager); //参数及属性相关设置end //将属性注入start studentAO.setStudentManager(studentManager); studentAO.setInfomationManager(infomationManager); //将属性注入end //调用测试方法 boolean booleanResult=studentAO.sameStudent(studentDO,false); //assert验证区块start //assert验证区块end }