MOCKITO 应用示例

package com.paic.wms.service.auditflow.impl;

 

import static org.junit.Assert.*;

 

import java.util.ArrayList;

import java.util.List;

 

import org.json.JSONArray;

import org.json.JSONObject;

import org.junit.After;

import org.junit.Before;

import org.junit.Test;

 

import com.paic.mms.dto.PsPaicEmpInfoDTO;

import com.paic.wp.dao.system.AuditFlowDAO;

import com.paic.wp.service.system.AuditFlowServiceImpl;

//Let's import Mockitostatically so that the code looks clearer

import static org.mockito.Mockito.*;

//import staticorg.powermock.api.mockito.PowerMockito.*; 

public class AuditFlowServiceImplTest {

    AuditFlowServiceImpl service;

    AuditFlowDAO daoMock;

    @Before

    public void setUp() throws Exception {

       service = new AuditFlowServiceImpl();

       daoMock = mock(AuditFlowDAO.class);

       service.setAuditFlowDAO(daoMock);

    }

    @Test

    public final void testSelectPsEmpInfoByUm() throws Exception {

       testUmIsNull();

       testUmIsNotNullAndPaPaicEmpInfoListIsNull();

       testUmIsNotNullAndPaPaicEmpInfoListIsNotNull();

    }

    @Test

    public final void testSelectPsEmpInfoByUmOrName() throws Exception {

       testUmIsNullForByUmOrName();

       testUmIsNotNullAndPaPaicEmpInfoListIsNullForByUmOrName();

       testUmIsNotNullAndPaPaicEmpInfoListIsNotNullForByUmOrName();

    }

    @After

    public void tearDown() throws Exception {

       service = null;

    }

   

    private void testUmIsNotNullAndPaPaicEmpInfoListIsNotNull() throws Exception{

       //given

       String um = "zhengxinliang001";

       List<PsPaicEmpInfoDTO> list = getPsPaicEmpInfoList();

       //when

       when(daoMock.selectPsEmpInfoByUm("zhengxinliang001")).thenReturn(list);

       PsPaicEmpInfoDTO dto = service.selectPsEmpInfoByUm(um);

       //then

       assertEquals("郑新良", dto.getLAST_NAME());

    }

 

    private List<PsPaicEmpInfoDTO> getPsPaicEmpInfoList() {

       PsPaicEmpInfoDTO dtoInServer = new PsPaicEmpInfoDTO();

      

       dtoInServer.setPAIC_UM_NUM("zhengxinliang001");

       dtoInServer.setLAST_NAME("郑新良");

       dtoInServer.setDEPTID("10000000");

       dtoInServer.setDEPT_NAME("平安机构");

      

       List<PsPaicEmpInfoDTO> list = newArrayList<PsPaicEmpInfoDTO>();

      

       list.add(dtoInServer);

       return list;

    }

 

    private void testUmIsNotNullAndPaPaicEmpInfoListIsNull() throws Exception {

       //given

       String um = "noBody001";

       //when

       when(daoMock.selectPsEmpInfoByUm("noBody001")).thenReturn(null);

       PsPaicEmpInfoDTO dto = service.selectPsEmpInfoByUm(um);

       //then

       assertEquals(dto, null);

    }

 

    private void testUmIsNull() throws Exception {

       //given

       String um = null;

       //when

       when(daoMock.selectPsEmpInfoByUm("")).thenReturn(null);

       PsPaicEmpInfoDTO dto = service.selectPsEmpInfoByUm(um);

       //then

       assertEquals(dto, null);

    }

   

    //当um账号存在的时候

    private void testUmIsNotNullAndPaPaicEmpInfoListIsNotNullForByUmOrName()throws Exception{

       //given

       String um = "zhengxinliang001";

       //when

       when(daoMock.selectPsEmpInfoByUmOrName(um)).thenReturn(getPsPaicEmpInfoList());

       String umInfoFromService = service.selectPsEmpInfoByUmOrName(um);

       //then

       JSONObject umInfo = getPsEmpInfoJSON();

       assertEquals(umInfo.toString(), umInfoFromService);

    }

 

    private JSONObject getPsEmpInfoJSON() {

       JSONObject umInfo = new JSONObject();

       umInfo.put("um", "zhengxinliang001");

       umInfo.put("name", "郑新良");

       umInfo.put("deptNo", "10000000");

       umInfo.put("deptName", "平安机构");

      

       JSONArray empArray = new JSONArray();

       empArray.put(umInfo);

      

       JSONObject returnJSON = new JSONObject();

       returnJSON.put("size", 1);

       returnJSON.put("zhengxinliang001", empArray);

       return returnJSON;

    }

   

    //um账号不存在的时候

    private void testUmIsNotNullAndPaPaicEmpInfoListIsNullForByUmOrName()throws Exception{

       //given

       String um = null;

       //when

       when(daoMock.selectPsEmpInfoByUm("noBody001")).thenReturn(null);

       String umInfoFromService = service.selectPsEmpInfoByUmOrName(um);

       //then

       JSONObject umInfo = new JSONObject();

       umInfo.put("size", "0");

      

       assertEquals(umInfoFromService, umInfo.toString());

    }

   

    //当um账号是空的时候

    private void testUmIsNullForByUmOrName() throws Exception {

       //given

       String um = null;

       //when

       when(daoMock.selectPsEmpInfoByUm("")).thenReturn(null);

       String umInfoFromService = service.selectPsEmpInfoByUmOrName(um);

       //then

       JSONObject umInfo = new JSONObject();

       umInfo.put("size", "0");

      

       assertEquals(umInfoFromService, umInfo.toString());

    }

 

}