控制urllib.urlopen返回的最佳方法是什么?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (75)

我需要测试一个函数,它需要使用urllib.urlopen在外部服务器上查询页面(它还使用urllib.urlencode)。服务器可能关闭,页面可能会改变;我不能依赖它进行测试。

控制urllib.urlopen返回的最佳方法是什么?

提问于
用户回答回答于

一个简单的方法是让您的测试覆盖urllib的urlopen()函数。例如,如果您的模块

import urllib

def some_function_that_uses_urllib():
    ...
    urllib.urlopen()
    ...

您可以这样定义您的测试:

import mymodule

def dummy_urlopen(url):
    ...

mymodule.urllib.urlopen = dummy_urlopen

然后,当您的测试调用mymoduledummy_urlopen()将被称为,而不是真正的urlopen()

用户回答回答于
from mock import patch

[...]

@patch('urllib.urlopen')
def test_foo(self, urlopen_mock):
    urlopen_mock.return_value = MyUrlOpenMock()

所属标签

可能回答问题的人

  • 天使的炫翼

    17 粉丝531 提问36 回答
  • 富有想象力的人

    3 粉丝0 提问30 回答
  • 学生

    3 粉丝476 提问28 回答
  • o o

    4 粉丝494 提问27 回答

扫码关注云+社区

领取腾讯云代金券