Add ws_upgrade/protocol_changed events

And ensure that Websocket triggers all the request/response
events.

1 files changed, 106 insertions, 3 deletions

diff --git a/test/event_SUITE.erl b/test/event_SUITE.erl
index 1863f15..466b7bd 100644
--- a/test/event_SUITE.erl
+++ b/test/event_SUITE.erl
@@ -32,7 +32,7 @@ groups() ->
 	Tests = ct_helper:all(?MODULE),
 	[
 		{http, [parallel], Tests},
-		{http2, [parallel], Tests}
+		{http2, [parallel], Tests -- [ws_upgrade, ws_upgrade_all_events, protocol_changed]}
 	].
 
 init_per_suite(Config) ->
@@ -42,7 +42,8 @@ init_per_suite(Config) ->
 			{"/empty", empty_h, []},
 			{"/inform", inform_h, []},
 			{"/stream", stream_h, []},
-			{"/trailers", trailers_h, []}
+			{"/trailers", trailers_h, []},
+			{"/ws", ws_echo, []}
 		]}])
 	}}),
 	OriginPort = ranch:get_port(?MODULE),
@@ -165,7 +166,8 @@ request_end(Config) ->
 	doc("Confirm that the request_end event callback is called."),
 	do_request_end(Config, ?FUNCTION_NAME),
 	do_request_end_headers(Config, ?FUNCTION_NAME),
-	do_request_end_headers_content_length(Config, ?FUNCTION_NAME).
+	do_request_end_headers_content_length(Config, ?FUNCTION_NAME),
+	do_request_end_headers_content_length_0(Config, ?FUNCTION_NAME).
 
 do_request_end(Config, EventName) ->
 	{ok, Pid, _} = do_gun_open(Config),
@@ -209,6 +211,21 @@ do_request_end_headers_content_length(Config, EventName) ->
 	} = do_receive_event(EventName),
 	gun:close(Pid).
 
+do_request_end_headers_content_length_0(Config, EventName) ->
+	{ok, Pid, _} = do_gun_open(Config),
+	{ok, _} = gun:await_up(Pid),
+	StreamRef = gun:put(Pid, "/", [
+		{<<"content-type">>, <<"text/plain">>},
+		{<<"content-length">>, <<"0">>}
+	]),
+	gun:data(Pid, StreamRef, fin, <<>>),
+	ReplyTo = self(),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo
+	} = do_receive_event(EventName),
+	gun:close(Pid).
+
 response_start(Config) ->
 	doc("Confirm that the request_start event callback is called."),
 	{ok, Pid, _} = do_gun_open(Config),
@@ -286,6 +303,84 @@ do_response_end(Config, EventName, Path) ->
 	} = do_receive_event(EventName),
 	gun:close(Pid).
 
+ws_upgrade(Config) ->
+	doc("Confirm that the ws_upgrade event callback is called."),
+	{ok, Pid, _} = do_gun_open(Config),
+	{ok, _} = gun:await_up(Pid),
+	StreamRef = gun:ws_upgrade(Pid, "/ws"),
+	ReplyTo = self(),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo,
+		opts := #{}
+	} = do_receive_event(?FUNCTION_NAME),
+	gun:close(Pid).
+
+ws_upgrade_all_events(Config) ->
+	doc("Confirm that a Websocket upgrade triggers all relevant events."),
+	{ok, Pid, OriginPort} = do_gun_open(Config),
+	{ok, _} = gun:await_up(Pid),
+	StreamRef = gun:ws_upgrade(Pid, "/ws"),
+	ReplyTo = self(),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo,
+		opts := #{}
+	} = do_receive_event(ws_upgrade),
+	Authority = iolist_to_binary([<<"localhost:">>, integer_to_list(OriginPort)]),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo,
+		function := ws_upgrade,
+		method := <<"GET">>,
+		authority := EventAuthority1,
+		path := "/ws",
+		headers := [_|_]
+	} = do_receive_event(request_start),
+	Authority = iolist_to_binary(EventAuthority1),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo,
+		function := ws_upgrade,
+		method := <<"GET">>,
+		authority := EventAuthority2,
+		path := "/ws",
+		headers := [_|_]
+	} = do_receive_event(request_headers),
+	Authority = iolist_to_binary(EventAuthority2),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo
+	} = do_receive_event(request_end),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo
+	} = do_receive_event(response_start),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo,
+		status := 101,
+		headers := [_|_]
+	} = do_receive_event(response_inform),
+	#{
+		protocol := ws
+	} = do_receive_event(protocol_changed),
+	gun:close(Pid).
+
+protocol_changed(Config) ->
+	doc("Confirm that the protocol_changed event callback is called."),
+	do_protocol_changed_ws(Config, ?FUNCTION_NAME).
+	%% @todo do_protocol_changed_connect
+
+do_protocol_changed_ws(Config, EventName) ->
+	{ok, Pid, _} = do_gun_open(Config),
+	{ok, _} = gun:await_up(Pid),
+	_ = gun:ws_upgrade(Pid, "/ws"),
+	#{
+		protocol := ws
+	} = do_receive_event(EventName),
+	gun:close(Pid).
+
 disconnect(Config) ->
 	doc("Confirm that the disconnect event callback is called on disconnect."),
 	{ok, OriginPid, OriginPort} = init_origin(tcp),
@@ -380,6 +475,14 @@ response_end(EventData, Pid) ->
 	Pid ! {?FUNCTION_NAME, EventData},
 	Pid.
 
+ws_upgrade(EventData, Pid) ->
+	Pid ! {?FUNCTION_NAME, EventData},
+	Pid.
+
+protocol_changed(EventData, Pid) ->
+	Pid ! {?FUNCTION_NAME, EventData},
+	Pid.
+
 disconnect(EventData, Pid) ->
 	Pid ! {?FUNCTION_NAME, EventData},
 	Pid.