2010-09-28 16 views
2

J'essaie de tester si le paramètre extra_context fourni par l'utilisateur a été correctement traité dans la vue.Test de extra_context dans django

Voici mon approche de test:

# 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') 

Et voici ma fonction de vue:

# 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) 

Lors du test, je suis en train suivante exception 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' 

Je pense que je suis trop fatigué et je ne vois pas le problème - des indices sur ce qui me manque?

Répondre

3

Vous devez utiliser la solution suivante:

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

Lors du test, il vous faut passer extra_context soit en url.py ou en vue qui appelle votre point de vue.

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