Question

我的项目使用Python urllib2.urlopen对外部API进行各种调用。我使用NoseTests进行单元测试，使用MiniMock模拟对urllib2.urlopen的调用。

模拟代码：

from hashlib import md5
from os.path import dirname, join
from urllib2 import Request, urlopen

from minimock import mock, restore

def urlopen_stub(url, data=None, timeout=30):
    """
    Mock urllib2.urlopen and return a local file handle or create file if
    not existent and then return it.
    """

    if isinstance(url, Request):
        key = md5(url.get_full_url()).hexdigest()
    else:
        key = md5(url).hexdigest()
    data_file = join(dirname(__file__), 'cache', '%s.xml' % key)
    try:
        f = open(data_file)
    except IOError:
        restore() # restore normal function
        data = urlopen(url, data).read()
        mock('urlopen', returns_func=urlopen_stub, tracker=None) # re-mock it.
        with open(data_file, 'w') as f:
            f.write(data)
        f = open(data_file, 'r')
    return f

mock('urlopen', returns_func=urlopen_stub, tracker=None)

我正在运行我的测试：

from os.path import isfile, join
from shutil import copytree, rmtree

from nose.tools import assert_raises, assert_true

import urlopenmock

class TestMain(object):
    working = 'testing/working'

    def setUp(self):
        files = 'testing/files'
        copytree(files, self.working)

    def tearDown(self):
        rmtree(self.working)

    def test_foo(self):
        func_a_calling_urlopen()
        assert_true(isfile(join(self.working, 'file_b.txt')))

    def test_bar(self):
        func_b_calling_urlopen()
        assert_true(isfile(join(self.working, 'file_b.txt')))

    def test_bar_exception(self):
        assert_raises(AnException, func_c_calling_urlopen)

最初，我在一个单独的模块中检查了异常，该模块导入了一个不同的模拟文件，该文件在调用urlopen时返回了一个损坏的XML文件。但是，导入该模拟类会覆盖上面显示的模拟类，因为每次都使用了损坏的XML，所以打破了所有测试。

我认为这是因为异常测试模块是在其他模块之后加载的，因此它的导入被称为last，而返回破碎的XML的模拟函数覆盖了原始的模拟函数。

我希望能够告诉模拟代码在运行test_bar_exception时使用损坏的XML文件，以便引发异常。我该怎么做呢？

Answer 1

在我看来，您需要在需要使用它的每个测试上设置和拆除模拟urlopen，并使用不同的mock urlopen为处理该错误条件的那些测试返回一个损坏的xml文件。类似的东西：

class TestMain(object):
    # ...

    def test_foo(self):
        mock('urlopen', returns_func=urlopen_stub, tracker=None)
        func_a_calling_urlopen()
        assert_true(isfile(join(self.working, 'file_b.txt')))
        restore()

    # ...

    def test_bar_exception(self):
        mock('urlopen', 
                returns_func=urlopen_stub_which_returns_broken_xml_file, 
                tracker=None)
        assert_raises(AnException, func_c_calling_urlopen)
        restore()

但是，如果测试引发异常并且未达到restore()调用，则会出现上述问题。你可以用try: ... finally:包裹，但每次测试都需要很多繁文缛节。

你可能想看看Michael Foord的mock库。

Pycon演示：http://blip.tv/file/4881513

查看patch，它为您提供装饰器和上下文管理器选项，用于替换和恢复您在urlopen上的调用。它很漂亮！

Answer 2

假设你的请求输入是'url'并且输出是'aresponse'，输入'burl'输出是'bresponse'，所以使用

@mock.patch('requests.get', mock.Mock(side_effect = lambda k:{'aurl': 'a response', 'burl' : 'b response'}.get(k, 'unhandled request %s'%k)))

如何在Python中为mock函数提供条件参数？

2 个答案: