def test_run(self, mock_ssh_client): mock_ssh_client.exec_command.return_value = (None, 'stdout', 'stderr') a = ACSClient(mock_ssh_client) a.connect('somehost', 'someusername') stdout, stderr = a.run('somecommand') self.assertEqual(stdout, 'stdout') self.assertEqual(stderr, 'stderr')