diff options
Diffstat (limited to 'test/support')
| -rw-r--r-- | test/support/websocket_client.ex | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/test/support/websocket_client.ex b/test/support/websocket_client.ex index cf2972c38..7163bbd41 100644 --- a/test/support/websocket_client.ex +++ b/test/support/websocket_client.ex @@ -41,6 +41,12 @@ defmodule Pleroma.Integration.WebsocketClient do {:ok, state} end + @impl true + def handle_disconnect(conn_status, state) do + send(state.sender, {:close, conn_status}) + {:ok, state} + end + @doc false @impl true def handle_info({:text, msg}, state) do |