Question

我有一个基类，它定义了一个类属性和一些依赖它的子类，例如

class Base(object):
    assignment = dict(a=1, b=2, c=3)

我想使用不同的分配对此类进行单元测试，例如空字典，单项等。当然，这是非常简化的，不是重构我的课程或测试的问题

我提出的（pytest）测试，最终，这项工作是

from .base import Base

def test_empty(self):
    with mock.patch("base.Base.assignment") as a:
        a.__get__ = mock.Mock(return_value={})
        assert len(Base().assignment.values()) == 0

def test_single(self):
    with mock.patch("base.Base.assignment") as a:
        a.__get__ = mock.Mock(return_value={'a':1})
        assert len(Base().assignment.values()) == 1

这感觉相当复杂和hacky - 我甚至不完全理解为什么它有效（尽管我熟悉描述符）。 mock是否自动将类属性转换为描述符？

一种感觉更符合逻辑的解决方案不起作用：

def test_single(self):
    with mock.patch("base.Base") as a:
        a.assignment = mock.PropertyMock(return_value={'a':1})
        assert len(Base().assignment.values()) == 1

或只是

def test_single(self):
    with mock.patch("base.Base") as a:
        a.assignment = {'a':1}
        assert len(Base().assignment.values()) == 1

我尝试过的其他变种也不起作用（测试中的作业保持不变）。

模拟类属性的正确方法是什么？有没有比上面更好/更容易理解的方式？

Answer 1

base.Base.assignment只是替换为Mock对象。您通过添加__get__方法使其成为。

有点冗长，有点不必要;你可以直接设置base.Base.assignment：

def test_empty(self):
    Base.assignment = {}
    assert len(Base().assignment.values()) == 0

当然，使用测试并发时这不太安全。

要使用PropertyMock，我会使用：

with patch('base.Base.assignment', new_callable=PropertyMock) as a:
    a.return_value = {'a': 1}

甚至：

with patch('base.Base.assignment', new_callable=PropertyMock, 
           return_value={'a': 1}):

Answer 2

为了提高可读性，您可以使用@patch装饰器：

from mock import patch
from unittest import TestCase

from base import Base

class MyTest(TestCase):
    @patch('base.Base.assignment')
    def test_empty(self, mock_assignment):
        # The `mock_assignment` is a MagicMock instance,
        # you can do whatever you want to it.
        mock_assignment.__get__.return_value = {}

        self.assertEqual(len(Base().assignment.values()), 0)
        # ... and so on

您可以在http://www.voidspace.org.uk/python/mock/patch.html#mock.patch找到更多详情。

Answer 3

如果您的测试中已经导入了中的类（例如队列），而您想修补 MAX_RETRY attr - 您可以使用 {{3} } 或者只是更好 @patch.object

from mock import patch, PropertyMock, Mock
from somewhere import Queue

@patch.multiple(Queue, MAX_RETRY=1, some_class_method=Mock)
def test_something(self):
    do_something()


@patch.object(Queue, 'MAX_RETRY', return_value=1, new_callable=PropertyMock)
def test_something(self, _mocked):
    do_something()

Answer 4

也许我错过了一些东西，但如果不使用PropertyMock？

就不可能

with mock.patch.object(Base, 'assignment', {'bucket': 'head'}):
   # do stuff

Answer 5

以下是如何对Base类进行单元测试的示例：

模拟不同类型的多个类属性（即：dict和int）
使用@patch装饰器和pytest框架与python 2.7+或3+。

# -*- coding: utf-8 -*-
try: #python 3
    from unittest.mock import patch, PropertyMock
except ImportError as e: #python 2
    from mock import patch, PropertyMock 

from base import Base

@patch('base.Base.assign_dict', new_callable=PropertyMock, return_value=dict(a=1, b=2, c=3))
@patch('base.Base.assign_int',  new_callable=PropertyMock, return_value=9765)
def test_type(mock_dict, mock_int):
    """Test if mocked class attributes have correct types"""
    assert isinstance(Base().assign_dict, dict)
    assert isinstance(Base().assign_int , int)

在python单元测试中模拟类属性的更好方法

5 个答案: