From 452d4c40c58d832ac3b1e85e6ffc523690814b3c Mon Sep 17 00:00:00 2001
From: Gyandeep Singh <gyandeeps@gmail.com>
Date: Wed, 16 Oct 2024 09:27:15 -0500
Subject: [PATCH] add tests

---
 providers/tests/ssh/hooks/test_ssh.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/providers/tests/ssh/hooks/test_ssh.py b/providers/tests/ssh/hooks/test_ssh.py
index 6b3e5dcdae116..ac40d01c6f68c 100644
--- a/providers/tests/ssh/hooks/test_ssh.py
+++ b/providers/tests/ssh/hooks/test_ssh.py
@@ -362,6 +362,7 @@ def test_ssh_connection_with_password(self, ssh_mock):
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=None,
             )
 
     @mock.patch("airflow.providers.ssh.hooks.ssh.paramiko.SSHClient")
@@ -381,6 +382,7 @@ def test_ssh_connection_without_password(self, ssh_mock):
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=None,
             )
 
     @mock.patch("airflow.providers.ssh.hooks.ssh.SSHTunnelForwarder")
@@ -569,6 +571,7 @@ def test_ssh_connection_with_private_key_extra(self, ssh_mock):
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=None,
             )
 
     @mock.patch("airflow.providers.ssh.hooks.ssh.paramiko.SSHClient")
@@ -592,6 +595,7 @@ def test_ssh_connection_with_private_key_passphrase_extra(self, ssh_mock):
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=None,
             )
 
     @mock.patch("airflow.providers.ssh.hooks.ssh.paramiko.SSHClient")
@@ -654,6 +658,7 @@ def test_ssh_connection_with_conn_timeout(self, ssh_mock):
             password="password",
             conn_timeout=20,
             key_file="fake.file",
+            auth_timeout=10,
         )
 
         with hook.get_conn():
@@ -668,6 +673,7 @@ def test_ssh_connection_with_conn_timeout(self, ssh_mock):
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=10,
             )
 
     @mock.patch("airflow.providers.ssh.hooks.ssh.paramiko.SSHClient")
@@ -695,6 +701,7 @@ def test_ssh_connection_with_conn_timeout_and_timeout(self, ssh_mock):
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=None,
             )
 
     @mock.patch("airflow.providers.ssh.hooks.ssh.paramiko.SSHClient")
@@ -718,6 +725,7 @@ def test_ssh_connection_with_timeout_extra(self, ssh_mock):
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=None,
             )
 
     @mock.patch("airflow.providers.ssh.hooks.ssh.paramiko.SSHClient")
@@ -743,6 +751,7 @@ def test_ssh_connection_with_conn_timeout_extra(self, ssh_mock):
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=None,
             )
 
     @mock.patch("airflow.providers.ssh.hooks.ssh.paramiko.SSHClient")
@@ -768,6 +777,7 @@ def test_ssh_connection_with_timeout_extra_and_conn_timeout_extra(self, ssh_mock
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=None,
             )
 
     @pytest.mark.parametrize(
@@ -837,6 +847,7 @@ def test_ssh_connection_with_all_timeout_param_and_extra_combinations(
                 port="port",
                 sock=None,
                 look_for_keys=True,
+                auth_timeout=None,
             )
 
     @pytest.mark.parametrize(
@@ -901,6 +912,7 @@ def test_ssh_with_extra_disabled_algorithms(self, ssh_mock):
                 sock=None,
                 look_for_keys=True,
                 disabled_algorithms=TEST_DISABLED_ALGORITHMS,
+                auth_timeout=None,
             )
 
     @mock.patch("airflow.providers.ssh.hooks.ssh.paramiko.SSHClient")