4

我有一个为我的用户添加订阅的功能。在这个函数的最后，一个指令会检查订阅并在必要时与赞助人一起做一些事情。对于我的测试，我需要跳过最后一条指令；所以我尝试使用mock.patch ，但尽管我做了所有尝试，代码总是被执行（我的测试失败了）

我的模块/classes.py

class User:

    def add_subscription(self, start_date, end_date):
        subscriptions = self.get('subscriptions', [])
        subscriptions.append({'start': start_date, 'end': end_date})
        self['subscriptions'] = subscriptions
        check_subscriptions(self)  # <-- skip this instruction for unitesting


def check_subscriptions(user):
    print("Not need to print for unitesting")
    # next doing more stuff...
    ...

测试/user_unitest.py

import mock
from mymodule.classes import User
from datetime import datetime, timedelta

...

def test_subscriptions(user_with_no_subscription_fixture):
    user = user_with_no_subscription_fixture
    start = datetime.now()
    end = start + timedelta(days=10)
    user.add_subscription(start, end)
    ...

如何使用@mock.patch（或其他模拟函数）不进入 User.check_subscriptions方法？

我尝试了什么（但没有奏效）

@mock.patch('mymodule.classes.check_subscriptions')
def test_subscriptions(user_with_no_subscription_fixture):
    user = user_with_no_subscription_fixture
    start = datetime.now()
    end = start + timedelta(days=10)
    user.add_subscription(start, end)

def test_subscriptions(user_with_no_subscription_fixture):
    with mock.patch('mymodule.classes.check_subscriptions'):
        user = user_with_no_subscription_fixture
        start = datetime.now()
        end = start + timedelta(days=10)
        user.add_subscription(start, end)

@mock.patch('mymodule.classes.check_subscriptions', MagicMock())
def test_subscriptions(user_with_no_subscription_fixture):
    user = user_with_no_subscription_fixture
    start = datetime.now()
    end = start + timedelta(days=10)
    user.add_subscription(start, end)

@mock.patch('mymodule.classes.check_subscriptions', side_effect=...)
def test_subscriptions(user_with_no_subscription_fixture):
    user = user_with_no_subscription_fixture
    start = datetime.now()
    end = start + timedelta(days=10)
    user.add_subscription(start, end)

感谢您一直以来的有用帮助

4

0 回答 0