From 8ff5fc0f83912e0fbaa9f834bef12778537c8433 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Fri, 21 Aug 2020 15:43:26 +0800 Subject: [PATCH] modify --- src/test/java/com/xcong/excoin/WholeTest.java | 29 +++++++++++++++++++++++++++++ 1 files changed, 29 insertions(+), 0 deletions(-) diff --git a/src/test/java/com/xcong/excoin/WholeTest.java b/src/test/java/com/xcong/excoin/WholeTest.java index e685505..72454e6 100644 --- a/src/test/java/com/xcong/excoin/WholeTest.java +++ b/src/test/java/com/xcong/excoin/WholeTest.java @@ -1,10 +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; @@ -30,6 +34,9 @@ @Autowired private RabbitOrderService rabbitOrderService; + @Autowired + private MemberWalletContractDao memberWalletContractDao; + @Test public void forceClosePriceTest() { MemberEntity memberEntity = memberDao.selectById(5L); @@ -47,5 +54,27 @@ 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); + } + } + } + } -- Gitblit v1.9.1