diff options
Diffstat (limited to 'test/units/plugins/connection/test_ssh.py')
-rw-r--r-- | test/units/plugins/connection/test_ssh.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/test/units/plugins/connection/test_ssh.py b/test/units/plugins/connection/test_ssh.py index 2ca4d93a8e..0bb5c19cc5 100644 --- a/test/units/plugins/connection/test_ssh.py +++ b/test/units/plugins/connection/test_ssh.py @@ -80,6 +80,8 @@ class TestConnectionBaseClass(unittest.TestCase): conn._build_command.return_value = 'ssh something something' conn._run = MagicMock() conn._run.return_value = (0, 'stdout', 'stderr') + conn.get_option = MagicMock() + conn.get_option.return_value = True res, stdout, stderr = conn.exec_command('ssh') res, stdout, stderr = conn.exec_command('ssh', 'this is some data') @@ -538,6 +540,8 @@ class TestSSHConnectionRetries(object): self.conn._build_command = MagicMock() self.conn._build_command.return_value = 'ssh' + self.conn.get_option = MagicMock() + self.conn.get_option.return_value = True return_code, b_stdout, b_stderr = self.conn.exec_command('ssh', 'some data') assert return_code == 0 @@ -563,6 +567,8 @@ class TestSSHConnectionRetries(object): self.conn._build_command = MagicMock() self.conn._build_command.return_value = 'ssh' + self.conn.get_option = MagicMock() + self.conn.get_option.return_value = True pytest.raises(AnsibleConnectionFailure, self.conn.exec_command, 'ssh', 'some data') assert self.mock_popen.call_count == 10 @@ -575,6 +581,8 @@ class TestSSHConnectionRetries(object): self.conn._build_command = MagicMock() self.conn._build_command.return_value = 'ssh' + self.conn.get_option = MagicMock() + self.conn.get_option.return_value = True self.mock_popen.side_effect = [Exception('bad')] * 10 pytest.raises(Exception, self.conn.exec_command, 'ssh', 'some data') |