add tests

apache · Oct 16, 2024 · 452d4c4 · 452d4c4
1 parent 5064bfa
commit 452d4c4
Showing 1 changed file with 12 additions and 0 deletions.
diff --git 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")