完善Excel模板文本格式与资产状态下拉

ccdi-info-collection/src/test/java/com/ruoyi/info/collection/utils/EasyExcelUtilTemplateTest.java

@@ -0,0 +1,98 @@
+package com.ruoyi.info.collection.utils;
+
+import com.ruoyi.common.core.domain.entity.SysDictData;
+import com.ruoyi.common.utils.DictUtils;
+import com.ruoyi.info.collection.domain.excel.CcdiAssetInfoExcel;
+import com.ruoyi.info.collection.domain.excel.CcdiStaffFmyRelationExcel;
+import org.apache.poi.ss.usermodel.CellStyle;
+import org.apache.poi.ss.usermodel.DataValidation;
+import org.apache.poi.ss.usermodel.Sheet;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.ss.util.CellRangeAddress;
+import org.apache.poi.ss.usermodel.WorkbookFactory;
+import org.junit.jupiter.api.Test;
+import org.mockito.MockedStatic;
+import org.springframework.mock.web.MockHttpServletResponse;
+
+import java.io.ByteArrayInputStream;
+import java.util.List;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.mockito.Mockito.mockStatic;
+
+class EasyExcelUtilTemplateTest {
+
+    @Test
+    void importTemplateWithDictDropdown_shouldAddAssetStatusDropdownToAssetTemplate() throws Exception {
+        MockHttpServletResponse response = new MockHttpServletResponse();
+
+        try (MockedStatic<DictUtils> mocked = mockStatic(DictUtils.class)) {
+            mocked.when(() -> DictUtils.getDictCache("ccdi_asset_status"))
+                    .thenReturn(List.of(
+                            buildDictData("正常"),
+                            buildDictData("冻结"),
+                            buildDictData("处置中"),
+                            buildDictData("报废")
+                    ));
+
+            EasyExcelUtil.importTemplateWithDictDropdown(response, CcdiAssetInfoExcel.class, "亲属资产信息");
+        }
+
+        try (Workbook workbook = WorkbookFactory.create(new ByteArrayInputStream(response.getContentAsByteArray()))) {
+            Sheet sheet = workbook.getSheetAt(0);
+            assertTrue(hasValidationOnColumn(sheet, 9), "资产状态列应包含下拉校验");
+        }
+    }
+
+    @Test
+    void importTemplateWithDictDropdown_shouldFormatCertificateColumnsAsText() throws Exception {
+        MockHttpServletResponse assetResponse = new MockHttpServletResponse();
+        MockHttpServletResponse familyResponse = new MockHttpServletResponse();
+
+        try (MockedStatic<DictUtils> mocked = mockStatic(DictUtils.class)) {
+            mocked.when(() -> DictUtils.getDictCache("ccdi_asset_status"))
+                    .thenReturn(List.of(buildDictData("正常")));
+            mocked.when(() -> DictUtils.getDictCache("ccdi_relation_type"))
+                    .thenReturn(List.of(buildDictData("配偶")));
+            mocked.when(() -> DictUtils.getDictCache("ccdi_indiv_gender"))
+                    .thenReturn(List.of(buildDictData("男")));
+            mocked.when(() -> DictUtils.getDictCache("ccdi_certificate_type"))
+                    .thenReturn(List.of(buildDictData("居民身份证")));
+
+            EasyExcelUtil.importTemplateWithDictDropdown(assetResponse, CcdiAssetInfoExcel.class, "亲属资产信息");
+            EasyExcelUtil.importTemplateWithDictDropdown(familyResponse, CcdiStaffFmyRelationExcel.class, "员工亲属关系信息");
+        }
+
+        try (Workbook assetWorkbook = WorkbookFactory.create(new ByteArrayInputStream(assetResponse.getContentAsByteArray()));
+             Workbook familyWorkbook = WorkbookFactory.create(new ByteArrayInputStream(familyResponse.getContentAsByteArray()))) {
+            assertTextColumn(assetWorkbook.getSheetAt(0), 0);
+            assertTextColumn(familyWorkbook.getSheetAt(0), 6);
+        }
+    }
+
+    private void assertTextColumn(Sheet sheet, int columnIndex) {
+        CellStyle style = sheet.getColumnStyle(columnIndex);
+        assertNotNull(style, "文本列应设置默认样式");
+        assertEquals("@", style.getDataFormatString(), "证件号列应使用文本格式");
+    }
+
+    private boolean hasValidationOnColumn(Sheet sheet, int columnIndex) {
+        for (DataValidation validation : sheet.getDataValidations()) {
+            for (CellRangeAddress address : validation.getRegions().getCellRangeAddresses()) {
+                if (address.getFirstColumn() <= columnIndex && address.getLastColumn() >= columnIndex) {
+                    return true;
+                }
+            }
+        }
+        return false;
+    }
+
+    private SysDictData buildDictData(String label) {
+        SysDictData dictData = new SysDictData();
+        dictData.setDictLabel(label);
+        dictData.setDictValue(label);
+        return dictData;
+    }
+}