aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/handlers/ws_set_options_commands_h.erl20
-rw-r--r--test/ws_handler_SUITE.erl31
2 files changed, 50 insertions, 1 deletions
diff --git a/test/handlers/ws_set_options_commands_h.erl b/test/handlers/ws_set_options_commands_h.erl
new file mode 100644
index 0000000..88d4e72
--- /dev/null
+++ b/test/handlers/ws_set_options_commands_h.erl
@@ -0,0 +1,20 @@
+%% This module sets options based on the frame received.
+
+-module(ws_set_options_commands_h).
+-behavior(cowboy_websocket).
+
+-export([init/2]).
+-export([websocket_handle/2]).
+-export([websocket_info/2]).
+
+init(Req, RunOrHibernate) ->
+ {cowboy_websocket, Req, RunOrHibernate,
+ #{idle_timeout => infinity}}.
+
+websocket_handle(Frame={text, <<"idle_timeout_short">>}, State=run) ->
+ {[{set_options, #{idle_timeout => 500}}, Frame], State};
+websocket_handle(Frame={text, <<"idle_timeout_short">>}, State=hibernate) ->
+ {[{set_options, #{idle_timeout => 500}}, Frame], State, hibernate}.
+
+websocket_info(_Info, State) ->
+ {[], State}.
diff --git a/test/ws_handler_SUITE.erl b/test/ws_handler_SUITE.erl
index 6e2bb41..8c45dc7 100644
--- a/test/ws_handler_SUITE.erl
+++ b/test/ws_handler_SUITE.erl
@@ -51,7 +51,8 @@ init_dispatch(Name) ->
{"/handle", ws_handle_commands_h, RunOrHibernate},
{"/info", ws_info_commands_h, RunOrHibernate},
{"/active", ws_active_commands_h, RunOrHibernate},
- {"/deflate", ws_deflate_commands_h, RunOrHibernate}
+ {"/deflate", ws_deflate_commands_h, RunOrHibernate},
+ {"/set_options", ws_set_options_commands_h, RunOrHibernate}
]}]).
%% Support functions for testing using Gun.
@@ -257,3 +258,31 @@ websocket_deflate_ignore_if_not_negotiated(Config) ->
{ok, {text, <<"Hello.">>}} = receive_ws(ConnPid, StreamRef)
end || _ <- lists:seq(1, 10)],
ok.
+
+websocket_set_options_idle_timeout(Config) ->
+ doc("The idle_timeout option can be modified using the "
+ "command {set_options, Opts} at runtime."),
+ ConnPid = gun_open(Config),
+ StreamRef = gun:ws_upgrade(ConnPid, "/set_options"),
+ receive
+ {gun_upgrade, ConnPid, StreamRef, [<<"websocket">>], _} ->
+ ok;
+ {gun_response, ConnPid, _, _, Status, Headers} ->
+ exit({ws_upgrade_failed, Status, Headers});
+ {gun_error, ConnPid, StreamRef, Reason} ->
+ exit({ws_upgrade_failed, Reason})
+ after 1000 ->
+ error(timeout)
+ end,
+ %% We don't send anything for a short while and confirm
+ %% that idle_timeout does not trigger.
+ {error, timeout} = gun:await(ConnPid, StreamRef, 2000),
+ %% Trigger the change in idle_timeout and confirm that
+ %% the connection gets closed soon after.
+ gun:ws_send(ConnPid, {text, <<"idle_timeout_short">>}),
+ receive
+ {gun_down, ConnPid, _, _, _, _} ->
+ ok
+ after 2000 ->
+ error(timeout)
+ end.