package com.xcong.excoin;

import com.xcong.excoin.common.contants.AppContants;
import com.xcong.excoin.common.enumerates.CoinTypeEnum;
import com.xcong.excoin.modules.contract.dao.ContractHoldOrderDao;
import com.xcong.excoin.modules.contract.entity.ContractHoldOrderEntity;
import com.xcong.excoin.modules.contract.service.RabbitOrderService;
import com.xcong.excoin.modules.member.dao.MemberDao;
import com.xcong.excoin.modules.member.dao.MemberWalletContractDao;
import com.xcong.excoin.modules.member.entity.MemberEntity;
import com.xcong.excoin.modules.member.entity.MemberWalletContractEntity;
import com.xcong.excoin.utils.CalculateUtil;
import com.xcong.excoin.utils.ThreadPoolUtils;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;

/**
 * @author wzy
 * @date 2020-08-14
 **/
@SpringBootTest
public class WholeTest {

    @Resource
    private MemberDao memberDao;
    @Autowired
    private ContractHoldOrderDao contractHoldOrderDao;

    @Autowired
    private RabbitOrderService rabbitOrderService;

    @Autowired
    private MemberWalletContractDao memberWalletContractDao;

    @Test
    public void forceClosePriceTest() {
        MemberEntity memberEntity = memberDao.selectById(5L);
        try {
            CalculateUtil.getForceSetPriceForWhole("BTC/USDT", memberEntity);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @Test
    public void closingOrderTest() {
        List<Long> ids = new ArrayList<>();
        ids.add(836L);
        rabbitOrderService.cancelHoldOrder(ids);
    }

    @Test
    public void wholeContractWalletTest() {
        List<MemberEntity> memberEntities = memberDao.selectAllMember();
        for (MemberEntity member : memberEntities) {
            CoinTypeEnum[] values = CoinTypeEnum.values();
            for (CoinTypeEnum value : values) {
                MemberWalletContractEntity walletContract = new MemberWalletContractEntity();
                if (value.name().equals(CoinTypeEnum.USDT.name())) {
                    continue;
                }

                walletContract.setMemberId(member.getId());
                walletContract.setAvailableBalance(AppContants.INIT_MONEY);
                walletContract.setFrozenBalance(AppContants.INIT_MONEY);
                walletContract.setTotalBalance(AppContants.INIT_MONEY);
                walletContract.setBorrowedFund(AppContants.INIT_MONEY);
                walletContract.setWalletCode(value.name());
                memberWalletContractDao.insert(walletContract);
            }
        }
    }


}