From 3aa7168404e62f9ea9e290380b93569eff4372a8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= <essen@ninenines.eu>
Date: Mon, 13 May 2019 13:18:01 +0200
Subject: Add the disconnect event callback

---
 test/event_SUITE.erl | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'test')

diff --git a/test/event_SUITE.erl b/test/event_SUITE.erl
index 383ac46..50a1ce2 100644
--- a/test/event_SUITE.erl
+++ b/test/event_SUITE.erl
@@ -20,6 +20,7 @@
 
 -import(ct_helper, [config/2]).
 -import(ct_helper, [doc/1]).
+-import(gun_test, [init_origin/1]).
 
 all() ->
 	ct_helper:all(?MODULE).
@@ -99,6 +100,21 @@ connect_end_ok(Config) ->
 	} = do_receive_event(connect_end),
 	gun:close(Pid).
 
+disconnect(_) ->
+	doc("Confirm that the disconnect event callback is called on disconnect."),
+	Self = self(),
+	Opts = #{event_handler => {?MODULE, Self}},
+	{ok, OriginPid, OriginPort} = init_origin(tcp),
+	{ok, Pid} = gun:open("localhost", OriginPort, Opts),
+	{ok, http} = gun:await_up(Pid),
+	%% We make the origin exit to trigger a disconnect.
+	unlink(OriginPid),
+	exit(OriginPid, shutdown),
+	#{
+		reason := closed
+	} = do_receive_event(disconnect),
+	gun:close(Pid).
+
 %% Internal.
 
 do_receive_event(Event) ->
@@ -122,3 +138,7 @@ connect_start(EventData, Pid) ->
 connect_end(EventData, Pid) ->
 	Pid ! {?FUNCTION_NAME, EventData},
 	Pid.
+
+disconnect(EventData, Pid) ->
+	Pid ! {?FUNCTION_NAME, EventData},
+	Pid.
-- 
cgit v1.2.3