实现测试详情持久化接口

yeg

2026-09-03 e35881f24e349e865d7a1dbbebc1a7e55ecc7a16

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
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"));
    }
}