3 files changed, 93 insertions, 0 deletions
diff --git a/test/handlers/ws_handle_commands_h.erl b/test/handlers/ws_handle_commands_h.erl
new file mode 100644
index 0000000..da3ffad
--- /dev/null
+++ b/test/handlers/ws_handle_commands_h.erl
@@ -0,0 +1,31 @@
+%% This module takes commands from the x-commands header
+%% and returns them in the websocket_handle/2 callback.
+
+-module(ws_handle_commands_h).
+-behavior(cowboy_websocket).
+
+-export([init/2]).
+-export([websocket_init/1]).
+-export([websocket_handle/2]).
+-export([websocket_info/2]).
+
+init(Req=#{pid := Pid}, RunOrHibernate) ->
+	Commands0 = cowboy_req:header(<<"x-commands">>, Req),
+	Commands = binary_to_term(base64:decode(Commands0)),
+	case Commands of
+		bad -> ct_helper_error_h:ignore(Pid, cowboy_websocket, handler_call, 6);
+		_ -> ok
+	end,
+	{cowboy_websocket, Req, {Commands, RunOrHibernate}}.
+
+websocket_init(State) ->
+	{[], State}.
+
+websocket_handle(_, State={Commands, run}) ->
+	{Commands, State};
+websocket_handle(_, State={Commands, hibernate}) ->
+	{Commands, State, hibernate}.
+
+websocket_info(_, State) ->
+	{[], State}.
+
diff --git a/test/handlers/ws_info_commands_h.erl b/test/handlers/ws_info_commands_h.erl
new file mode 100644
index 0000000..d596473
--- /dev/null
+++ b/test/handlers/ws_info_commands_h.erl
@@ -0,0 +1,32 @@
+%% This module takes commands from the x-commands header
+%% and returns them in the websocket_info/2 callback.
+%% This callback is triggered via a message.
+
+-module(ws_info_commands_h).
+-behavior(cowboy_websocket).
+
+-export([init/2]).
+-export([websocket_init/1]).
+-export([websocket_handle/2]).
+-export([websocket_info/2]).
+
+init(Req=#{pid := Pid}, RunOrHibernate) ->
+	Commands0 = cowboy_req:header(<<"x-commands">>, Req),
+	Commands = binary_to_term(base64:decode(Commands0)),
+	case Commands of
+		bad -> ct_helper_error_h:ignore(Pid, cowboy_websocket, handler_call, 6);
+		_ -> ok
+	end,
+	{cowboy_websocket, Req, {Commands, RunOrHibernate}}.
+
+websocket_init(State) ->
+	self() ! shoot,
+	{[], State}.
+
+websocket_handle(_, State) ->
+	{[], State}.
+
+websocket_info(_, State={Commands, run}) ->
+	{Commands, State};
+websocket_info(_, State={Commands, hibernate}) ->
+	{Commands, State, hibernate}.
diff --git a/test/handlers/ws_init_commands_h.erl b/test/handlers/ws_init_commands_h.erl
new file mode 100644
index 0000000..8bae352
--- /dev/null
+++ b/test/handlers/ws_init_commands_h.erl
@@ -0,0 +1,30 @@
+%% This module takes commands from the x-commands header
+%% and returns them in the websocket_init/1 callback.
+
+-module(ws_init_commands_h).
+-behavior(cowboy_websocket).
+
+-export([init/2]).
+-export([websocket_init/1]).
+-export([websocket_handle/2]).
+-export([websocket_info/2]).
+
+init(Req=#{pid := Pid}, RunOrHibernate) ->
+	Commands0 = cowboy_req:header(<<"x-commands">>, Req),
+	Commands = binary_to_term(base64:decode(Commands0)),
+	case Commands of
+		bad -> ct_helper_error_h:ignore(Pid, cowboy_websocket, handler_call, 6);
+		_ -> ok
+	end,
+	{cowboy_websocket, Req, {Commands, RunOrHibernate}}.
+
+websocket_init(State={Commands, run}) ->
+	{Commands, State};
+websocket_init(State={Commands, hibernate}) ->
+	{Commands, State, hibernate}.
+
+websocket_handle(_, State) ->
+	{[], State}.
+
+websocket_info(_, State) ->
+	{[], State}.