Helius
2020-08-20 665aae9e293c4e97be597ddc4f21b4c1b1edca61
src/test/java/com/xcong/excoin/WholeTest.java
@@ -1,9 +1,14 @@
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;
@@ -11,6 +16,8 @@
import org.springframework.boot.test.context.SpringBootTest;
import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
/**
 * @author wzy
@@ -24,6 +31,12 @@
    @Autowired
    private ContractHoldOrderDao contractHoldOrderDao;
    @Autowired
    private RabbitOrderService rabbitOrderService;
    @Autowired
    private MemberWalletContractDao memberWalletContractDao;
    @Test
    public void forceClosePriceTest() {
        MemberEntity memberEntity = memberDao.selectById(5L);
@@ -36,8 +49,32 @@
    @Test
    public void closingOrderTest() {
        ContractHoldOrderEntity holdOrderEntity = contractHoldOrderDao.selectById(822L);
        System.out.println(holdOrderEntity.getPositionType());
        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);
            }
        }
    }
}