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());

}

}