update websocket exception caught in api_ws

This commit is contained in:
Timothy Pogue 2022-10-08 18:39:33 -06:00
parent f9b3b0ef77
commit 85edcaaad5
2 changed files with 3 additions and 3 deletions

View File

@ -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:

View File

@ -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