From e570e277165ef5ffe2f6eac058629d33ae8a3f18 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mikl=C3=B3s=20Fazekas?= <mfazekas@szemafor.com>
Date: Mon, 27 Feb 2023 11:02:37 +0100
Subject: test: added integration test for connection timeout

---
 test/integration/test_handshake_timeout.rb | 32 ++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 test/integration/test_handshake_timeout.rb

diff --git a/test/integration/test_handshake_timeout.rb b/test/integration/test_handshake_timeout.rb
new file mode 100644
index 0000000..c297377
--- /dev/null
+++ b/test/integration/test_handshake_timeout.rb
@@ -0,0 +1,32 @@
+require_relative 'common'
+require 'net/ssh'
+
+class TestHandshakeTimeout < NetSSHTest
+  include IntegrationTestHelpers
+
+  def with_non_responding_server(&block)
+    port = "4444"
+    pipe = IO.popen("/bin/nc -l -k -p #{port}")
+    begin
+      yield(port)
+    ensure
+      Process.kill("TERM", pipe.pid)
+    end
+  end
+
+  def nc_port_open?(port)
+    Socket.tcp("localhost", port, connect_timeout: 1) { true } rescue false # rubocop:disable Style/RescueModifier
+  end
+
+  def test_error_exitstatus
+    with_non_responding_server do |port|
+      sleep(0.1) until nc_port_open?(port.to_i)
+
+      assert_raises(Net::SSH::ConnectionTimeout, 'timeout during server version negotiating') do
+        Net::SSH.start("localhost", "net_ssh_1", password: 'foopwd', port: port, timeout: 1) do |ssh|
+          ssh.exec! "exit 42"
+        end
+      end
+    end
+  end
+end
-- 
cgit v1.2.1