def test_wss_url(self):
url = utils.wss_url(None)
self.assertEqual(CONSTANTS.WSS_URLS.get("ndax_main"), url)
url = utils.wss_url("ndax_main")
self.assertEqual(CONSTANTS.WSS_URLS.get("ndax_main"), url)
url = utils.wss_url("ndax_testnet")
self.assertEqual(CONSTANTS.WSS_URLS.get("ndax_testnet"), url)
async def _create_websocket_connection(self) -> NdaxWebSocketAdaptor:
"""
Initialize WebSocket client for UserStreamDataSource
"""
try:
ws = await websockets.connect(ndax_utils.wss_url(self._domain))
return NdaxWebSocketAdaptor(throttler=self._throttler, websocket=ws)
except asyncio.CancelledError:
raise
except Exception as ex:
self.logger().network(f"Unexpected error occurred during {CONSTANTS.EXCHANGE_NAME} WebSocket Connection "
f"({ex})")
raise