Использование mock.patch для отключения поведения функции - PullRequest
Новая
       28

Использование mock.patch для отключения поведения функции

0 голосов
/ 03 апреля 2020

У меня есть функция добавления подписки для моих пользователей. В конце этой функции, инструкция проверяет подписки и делает некоторые вещи с патроном, если это необходимо. Для моего тестирования мне нужно пропустить эту последнюю инструкцию; поэтому я попытался использовать mock.patch, но, несмотря на все мои попытки, код всегда выполняется (и мои тесты не пройдены)

mymodule / 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...
    ...

tests / 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)

спасибо за ваше всегда полезная помощь

...