mock.patch和unittest.skip之间的冲突

时间:2015-08-20 11:46:33

标签: python python-3.x python-unittest

(Python 3.4.0)

我遇到了这个奇怪的错误,我花了一些时间来调试:

user.py

class User:

    def __init__(self, name):
        self.name = name


def new_user(name):
    user = User(name)

test.py

import unittest
from unittest.mock import Mock, patch

from user import new_user

@patch('user.User')
class TestUser(unittest.TestCase):

    @unittest.skip
    def test_new_user(self, mockUser):
        new_user('Frank')
        mockUser.assert_called_once_with('Frank')


unittest.main()

运行它会崩溃:

» python test.py 
E
======================================================================
ERROR: test_new_user (__main__.TestUser)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/usr/lib/python3.4/unittest/mock.py", line 1125, in patched
    return func(*args, **keywargs)
TypeError: decorator() takes 1 positional argument but 2 were given

----------------------------------------------------------------------
Ran 1 test in 0.001s

FAILED (errors=1)

删除跳过将使其正常运行。似乎patchskip堆叠不好。这是正确的,还是我做了一些愚蠢的事情?

1 个答案:

答案 0 :(得分:1)

unittest.skip需要一个自己的字符串参数,这是跳过测试的原因。

@unittest.skip("Not yet ready to test")
def test_new_user(self, mockUser):
    new_user('Frank')
    mockUser.assert_called_once_with('Frank')

您看到的交互来自skip装饰器,它使用方法本身作为reason参数(def skip(reason):),这会导致test_new_user绑定到一个-argument函数在装饰器中定义,而不是您在测试用例中定义的双参数函数。

请注意,如果您将呼叫留在skip并注释掉了patch,那么您的测试仍会通过,尽管test_new_user似乎没有收到mockUser参数。

unittest.skip本身在技术上不是装饰者;它是一个返回装饰器的函数,然后将其应用于test_new_user。使用常规函数调用语法,您的代码

def test_new_user(self, mockUser):
    ...
test_new_user = unittest.skip(test_new_user)

当你需要的是

test_new_user = unittest.skip("my reason")(test_new_user)

您的test_new_user绑定到装饰器本身,而不是装饰方法。