package cn.iocoder.yudao.module.mes.service.energy;

import cn.iocoder.yudao.module.mes.controller.admin.energy.vo.*;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.*;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.mock.web.MockHttpServletResponse;
import java.io.ByteArrayInputStream;
import java.time.LocalDate;
import java.util.List;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;

@ExtendWith(MockitoExtension.class)
class EnergyExportServiceTest {
    @Mock EnergyRecordService recordService;
    @InjectMocks EnergyExportService exportService;
    @Test void analysisProducesReadableWorkbookWithScopeAndSeparateUnits() throws Exception {
        when(recordService.analysis(any())).thenReturn(EnergyRecordService.aggregate(List.of()));
        var req = new EnergyAnalysisReqVO(); req.setStartDate(LocalDate.of(2026,10,1)); req.setEndDate(LocalDate.of(2026,10,12));
        var response = new MockHttpServletResponse(); exportService.analysis(req, response);
        try (var workbook = WorkbookFactory.create(new ByteArrayInputStream(response.getContentAsByteArray()))) {
            assertEquals(4, workbook.getNumberOfSheets());
            assertEquals("统计口径", workbook.getSheetAt(0).getSheetName());
            assertEquals("2026-10-01", workbook.getSheetAt(0).getRow(1).getCell(1).getStringCellValue());
            assertEquals("m³", workbook.getSheetAt(1).getRow(1).getCell(2).getStringCellValue());
            assertEquals("kWh", workbook.getSheetAt(1).getRow(2).getCell(2).getStringCellValue());
            assertEquals("t", workbook.getSheetAt(1).getRow(3).getCell(2).getStringCellValue());
            assertEquals(0, workbook.getSheetAt(1).getRow(1).getCell(4).getNumericCellValue());
        }
        assertTrue(response.getContentType().contains("spreadsheetml"));
    }
    @Test void oversizedLedgerExportFailsBeforeWritingResponse() {
        when(recordService.page(any())).thenReturn(new PageResult<>(List.of(), 50_001L));
        var response = new MockHttpServletResponse();
        assertThrows(RuntimeException.class, () -> exportService.records(new EnergyRecordPageReqVO(), response));
        assertEquals(0, response.getContentAsByteArray().length);
    }
}