From 5fdd4b505c388ba585b55feacf978f87d7f3e592 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 26 Aug 2020 14:41:12 +0800
Subject: [PATCH] modify

---
 src/test/java/com/xcong/excoin/WholeTest.java |   64 +++++++++++++++++++++++++++++++-
 1 files changed, 62 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..899aaf4 100644
--- a/src/test/java/com/xcong/excoin/WholeTest.java
+++ b/src/test/java/com/xcong/excoin/WholeTest.java
@@ -1,9 +1,18 @@
 package com.xcong.excoin;
 
+import com.alibaba.fastjson.JSONObject;
+import com.xcong.excoin.common.contants.AppContants;
+import com.xcong.excoin.common.enumerates.CoinTypeEnum;
+import com.xcong.excoin.common.enumerates.RabbitPriceTypeEnum;
 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.rabbit.pricequeue.OrderModel;
+import com.xcong.excoin.rabbit.producer.OrderProducer;
 import com.xcong.excoin.utils.CalculateUtil;
 import com.xcong.excoin.utils.ThreadPoolUtils;
 import org.junit.jupiter.api.Test;
@@ -11,6 +20,9 @@
 import org.springframework.boot.test.context.SpringBootTest;
 
 import javax.annotation.Resource;
+import java.math.BigDecimal;
+import java.util.ArrayList;
+import java.util.List;
 
 /**
  * @author wzy
@@ -24,6 +36,12 @@
     @Autowired
     private ContractHoldOrderDao contractHoldOrderDao;
 
+    @Autowired
+    private RabbitOrderService rabbitOrderService;
+
+    @Autowired
+    private MemberWalletContractDao memberWalletContractDao;
+
     @Test
     public void forceClosePriceTest() {
         MemberEntity memberEntity = memberDao.selectById(5L);
@@ -36,8 +54,50 @@
 
     @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);
+            }
+        }
+    }
+
+
+    @Test
+    public void bondAmountTest() {
+        BigDecimal openPrice = BigDecimal.valueOf(412.52124800);
+
+        System.out.println(CalculateUtil.getBondAmount(openPrice, BigDecimal.valueOf(6), 1, 5));
+    }
+
+    @Autowired
+    private OrderProducer orderProducer;
+
+    @Test
+    public void sendMsg() {
+        List<OrderModel> list = new ArrayList<>();
+        OrderModel orderModel = new OrderModel(1L, RabbitPriceTypeEnum.ENTRUST_OPEN_MORE.getValue(), "111", "BTC");
+        list.add(orderModel);
+        orderProducer.sendLimitClose(JSONObject.toJSONString(list));
     }
 
 }

--
Gitblit v1.9.1