eladkal commented on code in PR #28184:
URL: https://github.com/apache/airflow/pull/28184#discussion_r1041854170
##########
tests/providers/ssh/hooks/test_ssh.py:
##########
@@ -955,3 +955,15 @@ def
test_ssh_connection_with_no_host_key_check_true_and_allow_host_key_changes_f
ssh_mock.return_value.set_missing_host_key_policy.call_args[0][0],
paramiko.AutoAddPolicy
)
assert ssh_mock.return_value.load_host_keys.called is False
+
+ def test_connection_success(self):
+ hook = SSHHook(ssh_conn_id="ssh_default")
+ status, msg = hook.test_connection()
+ assert status is True
+ assert msg == "Connection successfully tested"
+ def test_connection_failure(self):
Review Comment:
```suggestion
assert msg == "Connection successfully tested"
def test_connection_failure(self):
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
