Test that a disconnect after the write request but before the data is an error

tests/acceptance/fakes/dest/close_after_hello.rb

@@ -9,7 +9,8 @@
 require 'flexnbd/fake_dest'
 include FlexNBD
 
-server = FakeDest.new( *ARGV )
+addr, port = *ARGV
+server = FakeDest.new( addr, port )
 client = server.accept( "Timed out waiting for a connection" )
 client.write_hello
 client.close

tests/acceptance/fakes/dest/close_after_write.rb

@@ -10,7 +10,8 @@
 require 'flexnbd/fake_dest'
 include FlexNBD
 
-server = FakeDest.new( *ARGV )
+addr, port = *ARGV
+server = FakeDest.new( addr, port )
 client = server.accept( "Timed out waiting for a connection" )
 client.write_hello
 client.read_request

tests/acceptance/fakes/dest/error_on_write.rb

@@ -4,7 +4,8 @@
 require 'flexnbd/fake_dest'
 include FlexNBD
 
-server = FakeDest.new( *ARGV )
+addr, port = *ARGV
+server = FakeDest.new( addr, port )
 client = server.accept
 
 client.write_hello

tests/acceptance/fakes/dest/hang_after_connect.rb

@@ -10,7 +10,8 @@
 require 'flexnbd/fake_dest'
 include FlexNBD
 
-server = FakeDest.new( *ARGV )
+addr, port = *ARGV
+server = FakeDest.new( addr, port )
 client = server.accept( "Client didn't make a connection" )
 
 # Sleep for one second past the timeout (a bit of slop in case ruby

tests/acceptance/fakes/dest/hang_after_write.rb

@@ -8,7 +8,8 @@
 require 'flexnbd/fake_dest'
 include FlexNBD
 
-server = FakeDest.new( *ARGV )
+addr, port = *ARGV
+server = FakeDest.new( addr, port )
 client1 = server.accept( server )
 client1.write_hello
 client1.read_request

tests/acceptance/fakes/dest/hello_wrong_magic.rb

@@ -6,7 +6,8 @@
 require 'flexnbd/fake_dest'
 include FlexNBD
 
-server = FakeDest.new( *ARGV )
+addr, port = *ARGV
+server = FakeDest.new( addr, port )
 client1 = server.accept
 
 # Launch a second thread so that we can spot the reconnection attempt

tests/acceptance/fakes/dest/hello_wrong_size.rb

@@ -7,7 +7,8 @@
 require 'flexnbd/fake_dest'
 include FlexNBD
 
-server = FakeDest.new( *ARGV )
+addr, port = *ARGV
+server = FakeDest.new( addr, port )
 client = server.accept
 
 t = Thread.new do

tests/acceptance/fakes/dest/reject_acl.rb

@@ -5,7 +5,8 @@
 require 'flexnbd/fake_dest'
 include FlexNBD
 
-server = FakeDest.new( *ARGV )
+addr, port = *ARGV
+server = FakeDest.new( addr, port )
 server.accept.close
 
 server.close

tests/acceptance/fakes/source/close_after_write.rb

@@ -0,0 +1,29 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+
+# We connect, pause the server, issue a write request, disconnect,
+# then cont the server. This ensures that our disconnect happens
+# while the server is trying to read the write data.
+
+require 'flexnbd/fake_source'
+include FlexNBD
+
+addr, port, srv_pid = *ARGV
+
+client = FakeSource.new( addr, port, "Timed out connecting" )
+client.read_hello
+Process.kill( "STOP", srv_pid.to_i )
+client.write_write_request( 0, 8 )
+client.close
+Process.kill( "CONT", srv_pid.to_i )
+
+# This sleep ensures that we don't return control to the test runner
+# too soon, giving the flexnbd process time to fall over if it's going
+# to.
+sleep(0.25)
+
+# ...and can we reconnect?
+client2 = FakeSource.new( addr, port, "Timed out connecting" )
+client2.close
+
+exit(0)