- html - 出于某种原因,IE8 对我的 Sass 文件中继承的 html5 CSS 不友好?
- JMeter 在响应断言中使用 span 标签的问题
- html - 在 :hover and :active? 上具有不同效果的 CSS 动画
- html - 相对于居中的 html 内容固定的 CSS 重复背景?
我想测试是否使用正确的参数调用了 ftp.storbinary()
函数。然而,这是与现场测试不同的模块的一部分。
在 io.py
模块中我有这个方法(它是类的一部分):
def to_ftp(self, output_path, file_name, host, username, password):
"""Upload file to FTP server."""
ftp = ftplib.FTP(host)
ftp.login(username, password)
full_path = os.path.join(output_path, file_name)
with open(full_path, "r") as ftp_file:
ftp.storbinary(" ".join(["STOR", file_name]), ftp_file.read)
我创建了一个 test_io.py
模块,其中包含一系列单元测试。我正在考虑同时修补 open
和 ftplib.FTP
,因为我正在传递给 ftp.storbinary()
ftp_file.read ()
。
@patch("ppc_model.io.ftplib.FTP")
def test_store_call(self, mock_ftp):
"""The *storbinary* call must use the right arguments."""
with patch("ppc_model.io.open", mock_open(read_data=None)) as m:
self.writer.to_ftp(output_path="./output", file_name="output.zip",
host="myftp", username="username", password="password")
mock_ftp.return_value.storbinary.assert_called_once_with(
"STOR output.zip", m.read())
然而,这会返回一个 AttributeError,因为该模块没有属性 open
。我如何确保 Mock 理解我正在尝试模拟内置函数?
是否还有更好的方法来测试我是否将正确的参数传递给 ftp.storbinary
?我对 mock 有点陌生。
编辑:我取得了一些进展。我认为问题是我试图修补错误的对象。使用 open
我想我必须修补 builtins.open
。有点违反直觉。
@patch("ppc_model.io.ftplib.FTP")
def test_store_call(self, mock_ftp):
"""The *storbinary* call must use the right arguments."""
with patch("builtins.open", mock_open(read_data=None), create=True) as m:
self.writer.to_ftp(output_path="./output", file_name="output.zip",
host="myftp", username="username", password="password")
mock_ftp.return_value.storbinary.assert_called_once_with(
"STOR output.zip", m.read())
不幸的是,现在编译器提示 Mock 对象没有 read
方法。
AttributeError: Mock 对象没有属性 'read'
编辑 2: 按照 RedCraig 的建议,我修补了本地命名空间中的 open
方法并传递给 ftp.storbinary
ftp_file 而不是ftp_file.read())。所以当前的单元测试是:
@patch("ppc_model.io.ftplib.FTP")
def test_store_call(self, mock_ftp):
"""The *storbinary* call must use the right arguments."""
with patch("{}.open".format(__name__), mock_open(read_data=None),
create=True) as m:
self.writer.to_ftp(output_path="./output", file_name="output.zip",
host="myftp", username="username", password="password")
mock_ftp.return_value.storbinary.assert_called_once_with(
"STOR output.zip", m)
以及我要测试的代码:
def to_ftp(self, output_path, file_name, host, username, password):
"""Upload file to FTP server."""
ftp = ftplib.FTP(host)
ftp.login(username, password)
full_path = os.path.join(output_path, file_name)
with open(full_path, "r") as ftp_file:
ftp.storbinary(" ".join(["STOR", file_name]), ftp_file)
我得到的当前错误是:
AssertionError: Expected call: storbinary('STOR output.zip', <MagicMock name='open' spec='builtin_function_or_method' id='140210704581632'>)
Actual call: storbinary('STOR output.zip', <_io.TextIOWrapper name='./output/output.zip' mode='r' encoding='UTF-8'>)
最佳答案
AssertionError: Expected call: storbinary('STOR output.zip', )
Actual call: storbinary('STOR output.zip', <_io.TextIOWrapper name='./output/output.zip' mode='r' encoding='UTF-8'>)
对于实际调用,storbinary
被赋予 <_io.TextIOWrapper...
而不是模拟对象。好像 open
的模拟不起作用,也许你的模块名称不正确。
这是 unittest.mock examples ,这个页面有一个模拟打开的例子:
>>> mock = MagicMock(return_value=sentinel.file_handle)
>>> with patch('builtins.open', mock):
... handle = open('filename', 'r')
...
>>> mock.assert_called_with('filename', 'r')
>>> assert handle == sentinel.file_handle, "incorrect file handle returned"
我尝试在另一个模块中模拟打开并且它有效,将它与 assert_called_once_with()
一起使用时有一个问题。当 patching open mock 使用 MagicMock 对象时,每次调用 open()
都会返回一个新对象。因此,使用 open
样式语句修补 with as m
的测试用例的 m
值将不同于来自 to_ftp
的 open 调用,而 assert_called_once_with()
将失败。
虽然我确信 Mock
和 assert_called_once_with
有更好的方法,但我在模拟 StringIO
时求助于使用 open
对象,以便每次调用都返回相同 StringIO
实例:
import ftplib
import os
from io import StringIO
from unittest import TestCase
from unittest.mock import patch
def to_ftp(output_path, file_name, host, username, password):
"""Upload file to FTP server."""
ftp = ftplib.FTP(host)
ftp.login(username, password)
full_path = os.path.join(output_path, file_name)
with open(full_path, "r") as ftp_file:
ftp.storbinary(" ".join(["STOR", file_name]), ftp_file)
class TestOpenMock(TestCase):
@patch("ftplib.FTP")
def test_store_call(self, mock_ftp):
"""The *storbinary* call must use the right arguments."""
open_file = StringIO('test_data')
with patch('%s.open' % __name__, return_value=open_file):
to_ftp(output_path="./output", file_name="output.zip",
host="myftp", username="username", password="password")
mock_ftp.return_value.storbinary.assert_called_once_with(
"STOR output.zip", open_file)
这就过去了。显然,您必须将 '%s.open' % __name__
更改为包含 to_ftp
的模块。
1) 这看起来像您正在寻找的答案: How do I mock an open used in a with statement (using the Mock framework in Python)?它使用 open
模拟 <local module name>.open
。
2)
AttributeError: Mock object has no attribute 'read'
它提示读取,因为您将读取方法传递给 storbinary
:
ftp.storbinary(" ".join(["STOR", file_name]), ftp_file.read)
并且您的 open ftp_file
模拟实例没有该方法。
3) 您正在检查是否使用 storbinary
调用了 m.read()
但您传入了 m.read
。 m.read()
调用读取方法,而 m.read
是对该方法的引用。
mock_ftp.return_value.storbinary.assert_called_once_with("STOR output.zip", m.read())
4) 查看 ftp.storbinary
的文档,它需要文件对象本身,而不是对文件对象 read
方法的引用。应该是:
ftp.storbinary(" ".join(["STOR", file_name]), ftp_file)
传入(模拟的)文件对象将解决上面的第 3) 点。
关于python - 模拟内置函数在导入模块中打开,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37752485/
当我这样做时... import numpy as np ...我可以使用它但是... import pprint as pp ...不能,因为我需要这样做... from pprint import
我第一次尝试将 OpenCV 用于 Python 3。要安装,我只需在终端中输入“pip3 install opencv-python”。当我这样做时,我在 Finder(我在 Mac 上)中看到,在
如果有一个库我将使用至少两种方法,那么以下之间在性能或内存使用方面是否有任何差异? from X import method1, method2 和 import X 最佳答案 有区别,因为在 imp
我正在从 lodash 导入一些函数,我的同事告诉我,单独导入每个函数比将它们作为一个组导入更好。 当前方法: import {fn1, fn2, fn3} from 'lodash'; 首选方法:
之间有什么关系: import WSDL 中的元素 -和- import元素和在 XML Schema ...尤其是 location 之间的关系前者和 schemaLocation 的属性后者的属性
我在从 'theano.configdefaults' 导入 'local_bitwidth' 时遇到问题。并显示以下消息: ImportError
我注意到 React 可以这样导入: import * as React from 'react'; ...或者像这样: import React from 'react'; 第一个导入 react
对于当前的项目,我必须使用矩阵中提供的信息并对其进行数学计算,以及使用 ITK/VTK 函数来显示医疗信息/渲染。基本上我必须以(我猜)50/50 的方式同时使用 matlab 例程和 VTK/ITK
当我看到 pysqlite 的示例时,SQLite 库有两个用例。 from sqlite3 import dbapi2 as sqlite3 和 import sqlite3 为什么有两种方式支持s
我使用 Anaconda Python 发行版:Python 2.7 x64 和 Windows 7 SP1 x64 Ultimate。 当我import matplotlib.pyplot时,我得到
目录 【容器】镜像导出/导入 导出 导入 带标签 不带标签,后期修改 【仓库】镜像导出/导入
我正在寻找一种导入模块的方法,以便我可以从子文件夹 project/v0 和根文件夹 project 运行脚本。/p> 我在 python 3.6 中的文件结构(这就是没有初始化文件的原因) proj
我通常被告知以下是不好的做法。 from module import * 主要原因(或者有人告诉我)是,您可能会导入一些您不想要的东西,并且它可能会隐藏另一个模块中具有类似名称的函数或类。 但是,Py
我为 urllib (python3) 编写了一个小包装器。在if中导入模块是否正确且安全? if self.response_encoding == 'gzip': import gzip
我正在 pimcore 中创建一个新站点。有没有办法导出/导入 pimcore 站点的完整数据,以便我可以导出 xml/csv 格式的 pimcore 数据进行必要的更改,然后将其导入回来? 最佳答案
在 Node JS 中测试以下模块布局,看起来本地导出的定义总是在名称冲突的情况下替换外部导出的定义(参见 B.js 中的 f1)。 A.js export const f1 = 'A' B.js e
我在使用 VBA 代码时遇到了一些问题,该代码应该将 excel 数据导入我的 Access 数据库。当我运行代码时,我收到一个运行时错误“运行时错误 438 对象不支持此属性或方法”。来自我在其他论
我有一个名为 elements 的包,其中包含按钮、trifader、海报等内容。在 Button 类中,我正在执行 from elements import * 这执行正常,当我尝试 print(p
在我长期使用 python 的经验中,我遇到了一个非常奇怪的问题。 提前我想说我想知道为什么会发生这种情况 ,而不是如何更改我的代码或如何修复它,因为我也可以做到。 我正在使用 python2.7.3
我正在更新我的包。但是,我正在为依赖项/导入而苦苦挣扎。我使用了两个冲突的包 - ggplot2和 psych及其功能 alpha当然还有 alpha ggplot2 的对象不同于 alpha psy
我是一名优秀的程序员,十分优秀!