<pre class="java" name="code">/**
* JUnit單元測試父類，配置了Spring的基礎環境。 <br/>
* 可以作為Controller、Service、Dao單元測試的父類。
*
* @author leiwen@fansunion.cn
*/
public class JUnitTestBase {
public static XmlWebApplicationContext context = null;
private static HandlerMapping handlerMapping;
private static HandlerAdapter handlerAdapter;
// /public static String[] CONFIG_FILES = { "classpath:spring-*.xml" };
// public static String[] configs = { "file:src/main/resources/spring-*.xml"
// };
public static String[] CONFIG_FILES = { "file:src/main/resources/spring-*.xml" };
/**
* 讀取spring配置文件，初始化上下文。
*/
@BeforeClass
public static void setUp() {
System.out.println("Test start...");
context = new XmlWebApplicationContext();
context.setConfigLocations(CONFIG_FILES);
MockServletContext msc = new MockServletContext();
context.setServletContext(msc);
context.refresh();
msc.setAttribute(
WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE，
context);
handlerMapping = (HandlerMapping) context
.getBean(DefaultAnnotationHandlerMapping.class);
handlerAdapter = (HandlerAdapter) context.getBean(context
.getBeanNamesForType(AnnotationMethodHandlerAdapter.class)[0]);
}
// 執行request對象請求的action
public ModelAndView excuteAction(HttpServletRequest request，
HttpServletResponse response) throws Exception {
// 這里需要聲明request的實際類型，否則會報錯
request.setAttribute(HandlerMapping.INTROSPECT_TYPE_LEVEL_MAPPING， true);
HandlerExecutionChain chain = handlerMapping.getHandler(request);
Object handler = chain.getHandler();
final ModelAndView model = handlerAdapter.handle(request， response，
handler);
return model;
}
@AfterClass
public static void tearUp() {
System.out.println("Test end!");
}
}
//測試Controller的2種方法
/**
* CampaignGroupController單元測試。
*
* @author leiwen@fansunion.cn
*/
public class FansUnionControllerTestextends JUnitTestBase {
@Test
public void prevAddCampaignGroup() throws Exception {
MockHttpServletRequest request = new MockHttpServletRequest();
MockHttpServletResponse response = new MockHttpServletResponse();
String requestURI = "/campaignGroup/prevAdd";
request.setRequestURI(requestURI);
request.setMethod("POST");
final ModelAndView mav = this.excuteAction(request， response);
Assert.assertEquals("createCampaignGroup"， mav.getViewName());
}
@Test
public void prevAddCampaignGroup2() throws Exception {
CampaignGroupController userController = context.getBean(CampaignGroupController.class);
MockMvc mockMvc = MockMvcBuilders.standaloneSetup(userController)
.build();
ResultActions perform = mockMvc.perform(MockMvcRequestBuilders.post("/campaignGroup/prevAdd"));
ResultActions andExpect = perform
.andExpect(MockMvcResultMatchers.status().is(200));
andExpect
.andExpect(
MockMvcResultMatchers.view()
.name("createCampaignGroup"));
}
}
----需要配置2個bean，測試Controller需要用到
<bean
class="org.springframework.web.servlet.mvc.annotation.DefaultAnnotationHandlerMapping">
</bean>
<bean
class="org.springframework.web.servlet.mvc.annotation.AnnotationMethodHandlerAdapter">
</bean>
測試Service(Dao類似)
/**
* 這種方式 的單元測試，不需要重復指定 資源文件，減少重復。 麻煩的地方在于，需要手動獲取bean。
*
* @author leiwen@fansunion.cn
*/
public class FansUnionServiceTest extends JUnitTestBase {
private static CampaignGroupService campaignGroupService = context
.getBean(CampaignGroupService.class);
// 測試add，remove，update方法的時候，已經測試了find？
@Test
public void addCampaignGroup() {
CampaignGroup cg = buildOneGroup();
campaignGroupService.add(cg);
CampaignGroup dbCampaignGroup = campaignGroupService.find(cg.getId());
Assert.assertEquals(cg， dbCampaignGroup);
campaignGroupService.delete(cg.getId());
}
基于注解的測試
/**
*
* 基于注解的單元測試。
*
* @author leiwen@fansunion.cn
*/
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = { "classpath:spring-mvc-context.xml"，
"classpath:spring-common-context.xml" })
public class FansUnionServiceTestWithAnnotation {
@Autowired
private CampaignGroupService campaignGroupService;
/**
* 根據campaignId獲取campaignGroup的name集合<br>
* 測試場景：<br>
*
*/
@Test
public void testListCampaignGroupNameByCampaignId(){
System.out.println("**********");
System.out.println(campaignGroupService.listCampaignGroupNameByCampaignId(1005899));
}
}
