Add the cancel event for local/remote stream cancellation

1 files changed, 37 insertions, 3 deletions

diff --git a/test/event_SUITE.erl b/test/event_SUITE.erl
index ace509a..d043a11 100644
--- a/test/event_SUITE.erl
+++ b/test/event_SUITE.erl
@@ -35,7 +35,7 @@ groups() ->
 	%% We currently do not support Websocket over HTTP/2.
 	WsTests = [T || T <- Tests, lists:sublist(atom_to_list(T), 3) =:= "ws_"],
 	[
-		{http, [parallel], Tests -- PushTests},
+		{http, [parallel], Tests -- [cancel_remote|PushTests]},
 		{http2, [parallel], Tests -- [protocol_changed|WsTests]}
 	].
 
@@ -629,6 +629,36 @@ do_ws_send_frame(Config, EventName) ->
 	} = do_receive_event(EventName),
 	gun:close(Pid).
 
+cancel(Config) ->
+	doc("Confirm that the cancel event callback is called when we cancel a stream."),
+	{ok, Pid, _} = do_gun_open(Config),
+	{ok, _} = gun:await_up(Pid),
+	StreamRef = gun:post(Pid, "/stream", []),
+	gun:cancel(Pid, StreamRef),
+	ReplyTo = self(),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo,
+		endpoint := local,
+		reason := cancel
+	} = do_receive_event(?FUNCTION_NAME),
+	gun:close(Pid).
+
+cancel_remote(Config) ->
+	doc("Confirm that the cancel event callback is called "
+		"when the remote endpoint cancels the stream."),
+	{ok, Pid, _} = do_gun_open(Config),
+	{ok, _} = gun:await_up(Pid),
+	StreamRef = gun:post(Pid, "/stream", []),
+	ReplyTo = self(),
+	#{
+		stream_ref := StreamRef,
+		reply_to := ReplyTo,
+		endpoint := remote,
+		reason := _
+	} = do_receive_event(cancel),
+	gun:close(Pid).
+
 disconnect(Config) ->
 	doc("Confirm that the disconnect event callback is called on disconnect."),
 	{ok, OriginPid, OriginPort} = init_origin(tcp),
@@ -639,7 +669,7 @@ disconnect(Config) ->
 	exit(OriginPid, shutdown),
 	#{
 		reason := closed
-	} = do_receive_event(disconnect),
+	} = do_receive_event(?FUNCTION_NAME),
 	gun:close(Pid).
 
 terminate(Config) ->
@@ -649,7 +679,7 @@ terminate(Config) ->
 	#{
 		state := _,
 		reason := shutdown
-	} = do_receive_event(terminate),
+	} = do_receive_event(?FUNCTION_NAME),
 	ok.
 
 %% Internal.
@@ -785,6 +815,10 @@ ws_send_frame_end(EventData, Pid) ->
 	Pid ! {?FUNCTION_NAME, EventData},
 	Pid.
 
+cancel(EventData, Pid) ->
+	Pid ! {?FUNCTION_NAME, EventData},
+	Pid.
+
 disconnect(EventData, Pid) ->
 	Pid ! {?FUNCTION_NAME, EventData},
 	Pid.