修复模板新增后电池组保存找不到模板

yeg

2026-08-29 07494f2ce0ff6aa7b266b2b059553297043c1885

修复模板新增后电池组保存找不到模板

 ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/TBatteryGroupTemplateController.java

@@ -54,7 +54,12 @@
    public AjaxResult add(@Validated @RequestBody TBatteryGroupTemplate template)
    {
        template.setCreateBy(getUsername());
        return toAjax(batteryGroupTemplateService.insertBatteryGroupTemplate(template));
        int rows = batteryGroupTemplateService.insertBatteryGroupTemplate(template);
        if (rows <= 0 || template.getTemplateId() == null)
        {
            return error("模板保存失败");
        }
        return success(java.util.Map.of("templateId", template.getTemplateId()));
    }

    /** 编辑电池组模板并替换字段关系。 */

 ruoyi-admin/src/test/java/com/ruoyi/web/controller/system/TBatteryGroupTemplateControllerTest.java

New file
@@ -0,0 +1,45 @@
package com.ruoyi.web.controller.system;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.mock;

import java.util.Collections;
import java.util.Map;
import org.junit.jupiter.api.Test;
import org.springframework.test.util.ReflectionTestUtils;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.system.domain.TBatteryGroupTemplate;
import com.ruoyi.system.service.ITBatteryGroupTemplateService;

/** 电池组模板新增接口合同测试。 */
class TBatteryGroupTemplateControllerTest
{
    @Test
    void addShouldReturnGeneratedTemplateId()
    {
        ITBatteryGroupTemplateService service = mock(ITBatteryGroupTemplateService.class);
        doAnswer(invocation -> {
            invocation.<TBatteryGroupTemplate>getArgument(0).setTemplateId(21L);
            return 1;
        }).when(service).insertBatteryGroupTemplate(any(TBatteryGroupTemplate.class));

        TBatteryGroupTemplateController controller = new TBatteryGroupTemplateController()
        {
            @Override
            public String getUsername()
            {
                return "admin";
            }
        };
        ReflectionTestUtils.setField(controller, "batteryGroupTemplateService", service);
        TBatteryGroupTemplate template = new TBatteryGroupTemplate();
        template.setTemplateName("模板");
        template.setFieldIds(Collections.singletonList(3L));

        AjaxResult result = controller.add(template);

        assertEquals(21L, ((Map<?, ?>) result.get("data")).get("templateId"));
    }
}