package com.xjrsoft.xjrsoftboot;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.support.ExcelTypeEnum;
import com.xjrsoft.XjrSoftApplication;
import com.xjrsoft.module.ledger.dto.WfSubscriptionPageDto;
import com.xjrsoft.module.oa.service.IWfSubscriptionService;
import com.xjrsoft.module.ledger.vo.WfSubscriptionExcelVo;
import org.junit.jupiter.api.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.List;

/**
 * @author dzx
 * @date 2024/2/20
 */
@RunWith(SpringRunner.class)
@SpringBootTest(classes = XjrSoftApplication.class)
public class ExcelDownloadTest {
    @Autowired
    private IWfSubscriptionService wfSubscriptionService;
    @Test
    void downloadWfSubscription() throws IOException {
        WfSubscriptionPageDto dto = new WfSubscriptionPageDto();
        List<WfSubscriptionExcelVo> customerList = wfSubscriptionService.getList(dto);
        ByteArrayOutputStream bot = new ByteArrayOutputStream();
        EasyExcel.write(bot, WfSubscriptionExcelVo.class).automaticMergeHead(false).excelType(ExcelTypeEnum.XLSX).sheet().doWrite(customerList);
        byte[] byteArray = bot.toByteArray();

        File file = new File("D:\\workspace\\其他\\测试\\file.xlsx");
        FileOutputStream fos = new FileOutputStream(file);
        fos.write(byteArray);
        fos.close();
    }
}