update test

iohannes-long · iohannes-long · commit c656c8d73a35 · 2022-02-07T12:12:16.000+08:00
diff --git a/tests/coin_swap/test_rest_account.py b/tests/coin_swap/test_rest_account.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.coin_swap.rest.account import Account
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestAccount(unittest.TestCase):
diff --git a/tests/coin_swap/test_rest_order.py b/tests/coin_swap/test_rest_order.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.coin_swap.rest.order import Order
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestOrder(unittest.TestCase):
diff --git a/tests/coin_swap/test_rest_transfer.py b/tests/coin_swap/test_rest_transfer.py
@@ -2,7 +2,8 @@
 import unittest
 
 sys.path.append('../../src')
-from tests.config import ACCESS_KEY, SECRET_KEY
+sys.path.append('../')
+from config import ACCESS_KEY, SECRET_KEY
 from huobi.coin_swap.rest.transfer import Transfer
 from huobi.utils.logger import logger
 
diff --git a/tests/coin_swap/test_rest_trigger_order.py b/tests/coin_swap/test_rest_trigger_order.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.coin_swap.rest.trigger_order import TriggerOrder
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestTriggerOrder(unittest.TestCase):
diff --git a/tests/coin_swap/test_ws_account.py b/tests/coin_swap/test_ws_account.py
@@ -3,9 +3,10 @@
 import time
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.coin_swap.ws.account import Account
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestWsAccount(unittest.TestCase):
diff --git a/tests/futures/test_rest_account.py b/tests/futures/test_rest_account.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.futures.rest.account import Account
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestAccount(unittest.TestCase):
diff --git a/tests/futures/test_rest_order.py b/tests/futures/test_rest_order.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.futures.rest.order import Order
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestOrder(unittest.TestCase):
diff --git a/tests/futures/test_rest_transfer.py b/tests/futures/test_rest_transfer.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
-from tests.config import ACCESS_KEY, SECRET_KEY
+sys.path.append('../')
+from config import ACCESS_KEY, SECRET_KEY
 from huobi.futures.rest.transfer import Transfer
-from huobi.utils.logger import logger
+from utils.logger import logger
 
 class TestRestAccount(unittest.TestCase):
 
diff --git a/tests/futures/test_rest_trigger_order.py b/tests/futures/test_rest_trigger_order.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.futures.rest.trigger_order import TriggerOrder
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestTriggerOrder(unittest.TestCase):
diff --git a/tests/futures/test_ws_account.py b/tests/futures/test_ws_account.py
@@ -3,9 +3,10 @@
 import time
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.futures.ws.account import Account
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestWsAccount(unittest.TestCase):
diff --git a/tests/linear_swap/test_rest_account.py b/tests/linear_swap/test_rest_account.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.linear_swap.rest.account import Account
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestAccount(unittest.TestCase):
diff --git a/tests/linear_swap/test_rest_order.py b/tests/linear_swap/test_rest_order.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.linear_swap.rest.order import Order
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestOrder(unittest.TestCase):
diff --git a/tests/linear_swap/test_rest_transfer.py b/tests/linear_swap/test_rest_transfer.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
-from tests.config import ACCESS_KEY, SECRET_KEY
+sys.path.append('../')
+from config import ACCESS_KEY, SECRET_KEY
 from huobi.linear_swap.rest.transfer import Transfer
-from huobi.utils.logger import logger
+from utils.logger import logger
 
 class TestRestAccount(unittest.TestCase):
 
diff --git a/tests/linear_swap/test_rest_trigger_order.py b/tests/linear_swap/test_rest_trigger_order.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.utils.logger import logger
 from huobi.linear_swap.rest.trigger_order import TriggerOrder
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestTriggerOrder(unittest.TestCase):
diff --git a/tests/linear_swap/test_ws_account.py b/tests/linear_swap/test_ws_account.py
@@ -3,9 +3,10 @@
 import time
 
 sys.path.append('../../src')
+sys.path.append('..')
 from huobi.utils.logger import logger
 from huobi.linear_swap.ws.account import Account
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestWsAccount(unittest.TestCase):
diff --git a/tests/spot/test_rest_account.py b/tests/spot/test_rest_account.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.spot.rest.account import Account
 from huobi.utils.logger import logger
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestAccount(unittest.TestCase):
diff --git a/tests/spot/test_rest_order.py b/tests/spot/test_rest_order.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.spot.rest.order import Order
 from huobi.utils.logger import logger
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestOrder(unittest.TestCase):
diff --git a/tests/spot/test_rest_sub_user.py b/tests/spot/test_rest_sub_user.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.spot.rest.sub_user import SubUser
 from huobi.utils.logger import logger
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestSubUser(unittest.TestCase):
diff --git a/tests/spot/test_rest_wallet.py b/tests/spot/test_rest_wallet.py
@@ -2,9 +2,10 @@
 import unittest
 
 sys.path.append('../../src')
+sys.path.append('../')
 from huobi.spot.rest.wallet import Wallet
 from huobi.utils.logger import logger
-from tests.config import ACCESS_KEY, SECRET_KEY
+from config import ACCESS_KEY, SECRET_KEY
 
 
 class TestRestWallet(unittest.TestCase):
diff --git a/tests/spot/test_ws_account.py b/tests/spot/test_ws_account.py
@@ -0,0 +1,34 @@
+import sys
+import unittest
+import time
+
+sys.path.append('../../src')
+sys.path.append('..')
+from huobi.utils.logger import logger
+from huobi.spot.ws.account import Account
+from config import ACCESS_KEY, SECRET_KEY
+
+
+class TestWsAccount(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        cls.api = Account(ACCESS_KEY, SECRET_KEY)
+
+    def test_sub_orders(self):
+        self.api.sub({"action": "sub", "ch": "orders#trxusdt"},
+                     lambda x: logger.info(x))
+        time.sleep(10)
+        self.api.unsub({"action": "unsub", "ch": "orders#trxusdt"})
+        time.sleep(10)
+
+    def test_sub_accounts(self):
+        self.api.sub({"action": "sub", "ch": "accounts.update#1"},
+                     lambda x: logger.info(x))
+        time.sleep(30)
+        self.api.unsub({"action": "unsub", "ch": "accounts.update#1"})
+        time.sleep(30)
+
+
+if __name__ == '__main__':
+    unittest.main(verbosity=2)
diff --git a/tests/spot/test_ws_market.py b/tests/spot/test_ws_market.py
@@ -0,0 +1,63 @@
+import sys
+import unittest
+import time
+
+sys.path.append('../../src')
+from huobi.utils.logger import logger
+from huobi.spot.ws.market import Market
+
+
+class TestWsMarket(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        cls.api = Market()
+
+    def test_sub_kline(self):
+        ws = self.api.sub({"sub": "market.btcusdt.kline.1min"}, lambda x: logger.info(x))
+        time.sleep(10)
+        ws.close()
+
+    def test_sub_ticker(self):
+        ws = self.api.sub({"sub": "market.btcusdt.ticker"}, lambda x: logger.info(x))
+        time.sleep(10)
+        ws.close()
+
+    def test_sub_depth(self):
+        ws = self.api.sub({"sub": "market.btcusdt.depth.step0"}, lambda x: logger.info(x))
+        time.sleep(10)
+        ws.close()
+
+    def test_sub_mbp_incremental(self):
+        ws = self.api.sub_mbp({"sub": "market.btcusdt.mbp.150"}, lambda x: logger.info(x))
+        time.sleep(10)
+        ws.close()
+
+    def test_sub_mbp_full(self):
+        ws = self.api.sub({"sub": "market.btcusdt.mbp.refresh.20"}, lambda x: logger.info(x))
+        time.sleep(10)
+        ws.close()
+
+    def test_sub_bbo(self):
+        ws = self.api.sub({"sub": "market.btcusdt.bbo"}, lambda x: logger.info(x))
+        time.sleep(10)
+        ws.close()
+
+    def test_sub_trade_detail(self):
+        ws = self.api.sub({"sub": "market.btcusdt.trade.detail"}, lambda x: logger.info(x))
+        time.sleep(10)
+        ws.close()
+
+    def test_sub_detail(self):
+        ws = self.api.sub({"sub": "market.btcusdt.detail"}, lambda x: logger.info(x))
+        time.sleep(10)
+        ws.close()
+
+    def test_sub_etp(self):
+        ws = self.api.sub({"sub": "market.btc3lusdt.etp"}, lambda x: logger.info(x))
+        time.sleep(10)
+        ws.close()
+
+
+if __name__ == '__main__':
+    unittest.main(verbosity=2)
diff --git a/tests/utils/test_http.py b/tests/utils/test_http.py
@@ -2,7 +2,8 @@
 import unittest
 
 sys.path.append('../../src')
-from tests.config import ACCESS_KEY, SECRET_KEY
+sys.path.append('../')
+from config import ACCESS_KEY, SECRET_KEY
 from huobi.host import HOST_FUTURES
 from huobi.utils.logger import logger
 from huobi.utils.http import get, post
diff --git a/tests/utils/test_ws.py b/tests/utils/test_ws.py
@@ -3,7 +3,8 @@
 import time
 
 sys.path.append('../../src')
-from tests.config import ACCESS_KEY, SECRET_KEY
+sys.path.append('../')
+from config import ACCESS_KEY, SECRET_KEY
 from huobi.utils.ws import Ws
 from huobi.utils.logger import logger
 from huobi.host import HOST_FUTURES
