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