Как исправлять внутренние функции модуля с помощью макета?

Под "внутренней функцией" подразумевается функция, вызываемая из того же модуля, в которой она определена.

Я использую библиотеку mock, в частности, декораторы patch, в моем модуле тесты. Это модульные тесты Django, но это должно применяться к любым тестам python.

У меня есть один модуль с несколькими функциями, многие из которых называют друг друга. Например (фиктивный код, игнорировать отсутствие decimal.Decimal):

TAX_LOCATION = 'StateName, United States'

def add_tax(price, user):
    tax = 0
    if TAX_LOCATION == 'StateName, UnitedStates':
        tax = price * .75
    return (tax, price+tax)

def build_cart(...):
    # build a cart object for `user`
    tax, price = add_tax(cart.total, cart.user)
    return cart

Это часть более глубокой вызывающей цепочки (func1 → func2 → build_cart → add_tax), все из которых находятся в одном модуле.

В моих модульных тестах я хотел бы отключить налоги, чтобы получить согласованные результаты. Как я вижу, мои два варианта: 1) исправить TAX_LOCATION (с пустой строкой, скажем), чтобы add_tax ничего не делал или 2) исправил add_tax, чтобы просто вернуть (0, цена).

Однако, когда я пытаюсь установить патч, патч работает внешне (я могу импортировать исправленную часть внутри теста и распечатать его, получить ожидаемые значения), но, кажется, не оказывает внутреннего эффекта (результаты я get from code ведут себя так, как если бы патч не применялся).

Мои тесты подобны этому (опять же, фиктивный код):

from mock import patch
from django.test import TestCase

class MyTests(TestCase):

    @patch('mymodule.TAX_LOCATION', '')
    def test_tax_location(self):
        import mymodule
        print mymodule.TAX_LOCATION # ''
        mymodule.func1()
        self.assertEqual(cart.total, original_price) # fails, tax applied

    @patch('mymodule.add_tax', lambda p, u: (0, p))
    def test_tax_location(self):
        import mymodule
        print mymodule.add_tax(50, None) # (0, 50)
        mymodule.func1()
        self.assertEqual(cart.total, original_price) # fails, tax applied

Кто-нибудь знает, может ли макет исправлять функции, используемые внутри, например, или мне не повезло?

Ответы

mock.patch('function_name')

mock.patch(__name__ + '.' + 'function_name')