connect failure scenarios

tests/fakes/source/close_after_connect.rb

@@ -0,0 +1,28 @@
+#!/usr/bin/env ruby
+
+# Connects to the destination server, then immediately disconnects,
+# simulating a source crash.
+#
+# It then connects again, to check that the destination is still
+# listening.
+
+addr, port = *ARGV
+require 'socket'
+require 'timeout'
+
+begin
+  Timeout.timeout( 2 ) do
+    sock = TCPSocket.open( addr, port.to_i )
+    sock.close
+  end
+rescue Timeout::Error
+  $stderr.puts "Failed to connect"
+  exit 1
+end
+
+Timeout.timeout( 2 ) do
+  sock = TCPSocket.open( addr, port.to_i )
+  sock.close
+end
+
+exit 0