如何像 urllib 这样的模拟/存根 python 模块

问题描述

我需要测试一个需要使用 urllib.urlopen(它也使用 urllib.urlencode)在外部服务器上查询页面的函数.服务器可能已关闭，页面可能会更改；我不能依赖它进行测试.

I need to test a function that needs to query a page on an external server using urllib.urlopen (it also uses urllib.urlencode). The server could be down, the page could change; I can't rely on it for a test.

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

What is the best way to control what urllib.urlopen returns?

推荐答案

另一种简单的方法是让您的测试覆盖 urllib 的 urlopen() 函数.例如，如果您的模块有

Another simple approach is to have your test override urllib's urlopen() function. For example, if your module has

import urllib

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

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

You could define your test like this:

import mymodule

def dummy_urlopen(url):
    ...

mymodule.urllib.urlopen = dummy_urlopen

然后，当您的测试调用 mymodule 中的函数时，将调用 dummy_urlopen() 而不是真正的 urlopen().像 Python 这样的动态语言可以非常容易地将方法和类存根以进行测试.

Then, when your tests invoke functions in mymodule, dummy_urlopen() will be called instead of the real urlopen(). Dynamic languages like Python make it super easy to stub out methods and classes for testing.

请参阅我在 http://softwarecorner.wordpress.com/ 上的博文，了解有关剔除的更多信息测试依赖项.

See my blog posts at http://softwarecorner.wordpress.com/ for more information about stubbing out dependencies for tests.

这篇关于如何像 urllib 这样的模拟/存根 python 模块的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！