Merge pull request #2918 from rackspace/setup_disconnect

[rackspace|compute] Updated setup method to retry once on disconnect

lib/fog/rackspace/models/compute_v2/server.rb

@@ -570,6 +570,8 @@ def change_admin_password(password)
         def setup(credentials = {})
           requires :ssh_ip_address, :identity, :public_key, :username
 
+          retried_disconnect = false
+
           commands = [
             %{mkdir .ssh},
             %{echo "#{public_key}" >> ~/.ssh/authorized_keys},
@@ -585,6 +587,14 @@ def setup(credentials = {})
         rescue Errno::ECONNREFUSED
           sleep(1)
           retry
+        # Ubuntu 12.04 images seem to be disconnecting during the ssh setup process.
+        # This rescue block is an effort to address that issue.
+        rescue Net::SSH::Disconnect
+          unless retried_disconnect
+            retried_disconnect = true
+            sleep(1)
+            retry
+          end
         end
 
         def virtual_interfaces