Improved error handling.

socketry · Sep 11, 2024 · d4da564 · d4da564
1 parent 5b5288d
commit d4da564
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/async/websocket/client.rb b/test/async/websocket/client.rb
@@ -17,9 +17,11 @@
 					connection.write(message)
 				end
 
-				connection.close
+				connection.shutdown
 			rescue Protocol::WebSocket::ClosedError
 				# Ignore this error.
+			ensure
+				connection.close
 			end or Protocol::HTTP::Response[404, {}, []]
 		end
 	end
@@ -56,6 +58,7 @@
 				expect(message.to_str).to be == "Hello World!"
 
 				connection.close
+
 				expect(task.children).to be(:empty?)
 			end.wait
 		end
@@ -68,6 +71,7 @@
 				expect(message.to_str).to be == "Hello World!"
 
 				connection.close(Protocol::WebSocket::Error::GOING_AWAY, "Bye!")
+
 				expect(task.children).to be(:empty?)
 			end.wait
 		end