예제 #1
0
 def test_ssh_execute(self, mock_ssh, mock_ssh_execute):
     mock_ssh.return_value = '<SSH object at 0x1024641d0>'
     mock_ssh_execute.return_value = {'return_code': 0, 'stderr': '',
                                      'stdout': self.username + '\n'}
     remote_server = SSH(self.hostname, self.username, self.password)
     remote_server.execute(self.command)
     self.assertEqual(
         mock_ssh_execute.return_value,
         {'return_code': 0, 'stderr': '', 'stdout': '{0}\n'
             .format(self.username)}
     )
예제 #2
0
 def test_ssh_execute_timeout(self, mock_ssh, mock_ssh_execute):
     mock_ssh.return_value = '<SSH object at 0x1024641d0>'
     mock_ssh_execute.return_exception = "ValueError('TimeOut)"
     instance = SSH(self.hostname, self.username, self.password, timeout=30)
     self.assertRaises(Exception, instance.execute(self.command))