cfc/cfc-backend/src/test/java/com/etotem/cfc/service/FamilyWorkbenchPlanWriteTest.java

package com.etotem.cfc.service;

import com.etotem.cfc.dto.FamilyPlanCreateDTO;
import com.etotem.cfc.entity.HealthPlan;
import com.etotem.cfc.mapper.HealthPlanMapper;
import com.etotem.cfc.service.impl.HealthPlanServiceImpl;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

/**
 * 方案新增/编辑契约：create 固定 family_id/status 默认 active；update 校验归属（防跨家庭越权）。
 */
class FamilyWorkbenchPlanWriteTest {

    @Mock private HealthPlanMapper planMapper;
    @InjectMocks private HealthPlanServiceImpl service;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
    }

    private FamilyPlanCreateDTO dto() {
        FamilyPlanCreateDTO d = new FamilyPlanCreateDTO();
        d.setPlanType("gut");
        d.setMemberName("小明");
        d.setGoal("改善肠道菌群");
        d.setTotalDays(30);
        return d;
    }

    @Test
    void 创建方案_familyId固定_status默认active_返回id() {
        when(planMapper.insert(any(HealthPlan.class))).thenAnswer(inv -> {
            HealthPlan p = inv.getArgument(0);
            p.setId(321L);
            return 1;
        });
        Long id = service.createFamilyPlan(5L, 88L, dto());
        ArgumentCaptor<HealthPlan> cap = ArgumentCaptor.forClass(HealthPlan.class);
        verify(planMapper).insert(cap.capture());
        HealthPlan p = cap.getValue();
        assertEquals(5L, p.getFamilyId());
        assertEquals("active", p.getStatus());
        assertEquals("gut", p.getPlanType());
        assertEquals("改善肠道菌群", p.getGoal());
        assertEquals(Integer.valueOf(30), p.getTotalDays());
        assertEquals(321L, id);
    }

    @Test
    void 编辑方案_跨家庭_拒绝() {
        HealthPlan other = new HealthPlan();
        other.setId(55L);
        other.setFamilyId(9L);
        when(planMapper.selectById(55L)).thenReturn(other);
        RuntimeException e = assertThrows(RuntimeException.class,
                () -> service.updateFamilyPlan(55L, 5L, dto()));
        assertTrue(e.getMessage().contains("方案不属于该家庭"));
    }

    @Test
    void 编辑方案_同家庭_字段更新() {
        HealthPlan p = new HealthPlan();
        p.setId(55L);
        p.setFamilyId(5L);
        p.setGoal("旧目标");
        when(planMapper.selectById(55L)).thenReturn(p);
        when(planMapper.updateById(any(HealthPlan.class))).thenReturn(1);
        boolean ok = service.updateFamilyPlan(55L, 5L, dto());
        verify(planMapper).updateById(p);
        assertTrue(ok);
        assertEquals("改善肠道菌群", p.getGoal());
    }
}