如何用pytest fixture+autospec模拟带类属性的子类继承方法并维护API契约

问题1：正确关联Comment与User的Mock实例 #

autospec会严格遵循原类的结构，直接给mock_comment.user赋值普通Mock会触发属性不匹配问题。正确做法是：

• 先创建带autospec的User实例mock
• 通过configure_mock方法给Comment的mock设置user属性，避免直接赋值触发autospec检查失败

示例代码：

@pytest.fixture
def mocked_user():
    return mock.Mock(spec=User, autospec=True)

@pytest.fixture
def mocked_comment(mocked_user):
    comment_mock = mock.Mock(spec=Comment, autospec=True)
    # 用configure_mock设置符合spec的user属性
    comment_mock.configure_mock(user=mocked_user)
    return comment_mock

之后即可正常访问mock_top_comment.user.account_id，只需提前给mocked_user.account_id赋值测试用的ID即可。

问题2：解决工厂Fixture重复调用的InvalidSpecError #

该错误源于工厂fixture未每次生成全新的mock实例，复用了被autospec锁定的同一对象。解决方法是将fixture改为工厂函数模式，每次调用都生成独立的mock实例：

示例代码：

@pytest.fixture
def mocked_comment_factory():
    def _create_comment(user_mock=None):
        # 每次调用都创建新的Comment mock
        comment_mock = mock.Mock(spec=Comment, autospec=True)
        if user_mock:
            comment_mock.configure_mock(user=user_mock)
        return comment_mock
    return _create_comment

@pytest.fixture
def mocked_user_factory():
    def _create_user(account_id):
        user_mock = mock.Mock(spec=User, autospec=True)
        user_mock.account_id = account_id
        return user_mock
    return _create_user

在测试用例中可多次调用工厂函数，生成无冲突的独立mock：

def test_multiple_comments(mocked_comment_factory, mocked_user_factory):
    user1 = mocked_user_factory("user-123")
    comment1 = mocked_comment_factory(user1)
    user2 = mocked_user_factory("user-456")
    comment2 = mocked_comment_factory(user2)
    
    processor = ClassToTest()
    result1 = processor.get_author_id(comment1)
    result2 = processor.get_author_id(comment2)
    
    assert result1 == "user-123"
    assert result2 == "user-456"

完整验证示例（含契约校验） #

使用autospec=True可确保API契约变更时自动触发测试失败，比如User类移除account_id属性时，测试会直接报错：

import pytest
from unittest import mock
from atlassian import Comment, User
from your_module import ClassToTest

@pytest.fixture
def mocked_user_factory():
    def _create_user(account_id):
        user_mock = mock.Mock(spec=User, autospec=True)
        user_mock.account_id = account_id
        return user_mock
    return _create_user

@pytest.fixture
def mocked_comment_factory():
    def _create_comment(user_mock=None):
        comment_mock = mock.Mock(spec=Comment, autospec=True)
        if user_mock:
            comment_mock.configure_mock(user=user_mock)
        return comment_mock
    return _create_comment

def test_top_comment_author(mocked_comment_factory, mocked_user_factory):
    test_id = "TEST-ACCOUNT-001"
    test_user = mocked_user_factory(test_id)
    test_comment = mocked_comment_factory(test_user)
    
    processor = ClassToTest()
    author_id = processor.extract_top_comment_author(test_comment)
    
    assert author_id == test_id

关键注意事项 #

• 禁止直接给autospec生成的mock赋值属性，必须用configure_mock或创建时的关键字参数设置。
• 工厂fixture必须每次返回新的mock实例，确保测试用例间、同用例多次调用间无冲突。
• autospec=True会严格校验原类的方法签名与属性，API契约变更时会直接触发测试失败，达到契约验证目的。

内容的提问来源于stack exchange，提问作者Mark Arce