def test_execute_command(self, ssh_client_cls): # Arrange host = "localhost" cmd = "ls" ssh = SSHEntity(host) ssh._ssh_cls = ssh_client_cls session_mock = mock.Mock() session_mock.recv_ready.return_value = True session_mock.exit_status_ready.return_value = True session_mock.recv.return_value = "dummy" stream_mock = mock.Mock() stream_mock.channel = session_mock ssh.ssh_client.exec_command.return_value = None, stream_mock, None def change_status(y): session_mock.recv_ready.return_value = False return "dummy" session_mock.recv.side_effect = change_status ssh.get_new_session = lambda: session_mock # Act ssh.execute_command(cmd) # Assert ssh.ssh_client.exec_command.assert_called_once_with(cmd) self.assertTrue(session_mock.recv.call_count >= 1)
def test_session(self): # Arrange host, port, username, password = get_ssh_config() # Act ssh = SSHEntity(host, port, username, password) session = ssh.get_new_session() # Assert self.assertIsNotNone(session)
def test_session(self): # Arrange host, port, username, password = get_ssh_config() # Act ssh = SSHEntity(host, port, username, password) session = ssh.get_new_session() # Assert self.assertIsNotNone(session)
def test_get_new_session(self, ssh_client_cls): # Arrange host = "localhost" ssh = SSHEntity(host) ssh._ssh_client = ssh_client_cls() # Act session = ssh.get_new_session() # Assert self.assertEquals(ssh._ssh_client.get_transport.call_count, 1)
def test_get_new_session(self, ssh_client_cls): # Arrange host = "localhost" ssh = SSHEntity(host) ssh._ssh_client = ssh_client_cls() # Act session = ssh.get_new_session() # Assert self.assertEquals(ssh._ssh_client.get_transport.call_count, 1)
def test_execute_command(self, ssh_client_cls): # Arrange host = "localhost" cmd = "ls" ssh = SSHEntity(host) ssh._ssh_cls = ssh_client_cls session_mock = mock.Mock() session_mock.recv_ready.return_value = True session_mock.exit_status_ready.return_value = True session_mock.recv.return_value = "dummy" stream_mock = mock.Mock() stream_mock.channel = session_mock ssh.ssh_client.exec_command.return_value = None, stream_mock, None def change_status(y): session_mock.recv_ready.return_value = False return "dummy" session_mock.recv.side_effect = change_status ssh.get_new_session = lambda: session_mock # Act ssh.execute_command(cmd) # Assert ssh.ssh_client.exec_command.assert_called_once_with(cmd) self.assertTrue(session_mock.recv.call_count >= 1)