Question

我有一个自定义的simple_tag，其定义如下：

@register.simple_tag
# usage: {% get_contact_preference_string user %}
def get_contact_preference_string(user):
    if user.contact_choice == 'C':
       return '{} prefers phone calls.'.format(user.first_name)
    # method continues

还有一个模板，该模板可以正确加载标签并使用标签。

但是，我在单元测试中努力将模拟用户传递给它。这是我尝试编写测试的方法：

def test_get_contact_preference_string_returns_correctly_formatted_content(self):
    test_customer = Customer.objects.create('tfirst', 'C')
    template_to_render = Template(
        '{% load contact_preference_helpers %}'
        '{% get_contact_preference_string test_customer %}'
    )

    rendered = template_to_render.render(test_customer)
    expected = 'tfirst prefers phone calls.'

    self.assertEqual(rendered, expected)

在击中AttributeError: 'NoneType' object has no attribute 'contact_choice'时会提高render(test_customer)，所以我知道我没有正确传递模拟对象。我也尝试过传递{'user': test_customer}无效。

我在做什么错了？

Answer 1

您需要传递一个Context实例来呈现模板。试试

from django.template import Context, Template

...

test_customer = Customer.objects.create('tfirst', 'C')
template_to_render = Template(
    '{% load contact_preference_helpers %}'
    '{% get_contact_preference_string test_customer %}'
)
ctx = Context({'test_customer': test_customer})
rendered = template_to_render.render(ctx)

如何将模拟对象传递给自定义simple_tag的单元测试？

1 个答案: