package com.ruoyi.payment.controller;

import com.ruoyi.order.api.feignClient.ChargingOrderClient;
import com.ruoyi.order.api.feignClient.OrderClient;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import java.util.HashMap;
import java.util.Map;

import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoInteractions;

@ExtendWith(MockitoExtension.class)
class IcbcPayControllerDispatchTest {

    @Mock
    private ChargingOrderClient chargingOrderClient;

    @Mock
    private OrderClient orderClient;

    @InjectMocks
    private IcbcPayController controller;

    @Test
    void dispatchesVipOrderByOrderPrefix() {
        controller.dispatchPaymentCallback(callback("HY202607210001", "ICBC-VIP-1", "1"));

        verify(orderClient).vipCallBack("HY202607210001", "ICBC-VIP-1");
        verifyNoInteractions(chargingOrderClient);
    }

    @Test
    void dispatchesShoppingOrderByOrderPrefix() {
        controller.dispatchPaymentCallback(callback("GW202607210001", "ICBC-SHOP-1", "1"));

        verify(orderClient).callBack("GW202607210001", "ICBC-SHOP-1");
        verifyNoInteractions(chargingOrderClient);
    }

    @Test
    void keepsChargingOrderPaymentTypeDispatch() {
        controller.dispatchPaymentCallback(callback("CD202607210001", "ICBC-CHARGE-1", "1"));

        verify(chargingOrderClient).chargingOrderWXCallback("CD202607210001", "ICBC-CHARGE-1", "1");
        verifyNoInteractions(orderClient);
    }

    private Map<String, String> callback(String outTradeNo, String transactionId, String attach) {
        Map<String, String> callback = new HashMap<>();
        callback.put("out_trade_no", outTradeNo);
        callback.put("transaction_id", transactionId);
        callback.put("attach", attach);
        return callback;
    }
}