From d587d3e6d1865b37e2c1951dc53c776dece654c2 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 25 Aug 2020 10:14:00 +0800
Subject: [PATCH] modify whole close order

---
 src/test/java/com/xcong/excoin/WholeTest.java |   41 +++++++++++++++++++++++++++++++++++++++--
 1 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/src/test/java/com/xcong/excoin/WholeTest.java b/src/test/java/com/xcong/excoin/WholeTest.java
index 1243c34..72454e6 100644
--- a/src/test/java/com/xcong/excoin/WholeTest.java
+++ b/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);
+            }
+        }
+    }
+
+
 }

--
Gitblit v1.9.1