Tweak timeouts to fix intermittent failures

1 files changed, 62 insertions, 61 deletions

diff --git a/test/gun_SUITE.erl b/test/gun_SUITE.erl
index acaee64..f4587e3 100644
--- a/test/gun_SUITE.erl
+++ b/test/gun_SUITE.erl
@@ -52,66 +52,6 @@ atom_hostname(_) ->
 	[<<"host: localhost:", _/bits>>] = [L || <<"host: ", _/bits>> = L <- Lines],
 	gun:close(Pid).
 
-stream_info_http(_) ->
-	doc("Ensure the function gun:stream_info/2 works as expected for HTTP/1.1."),
-	{ok, _, OriginPort} = init_origin(tcp, http,
-		fun(_, ClientSocket, ClientTransport) ->
-			%% Give some time to detect the cancel.
-			timer:sleep(100),
-			%% Then terminate the stream.
-			ClientTransport:send(ClientSocket,
-				"HTTP/1.1 200 OK\r\n"
-				"content-length: 0\r\n"
-				"\r\n"
-			),
-			timer:sleep(200)
-		end),
-	{ok, Pid} = gun:open("localhost", OriginPort),
-	{ok, http} = gun:await_up(Pid),
-	{ok, undefined} = gun:stream_info(Pid, make_ref()),
-	StreamRef = gun:get(Pid, "/"),
-	Self = self(),
-	{ok, #{
-		ref := StreamRef,
-		reply_to := Self,
-		state := running
-	}} = gun:stream_info(Pid, StreamRef),
-	gun:cancel(Pid, StreamRef),
-	{ok, #{
-		ref := StreamRef,
-		reply_to := Self,
-		state := stopping
-	}} = gun:stream_info(Pid, StreamRef),
-	%% Wait a little for the stream to terminate.
-	timer:sleep(200),
-	{ok, undefined} = gun:stream_info(Pid, StreamRef),
-	%% Wait a little more for the connection to terminate.
-	timer:sleep(200),
-	{error, not_connected} = gun:stream_info(Pid, StreamRef),
-	gun:close(Pid).
-
-stream_info_http2(_) ->
-	doc("Ensure the function gun:stream_info/2 works as expected for HTTP/2."),
-	{ok, _, OriginPort} = init_origin(tcp, http2,
-		fun(_, _, _) -> timer:sleep(100) end),
-	{ok, Pid} = gun:open("localhost", OriginPort, #{
-		protocols => [http2]
-	}),
-	{ok, http2} = gun:await_up(Pid),
-	{ok, undefined} = gun:stream_info(Pid, make_ref()),
-	StreamRef = gun:get(Pid, "/"),
-	Self = self(),
-	{ok, #{
-		ref := StreamRef,
-		reply_to := Self,
-		state := running
-	}} = gun:stream_info(Pid, StreamRef),
-	gun:cancel(Pid, StreamRef),
-	%% Wait a little for the connection to terminate.
-	timer:sleep(200),
-	{error, not_connected} = gun:stream_info(Pid, StreamRef),
-	gun:close(Pid).
-
 connect_timeout(_) ->
 	doc("Ensure an integer value for connect_timeout is accepted."),
 	{ok, Pid} = gun:open("localhost", 12345, #{connect_timeout => 1000, retry => 0}),
@@ -422,10 +362,71 @@ retry_timeout(_) ->
 	receive
 		{'DOWN', Ref, process, Pid, {shutdown, _}} ->
 			ok
-	after 400 ->
+	after 800 ->
 		error(shutdown_too_late)
 	end.
 
+stream_info_http(_) ->
+	doc("Ensure the function gun:stream_info/2 works as expected for HTTP/1.1."),
+	{ok, _, OriginPort} = init_origin(tcp, http,
+		fun(_, ClientSocket, ClientTransport) ->
+			%% Give some time to detect the cancel.
+			timer:sleep(100),
+			%% Then terminate the stream.
+			ClientTransport:send(ClientSocket,
+				"HTTP/1.1 200 OK\r\n"
+				"content-length: 0\r\n"
+				"\r\n"
+			),
+			timer:sleep(200)
+		end),
+	{ok, Pid} = gun:open("localhost", OriginPort),
+	{ok, http} = gun:await_up(Pid),
+	{ok, undefined} = gun:stream_info(Pid, make_ref()),
+	StreamRef = gun:get(Pid, "/"),
+	Self = self(),
+	{ok, #{
+		ref := StreamRef,
+		reply_to := Self,
+		state := running
+	}} = gun:stream_info(Pid, StreamRef),
+	gun:cancel(Pid, StreamRef),
+	{ok, #{
+		ref := StreamRef,
+		reply_to := Self,
+		state := stopping
+	}} = gun:stream_info(Pid, StreamRef),
+	%% Wait a little for the stream to terminate.
+	timer:sleep(200),
+	{ok, undefined} = gun:stream_info(Pid, StreamRef),
+	%% Wait a little more for the connection to terminate.
+	timer:sleep(200),
+	{error, not_connected} = gun:stream_info(Pid, StreamRef),
+	gun:close(Pid).
+
+stream_info_http2(_) ->
+	doc("Ensure the function gun:stream_info/2 works as expected for HTTP/2."),
+	{ok, _, OriginPort} = init_origin(tcp, http2,
+		fun(_, _, _) -> timer:sleep(100) end),
+	{ok, Pid} = gun:open("localhost", OriginPort, #{
+		protocols => [http2]
+	}),
+	{ok, http2} = gun:await_up(Pid),
+	timer:sleep(100), %% Give enough time for the handshake to fully complete.
+	{ok, undefined} = gun:stream_info(Pid, make_ref()),
+	StreamRef = gun:get(Pid, "/"),
+	Self = self(),
+	{ok, #{
+		ref := StreamRef,
+		reply_to := Self,
+		state := running
+	}} = gun:stream_info(Pid, StreamRef),
+	gun:cancel(Pid, StreamRef),
+	%% Wait a little for the connection to terminate.
+	timer:sleep(200),
+	{error, not_connected} = gun:stream_info(Pid, StreamRef),
+	gun:close(Pid).
+
 shutdown_reason(_) ->
 	doc("The last connection failure must be propagated."),
 	{ok, Pid} = gun:open("localhost", 12345, #{retry => 0}),