package cn.iocoder.yudao.module.pay.service.transfer;

import cn.iocoder.yudao.framework.test.core.ut.BaseDbAndRedisUnitTest;
import cn.iocoder.yudao.module.pay.dal.dataobject.channel.PayChannelDO;
import cn.iocoder.yudao.module.pay.dal.dataobject.transfer.PayTransferDO;
import cn.iocoder.yudao.module.pay.dal.mysql.transfer.PayTransferMapper;
import cn.iocoder.yudao.module.pay.dal.redis.no.PayNoRedisDAO;
import cn.iocoder.yudao.module.pay.enums.transfer.PayTransferStatusEnum;
import cn.iocoder.yudao.module.pay.framework.pay.config.PayProperties;
import cn.iocoder.yudao.module.pay.framework.pay.core.client.dto.transfer.PayTransferRespDTO;
import cn.iocoder.yudao.module.pay.service.app.PayAppService;
import cn.iocoder.yudao.module.pay.service.channel.PayChannelService;
import cn.iocoder.yudao.module.pay.service.notify.PayNotifyService;
import jakarta.annotation.Resource;
import org.junit.jupiter.api.Test;
import org.springframework.context.annotation.Import;
import org.springframework.test.context.bean.override.mockito.MockitoBean;

import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
import static org.junit.jupiter.api.Assertions.assertEquals;

/**
 * {@link PayTransferServiceImpl} 的单元测试类
 *
 * @author 芋道源码
 */
@Import({PayTransferServiceImpl.class, PayNoRedisDAO.class})
public class PayTransferServiceTest extends BaseDbAndRedisUnitTest {

    @Resource
    private PayTransferServiceImpl transferService;

    @Resource
    private PayTransferMapper transferMapper;

    @MockitoBean
    private PayProperties payProperties;
    @MockitoBean
    private PayAppService appService;
    @MockitoBean
    private PayChannelService channelService;
    @MockitoBean
    private PayNotifyService notifyService;

    @Test
    public void testNotifyTransferProgressing_fillChannelPackageInfoIfAbsent() {
        // mock 数据（PayTransferDO）：同步任务先把 WAITING 更新为 PROCESSING，但是不返回 package 信息
        PayTransferDO transfer = randomPojo(PayTransferDO.class,
                o -> o.setAppId(10L)
                        .setNo("T110")
                        .setStatus(PayTransferStatusEnum.PROCESSING.getStatus())
                        .setChannelPackageInfo(null));
        transferMapper.insert(transfer);
        // 准备参数
        PayChannelDO channel = randomPojo(PayChannelDO.class, o -> o.setAppId(10L));
        PayTransferRespDTO createNotify = PayTransferRespDTO.processingOf("WX_TRANSFER_110",
                "T110", "create");
        createNotify.setChannelPackageInfo("package-info-110");

        // 调用：模拟发起转账接口随后返回 PROCESSING，并携带微信确认收款 package 信息
        transferService.notifyTransfer(channel, createNotify);

        // 断言：已是 PROCESSING 时仍会补写 package 信息
        PayTransferDO dbTransfer = transferMapper.selectById(transfer.getId());
        assertEquals(PayTransferStatusEnum.PROCESSING.getStatus(), dbTransfer.getStatus());
        assertEquals("package-info-110", dbTransfer.getChannelPackageInfo());
    }

    @Test
    public void testNotifyTransferProgressing_notOverwriteChannelPackageInfo() {
        // mock 数据（PayTransferDO）：已经存在 package 信息
        PayTransferDO transfer = randomPojo(PayTransferDO.class,
                o -> o.setAppId(10L)
                        .setNo("T110")
                        .setStatus(PayTransferStatusEnum.PROCESSING.getStatus())
                        .setChannelPackageInfo("package-info-110"));
        transferMapper.insert(transfer);
        // 准备参数
        PayChannelDO channel = randomPojo(PayChannelDO.class, o -> o.setAppId(10L));
        PayTransferRespDTO syncNotify = PayTransferRespDTO.processingOf("WX_TRANSFER_110",
                "T110", "sync");

        // 调用：后续同步任务不返回 package 信息
        transferService.notifyTransfer(channel, syncNotify);

        // 断言：已有 package 信息不被空值覆盖
        PayTransferDO dbTransfer = transferMapper.selectById(transfer.getId());
        assertEquals("package-info-110", dbTransfer.getChannelPackageInfo());
    }

}