業務經常涉及到Excel檔案的匯出，後臺給前端提供匯出服務介面，不可能自己都沒測試過，就提供給前端呼叫。最好的方案就是程式設計師編寫單元測試，可以說，程式設計師有責任編寫功能程式碼，同時也就有責任為自己的程式碼編寫單元測試。

單元測試

java 常用的單元測試框架是Junit，利用測試框架可以使我們在編寫測試用例的過程更加便捷優雅。通過測試用例，我們只需要關注：對於特定的輸入，被測物件的返回是否征程，下面是涉及到的相關類。

controller類提供匯出介面

    @GET
        @Path("/{version}/exportRecordDetail")
        @ApiOperation(value = "匯出電子支付結算明細"
 
)
        public String exportRecordDetail (@Context HttpServletResponse response, @BeanParam SingleWithdrawReq request){        
            //...
            response.setContentType("application/x-download");
            // 獲取模板檔案
            String templateFile = "/export/AccountRecordDetail.xlsx";       
            try
 
 (InputStream is = this.getClass().getResourceAsStream(templateFile);
                 OPCPackage pkg = OPCPackage.open(is);
                 XSSFWorkbook workbook = new XSSFWorkbook(pkg)) {
                outputFile = URLEncoder.encode(outputFile, "UTF-8");        
                //...
                workbook.write(response.getOutputStream());
            } catch (IOException | InvalidFormatException e) {
                //...
 
 
            }
            return null;
        }

核心程式碼： workbook.write(response.getOutputStream()); 這一句是指將檔案寫到輸出流裡，用到了HttpservletResponse 介面，因此在測試用例裡，我們並不能直接例項化HttpservletResponse，因而也就不能作為引數傳遞到controller層。

那怎麼辦呢？首先想到了引入mock方法。Mock的引入，可以幫助我們構建比較難構造的Object，這些Object 通常有很多以來，在單元測試中構造出這些物件通常花費比較大的成本。

Mock 單元測試

@Test 
public void exportService() {
    SingleWithdrawReq request = new SingleWithdrawReq();
    request.setParkId(12345);
    ParkingBalanceAccountResources resource = new ParkingBalanceAccountResources();
    // Mock HttpServletResponse 例項
    HttpServletResponse response = Mockito.mock(HttpServletResponse.class);
    // 定義一個輸出的本地檔案
    File file = new File("output.xls");
    try (ServletOutputStream sos = new MockServletOutputStreamUtil (file)) {
        //用MockServletOutputStreamUtil 代替ServletOutputStream
        Mockito.when(response.getOutputStream()).thenReturn(sos);
        resource.exportRecordDetail(response, request);
    }
 }

單元測試思路： 將輸出流寫到一個臨時檔案裡，但是在case裡我們如何傳入一個response並且得到輸出流呢？

工具類：構造ServletOutputStream

public class MockServletOutputStreamUtil extends ServletOutputStream {

    private FileOutputStream fos;    //定義檔案輸出流
    public MockServletOutputStreamUtil(File file) {
        try {
            this.fos = new FileOutputStream(file);    // 將輸出流指向檔案
        } catch (FileNotFoundException e) {
            throw new RuntimeException(e);
        }
    }
    @Override
    public boolean isReady() {
        return true;
    }
    @Override
    public void setWriteListener(WriteListener writeListener) {
        throw new RuntimeException("do not support this method");
    }
    @Override
    public void write(int b) throws IOException {
        fos.write(b);
    }
    @Override
    public void close() {
        try {
            fos.close();
        } catch (IOException e) {
            throw new RuntimeE
xception(e);
        }
    }