[英]How to unit test this code with pymox?
所以我已经安装了pymox,我想测试一下这个方法:
class HttpStorage():
def download(self, input, output):
try:
file_to_download = urllib2.urlopen(input)
except URLError:
raise IOError("Opening input URL failed")
f = open(output, "wb")
f.write(file_to_download.read())
f.close()
return True
我正在阅读pymox文档,但无法弄清楚该怎么做。 您能帮我一些示例代码吗?
import unittest
import mox
class HttpStorageTest(mox.MoxTestBase):
def setUp(self):
self.httpstorage = HttpStorage()
def test_download(self):
self.mox.StubOutWithMock(urllib2, "urlopen")
test_file = open("testfile")
urllib2.urlopen(mox.IgnoreArg()).AndReturn(test_file)
self.mox.ReplayAll()
feedback = self.httpstorage.download(test_input, test_output)
self.assertEqual(feedback, True)
您可以尝试使用这种格式来测试您的下载功能,因为urllib2.openurl()已被模拟进行单元测试。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.