Merge pull request #46 from Mai-Te-Pora/testfix/websocket_futures_chnage

Because of the two new markets(btc_h21 and eth_h21) the test did not …
Mai-Te-Pora · Mar 2, 2021 · ac26631 · ac26631
2 parents 6035e01 + af4a395
commit ac26631
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/tests/test_ws_get_market_stats.py b/tests/test_ws_get_market_stats.py
@@ -38,7 +38,9 @@ def test_get_market_stats_structure(self):
                 'nneo2_usdc1': market_stats,
                 'swth_eth1': market_stats,
                 'swth_usdc1': market_stats,
-                'wbtc1_usdc1': market_stats
+                'wbtc1_usdc1': market_stats,
+                'btc_h21': market_stats,
+                'eth_h21': market_stats,
 
             }
         }
@@ -99,8 +101,9 @@ def test_get_market_stats_with_param(self):
                 'nneo2_usdc1': market_stats,
                 'swth_eth1': market_stats,
                 'swth_usdc1': market_stats,
-                'wbtc1_usdc1': market_stats
-
+                'wbtc1_usdc1': market_stats,
+                'btc_h21': market_stats,
+                'eth_h21': market_stats
             }
         }
 

diff --git a/tests/test_ws_subscribe_market_stats.py b/tests/test_ws_subscribe_market_stats.py
@@ -43,8 +43,9 @@ def test_subscribe_market_stats_structure(self):
                 'nneo2_usdc1': market_stats,
                 'swth_eth1': market_stats,
                 'swth_usdc1': market_stats,
-                'wbtc1_usdc1': market_stats
-
+                'wbtc1_usdc1': market_stats,
+                'btc_h21': market_stats,
+                'eth_h21': market_stats,
             }
         }
 
@@ -58,7 +59,6 @@ async def on_connect():
 
         async def on_message(message: dict):
             # save response into self
-            print(message)
             self.response.append(message)
 
         try:
@@ -76,8 +76,6 @@ async def on_message(message: dict):
         if len(self.response) < 2:
             self.skipTest(f"Did not receive candlesticks within time, test can not finish.")
 
-        print(self.response[1]["result"].keys())
-
         channel_subscription: dict = self.response[0]
         self.assertDictStructure(expect_subscription, channel_subscription)