Mockito+PowerMock单元测试

POM文件配置

<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-all</artifactId>
    <version>2.0.2-beta</version>
    <scope>test</scope>
</dependency>

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito</artifactId>
    <version>1.7.4</version>
    <scope>test</scope>
</dependency>

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>1.7.4</version>
    <scope>test</scope>
</dependency>

 

普通对象方法测试用例

@RunWith(SpringJUnit4ClassRunner.class)
public abstract class MockitoBasedTest {
    @Before
    public void setUp() throws Exception {
        // 初始化测试用例类中由Mockito的注解标注的所有模拟对象
        MockitoAnnotations.initMocks(this);
    }
}

@RequestMapping("reportSetting")
public class ReportSettingController {
    @Resource
    private ReportSettingService reportSettingService;

    @PostMapping("addReportSetting")
    public BaseResponse addReportSetting(@RequestBody @Valid AddReportSettingRequest request) {
        BaseResponse response = new BaseResponse();
        try {
            String result = reportSettingService.addAllReportSetting(request);
            if (StringUtils.isBlank(result)) {
                response.setCode(ConfigConsts.STATUS_CODE_SUCCESS);
                response.setMessage("设置成功");
            } else {
                response.setCode(ConfigConsts.STATUS_CODE_PARAM_ERROR);
                response.setMessage("设置失败，" + result);
            }

            return response;
        } catch (Exception e) {
            response.setCode(ConfigConsts.STATUS_CODE_INNER_ERROR);
            response.setMessage("设置异常");
            return response;
        }
    }
}

import net.sf.jsqlparser.expression.operators.relational.EqualsTo;
import org.junit.Assert;
import org.junit.Test;
import org.mockito.InjectMocks;
import org.mockito.Mock;

import static org.hamcrest.CoreMatchers.any;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.junit.Assert.*;
import static org.mockito.Matchers.anyObject;
import static org.mockito.Mockito.when;

public class ReportSettingControllerTest extends MockitoBasedTest {

    @InjectMocks
    private ReportSettingController reportSettingController;
    @Mock
    private ReportSettingService reportSettingService;
    @Test
    public void addReportSetting() {
        when(reportSettingService.addAllReportSetting(anyObject()))
               .thenReturn(null)
                .thenReturn(" 验证失败")
                .thenThrow(new RuntimeException("验证异常"));
        BaseResponse responseSuccess = reportSettingController.addReportSetting(anyObject());
        Assert.assertThat(responseSuccess.getCode(), equalTo("200"));
        BaseResponse responseFailed = reportSettingController.addReportSetting(anyObject());
        Assert.assertThat(responseFailed.getCode(), equalTo("400"));
        BaseResponse responseException = reportSettingController.addReportSetting(anyObject());
        Assert.assertThat(responseException.getCode(), equalTo("500"));
    }
}

 

包含静态方法的测试用例

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.util.LinkedHashMap;
import java.util.List;

@RequestMapping("sscBenf")
public class HrPsLfSscBenfMthController {
    @Resource
    private HrPsLfSscBenfMthService hrPsLfSscBenfMthService;

    @Resource
    private CommonService commonService;

    @Resource
    private ReportSettingService reportSettingService;

    private static final LinkedHashMap<String, String> MAP_KEY_VALUE = new LinkedHashMap<>();
    private static final String CLASS_PATH_Benf = "";
    private static final String TITLE_Benf = "";

    @GetMapping("/exportData")
    public BaseResponse exportData(HttpServletRequest req, HttpServletResponse resp) {
        BaseResponse response = new BaseResponse();
        String strTitle = "";
        try {
            String authCode = req.getParameter("authCode");
            String cLfItemType = req.getParameter("cLfItemType");
            String taskBillNumber = req.getParameter("taskbillNumber");
            ReportSetting reportSet = reportSettingService.getByTaskbillNumber(taskBillNumber);
            String yearMonth = reportSet.getReportMonth();
            HrPsLfSscBenfMth hrPsLfSscBenfMth = new HrPsLfSscBenfMth();
            hrPsLfSscBenfMth.setCLfItemType(cLfItemType);
            hrPsLfSscBenfMth.setCLfPrdId(yearMonth);
            strTitle = yearMonth.replaceAll("/", "-") + TITLE_Benf + cLfItemType + "数据";
            String fileName = strTitle + ".xls";
            resp.setContentType("APPLICATION/OCTET-STREAM");
            resp.setHeader("Content-Disposition", "attachment; filename=" + java.net.URLEncoder.encode(fileName, "UTF-8"));
            OutputStream out = resp.getOutputStream();
            List<HrPsLfSscBenfMth> listOfBenfMthData = hrPsLfSscBenfMthService.selectListOfSscBenfMth(hrPsLfSscBenfMth);
            ExcelUtil.sayHello()
            return response;
        } catch (Exception e) {
            response.setCode(ConfigConsts.STATUS_CODE_INNER_ERROR);
            response.setMessage(strTitle + "导出异常");
            log.error(strTitle + "数据导出异常：Exception=", e);
            return response;
        }
    }

}

import net.sf.jsqlparser.expression.operators.relational.EqualsTo;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Matchers;
import org.mockito.Mock;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;

import java.util.ArrayList;

import static org.hamcrest.CoreMatchers.equalTo;
import static org.mockito.Matchers.anyObject;
import static org.mockito.Matchers.anyString;
import static org.powermock.api.mockito.PowerMockito.when;

@RunWith(PowerMockRunner.class)
@PrepareForTest({ExcelUtil.class})
public class HrPsLfSscBenfMthControllerTest {
    @InjectMocks
    private HrPsLfSscBenfMthController hrPsLfSscBenfMthController;
    @Mock
    private ReportSettingService reportSettingService;
    @Mock
    private CommonService commonService;
    @Mock
    private HrPsLfSscBenfMthService hrPsLfSscBenfMthService;

    @Before
    public void init() {
        PowerMockito.mockStatic(ExcelUtil.class);
    }

    @Test
    public void exportData() throws Exception {
        when(reportSettingService,"getByTaskbillNumber",anyString()).thenReturn(this.getReportSetting());
        when(commonService,"authCodeCheck",anyString()).thenReturn(true);
        when(hrPsLfSscBenfMthService,"selectListOfSscBenfMth", anyObject()).thenReturn(new ArrayList<>());

 　　　　　　　　PowerMockito.when(ExcelUtil.sayHello()).thenReturn("my hello");

        BaseResponse response_success = hrPsLfSscBenfMthController.exportData(new MockHttpServletRequest(), new MockHttpServletResponse());
        Assert.assertThat(response_success.getCode(), equalTo("200"));

        when(commonService,"authCodeCheck",anyString()).thenReturn(false);
        BaseResponse response_failed = hrPsLfSscBenfMthController.exportData(new MockHttpServletRequest(), new MockHttpServletResponse());
        Assert.assertThat(response_failed.getCode(), equalTo("400"));

        when(commonService,"authCodeCheck",anyString()).thenThrow(new RuntimeException("验证异常"));
        BaseResponse response_excepion = hrPsLfSscBenfMthController.exportData(new MockHttpServletRequest(), new MockHttpServletResponse());
        Assert.assertThat(response_excepion.getCode(), equalTo("500"));

    }

    private ReportSetting getReportSetting(){
        ReportSetting reportSetting = new ReportSetting();
        reportSetting.setReportMonth("2019/10");
        return reportSetting;
    }
}