Add a test for failure.

Author: samuel-williams-shopify

lib/async/container/supervisor/connection.rb

@@ -251,6 +251,7 @@ def write(**message)
 				#
 				# @parameter timeout [Numeric, nil] Optional timeout for the call.
 				# @parameter message [Hash] The call message.
+				# @raises [IOError | Errno::EPIPE | Errno::ECONNRESET] If the write fails.
 				# @returns [Hash] The response.
 				def call(timeout: nil, **message)
 					id = next_id

test/async/container/connection.rb

@@ -21,7 +21,7 @@ def dispatch(call)
 	let(:connection) {Async::Container::Supervisor::Connection.new(stream)}
 
 	with "dispatch" do
-		it "handles failed writes" do
+		it "handles failed writes when dispatching a call" do
 			stream.write(JSON.dump({id: 1, do: :test}) << "\n")
 			stream.rewind
 
@@ -39,6 +39,16 @@ def dispatch(call)
 
 			expect(connection.calls).to be(:empty?)
 		end
+		
+		it "handles failed writes when making a call" do
+			expect(stream).to receive(:write).and_raise(IOError, "Test error")
+			
+			expect do
+				connection.call(do: :test)
+			end.to raise_exception(IOError, message: be =~ /Test error/)
+			
+			expect(connection.calls).to be(:empty?)
+		end
 	end
 
 	with subject::Call do