From 0ffaebff959e5399c7d6fdc8672e32c288d34754 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= <essen@ninenines.eu>
Date: Fri, 11 Aug 2017 18:21:02 +0200
Subject: Add missing Websocket clause detecting owner is gone

---
 test/gun_SUITE.erl | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

(limited to 'test')

diff --git a/test/gun_SUITE.erl b/test/gun_SUITE.erl
index dbd51b5..842d40c 100644
--- a/test/gun_SUITE.erl
+++ b/test/gun_SUITE.erl
@@ -64,6 +64,35 @@ detect_owner_gone(_) ->
 		error(timeout)
 	end.
 
+detect_owner_gone_ws(_) ->
+	Self = self(),
+	spawn(fun() ->
+		{ok, ConnPid} = gun:open("echo.websocket.org", 80),
+		Self ! {conn, ConnPid},
+		gun:await_up(ConnPid),
+		gun:ws_upgrade(ConnPid, "/", []),
+		receive
+			{gun_ws_upgrade, Pid, ok, _} ->
+				ok
+		after 1000 ->
+			error(timeout)
+		end
+	end),
+	Pid = receive
+		{conn, C} ->
+			C
+	after 1000 ->
+		error(timeout)
+	end,
+	Ref = monitor(process, Pid),
+	receive
+		{'DOWN', Ref, process, Pid, {{owner_gone, _}, _}} ->
+			ok
+	after 1000 ->
+		true = erlang:is_process_alive(Pid),
+		error(timeout)
+	end.
+
 gone_reason(_) ->
 	doc("The last connection failure must be propagated."),
 	{ok, Pid} = gun:open("localhost", 12345, #{retry => 0}),
-- 
cgit v1.2.3