编写单元测试时的注意事项
根据软件开发过程中的TDD理论，在我们编写自己的代码时，要尽量使得该代码能够进行单元测试。为了能够使得代码可以进行单元测试，我们在给接口或方法传入参数时要尽量传入简单参数，避免传入HttpServletRequest , ServletContext等和web上下文相关的复杂对象。但仍有部分情况下基于代码简洁性和可维护性的考虑，我们需要传入HttpServletRequest对象，此时对此类方法进行JUnit单元测试时会较为困难，本文介绍一种在Spring中通过Mock来模拟HttpServletRequest对象进行JUnit单元测试的方法。
假设在HttpServletRequest中有一个userId字符串对象，我们想在queryUserById方法中调用该参数来获取用户信息，则正确的做法应如下所示:
1
2
3
4
5
6
7
|
String userId = request.getAttribute("userId").toString();//先获取userId对象
queryUserById(userId);//然后将获取的userId传入对应方法
public User queryUserById(String userId){//相关该方法
User userModel = userDao.findById(userId);
return userModel;
}
|
请尽量避免使用第二种方式
1
2
3
4
5
6
7
8
|
queryUserById(request);//直接传入request对象
public User queryUserById(HttpServletRequest request){//相关方法
String userId = request.getAttribute("userId").toString();//在该方法内部获取userId
User userModel = userDao.findById(userId);
return userModel;
}
|
若采用第一种方法，我们在进行单元测试时，可以很容易的自己制造一个String字符串来代表userId进行测试，但当采用第二种方法后，在进行单元测试时我们是比较难以模拟一个HttpServletRequest对象，从而影响我们的测试。
Spring和Mock在单元测试中的使用
在某些方法中，为了减少代码量和提高程序的可读性，我们有时候需要直接传入HttpServletRequest或ServletContext对象，如果我们想对这种方法进行测试，可以利用Mock来模拟相关的对象。
由于Spring自身已经整合了Mock相关的类，故在此处展示一个示例代码，以供参考:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
import java.io.File;
import org.junit.Test;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockServletContext;
public class SpringMockTest {
@Test
public void testHttpServletRequest(){
String realPath ="file:D:\\Java\\apache-tomcat-7.0.23\\webapps\\tmn";
//模拟ServletContext,同时初始化realPath，注意要有file:前缀否则会报错
MockServletContext context = new MockServletContext(realPath);
//获取realPath
System.out.println(context.getRealPath(File.separator));
//模拟HttpServletRequest
MockHttpServletRequest request = new MockHttpServletRequest(context);
//通过HttpServletRequest来获取realPath
System.out.println(request.getSession().getServletContext().getRealPath(File.separator));
}
}
|
注意:请在上下文路径的字符串前面加上file:前缀，否则程序会报错。如上面的程序，realPath的值应为 file:D:\Java\apache-tomcat-7.0.23\webapps\tmn ，若去掉file:前缀，改为 D:\Java\apache-tomcat-7.0.23\webapps\tmn ，则程序会报错。
