在Django中测试extra_context

Question

我正在尝试测试用户提供的额外上下文是否在视图中被正确处理。

这是我的测试方法：

# tests.py (of course it's a part of TestCase class)
def test_should_use_definied_extra_context(self):
    response = self.client.get(reverse('contact_question_create'), {
        'extra_context': {'foo': 'bar', 'callable': lambda: 'called'}
    })
    eq_(response.context['foo'], 'bar')
    eq_(response.context['callable'], 'called')

这是我的视图函数：

# views.py (my view function)
def contact_question_create(request, success_url=None, form_class=None,
                        template_name="contact/contact_form.html",
                        extra_context=None, **kwargs):
if form_class is None:
    form_class = forms.DefaultContactForm

if request.method == 'POST':
    form = form_class(data=request.POST, files=request.FILES)
    if form.is_valid():
        signals.question_posted.send(sender='contact_question_create_view',
                                     form_data=form.cleaned_data)
        if success_url is None:
            pass # TODO: implement this case
        else:
            return redirect(sucess_url)
else:
    form = form_class()

if extra_context is None:
    extra_context = {}
context = RequestContext(request)
for key, value in extra_context.items():
    context[key] = callable(value) and value() or value

return render_to_response(template_name,
                          {'form':form},
                          context_instance=context)

在运行测试时，我遇到了以下的 KeyError 异常：

======================================================================
ERROR: test_should_use_definied_extra_context (mpozyczka.apps.contact.tests.ContactViewShouldUseRequestedValues)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Library/Frameworks/Python.framework/Versions/2.6/lib/python2.6/unittest.py",
      line 279, in run testMethod()
  File "/Users/bx2/Dropbox/Projekty/mpozyczka/src/mpozyczka/apps/contact/tests.py", 
      line 56, in test_should_use_definied_extra_context 
      eq_(response.context['foo'], 'bar')
  File "/Users/bx2/Dropbox/Projekty/mpozyczka/parts/django/django/test/utils.py", 
      line 35, in __getitem__ raise KeyError(key)
  KeyError: 'foo'

我觉得我可能太累了，看不出问题所在——有没有什么线索可以告诉我我漏掉了什么？

Answer 1

你需要使用以下的解决方案：

def contact_question_create(request, success_url=None, form_class=None,
                        template_name="contact/contact_form.html",
                        extra_context=None, **kwargs):

    # your view code here

    context = {'defult':'foo'}

    if extra_context:
        context.update(extra_context)

    return render_to_response(template_name,
                              context,
                              context_instance=RequestContext(request, context))

在测试的时候，你需要在 url.py 文件或者调用你视图的视图中传递 extra_context。

def my_view(reuest):
    return contact_question_create(extra_context={'foo':'bar'})