From 85edcaaad5de321263d7e69bcb16543d8bffc213 Mon Sep 17 00:00:00 2001 From: Timothy Pogue Date: Sat, 8 Oct 2022 18:39:33 -0600 Subject: [PATCH] update websocket exception caught in api_ws --- freqtrade/rpc/api_server/api_ws.py | 4 ++-- tests/data/test_dataprovider.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/freqtrade/rpc/api_server/api_ws.py b/freqtrade/rpc/api_server/api_ws.py index 50a008bf6..986f18477 100644 --- a/freqtrade/rpc/api_server/api_ws.py +++ b/freqtrade/rpc/api_server/api_ws.py @@ -134,13 +134,13 @@ async def message_endpoint( except ( WebSocketDisconnect, - websockets.exceptions.ConnectionClosed + websockets.exceptions.WebSocketException ): # Handle client disconnects logger.info(f"Consumer disconnected - {channel}") except Exception as e: logger.info(f"Consumer connection failed - {channel}") - logger.exception(e) + logger.debug(e, exc_info=e) # Handle cases like - # RuntimeError('Cannot call "send" once a closed message has been sent') finally: diff --git a/tests/data/test_dataprovider.py b/tests/data/test_dataprovider.py index 8500fa06c..de5d7ee6f 100644 --- a/tests/data/test_dataprovider.py +++ b/tests/data/test_dataprovider.py @@ -177,7 +177,7 @@ def test_get_producer_df(mocker, default_conf, ohlcv_history): assert la == empty_la # the data is added, should return that added dataframe - dataprovider._add_external_df(pair, ohlcv_history, now, timeframe, candle_type) + dataprovider._add_producer_df(pair, ohlcv_history, now, timeframe, candle_type) dataframe, la = dataprovider.get_producer_df(pair, timeframe, candle_type) assert len(dataframe) > 0 assert la > empty_la