/**
 * JUnit單元測試父類，配置了Spring的基礎環境。 <br/>
 * 可以作為Controller、Service、Dao單元測試的父類。
 * 
 * @author [email protected]
 */
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 
 
[email protected]
 */
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 
 
[email protected]
 */
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 
 
[email protected]
 */
@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));
 }

}