Python mocking paramiko sftpclient打开

Question

我想嘲笑帕拉米科SFTPClient.open（）并返回一个文件以测试我的文件解析代码。但是，当我如下设置Mock时，它返回一个magicmock实例，而不是打开的文件，它的值等于打开的文件。在

我想要paramiko.SSHClient.open_sftp参数.open等于打开的测试文件。我觉得这应该发生，但事实并非如此。我错过什么了吗？有谁能解释一下，我做错了什么，我是怎么误解嘲笑的？在

这是我的代码：

@app.route('/ipmivlan/<girls_name>', methods=['POST'])
def find_the_girl(girls_name):
  this_is_the_girl = None
  remote_file = None
  try:
    client = paramiko.SSHClient()
    # Auto add host key if not found/unknown
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    client.connect(host, username=username, password=password)
    sftp_client = client.open_sftp()
    remote_filename = '/usr/local/{0}'.format(girls_name.lower())
    remote_file = sftp_client.open(remote_filename, mode='r')
    print remote_file
    pprint(remote_file.__dict__)
    this_is_the_girl = parse_config(remote_file)
  except paramiko.ssh_exception.AuthenticationException:
    return_message = "Authentication has failed."
    app.logger.info(return_message)
    app.logger.info("Traceback: {0}".format(traceback.format_exc()))
    return jsonify(return_code=1, message=return_message, traceback=traceback.format_exc())
  except paramiko.ssh_exception.BadHostKeyException:
    return_message = "Server hostkey could not be verified"
    app.logger.info(return_message)
    app.logger.info("Traceback: {0}".format(traceback.format_exc()))
    return jsonify(return_code=1, message=return_message, traceback=traceback.format_exc())
  except paramiko.ssh_exception.SSHException:
    return_message = "Error connecting or establishing SSH Session."
    app.logger.info(return_message)
    app.logger.info("Traceback: {0}".format(traceback.format_exc()))
    return jsonify(return_code=1, message=return_message, traceback=traceback.format_exc())
  except paramiko.ssh_exception.socket.error:
    return_message = "Socket error occurred while connecting."
    app.logger.info(return_message)
    app.logger.info("Traceback: {0}".format(traceback.format_exc()))
    return jsonify(return_code=1, message=return_message, traceback=traceback.format_exc())
  except IOError:
    return_message = "IOError while trying to read remote file {0}".format(remote_filename)
    app.logger.info(return_message)
    app.logger.info("Traceback: {0}".format(traceback.format_exc()))
    return jsonify(return_code=1, message=return_message, traceback=traceback.format_exc())
  except: 
    return_message = "Unknown exception: {0}".format(str(sys.exc_info()[0]))
    app.logger.info(return_message)
    app.logger.info("Traceback: {0}".format(traceback.format_exc()))
    return jsonify(return_code=1, message=return_message, traceback=traceback.format_exc())
  finally:
    if remote_file is not None:
      remote_file.close()

if this_is_the_girl is None:
  return jsonify(return_code=0, message="Girl not found.")
elif this_is_the_girl == 'this is the girl:
  return jsonify(return_code=0, message="Successfully found girl")
elif this_is_the_girl == 'it's no longer your film, this is the girl.'
  return jsonify(return_code=50, message="GG no re.")


def parse_config(self, file):
  this_is_the_girl = None
  for line in file:
    # print line
    if re.findall('what girl, for what? what is this ray?', line):
      return "this is the girl"
    elif re.findall('that is considered one of the finest espressos in the world', line):
      return "it's no longer your film, this is the girl."

这是我的测试代码：

^{pr2}$

这是我的测试返回的结果：

<MagicMock name='paramiko.SSHClient().open_sftp().open()' id='4426176848'>
{'__str__': <MagicMock name='paramiko.SSHClient().open_sftp().open().__str__' id='4426525136'>,
 '_mock_call_args': None,
 '_mock_call_args_list': [],
 '_mock_call_count': 0,
 '_mock_called': False,
 '_mock_children': {'__str__': <MagicMock name='paramiko.SSHClient().open_sftp().open().__str__' id='4426525136'>},
 '_mock_delegate': None,
 '_mock_methods': None,
 '_mock_mock_calls': [call.__str__()],
 '_mock_name': None,
 '_mock_new_name': '()',
 '_mock_new_parent': <MagicMock name='paramiko.SSHClient().open_sftp().open' id='4426130896'>,
 '_mock_parent': None,
 '_mock_return_value': <open file 'this_is_the_girl_test_data.txt', mode 'r' at 0x107ab1930>,
 '_mock_side_effect': None,
 '_mock_unsafe': False,
 '_mock_wraps': None,
 '_spec_class': None,
 '_spec_set': None,
 '_spec_signature': None,
 'method_calls': []}

Answer 1

我已经想好了。当我通读这些行时，它返回一个iterable，所以我实际上需要模仿iter。在

paramiko.SSHClient().open_sftp().open().return_value = self.test_file

必须是

^{pr2}$