测试Django视图:我们怎么知道是否实际引发了catched异常?

时间:2018-01-22 13:36:58

标签: python django unit-testing testing exception-handling

我想测试我的视图中的异常是否在预期时引发。我面临的“问题”是我在我的视图中使用tryexcept语句,所以我实际上捕获这些异常(为了做一些事情,当我他们被提出了。)

我的问题如下:如果在我的exception声明中找到了预期的except,我该如何测试?

例如:

myviews.py

class SomeView(LoginRequiredMixin, CreateView):

    model = models.MyModel
    template_name = "my_template.html"
    form_class = forms.MyForm       
    def form_valid(self, form):

        try:
             # normal stuff here
             return super(SomeView, self).form_valid(form)

        except Exception1 as e:
             # some other stuff here
             print("Exception1 occured: ", e)
             messages.error(self.request, "Message1")
             return self.render_to_response(self.get_context_data(form=form))

        except Exception2 as e:
             # some other stuff here
             print("Exception2 occured: ", e)
             messages.error(self.request, "Message2")
             return self.render_to_response(self.get_context_data(form=form))

        except Exception as e:
             # some other stuff here
             print("Something unexpected occured! ", e)
             messages.error(self.request, "Message_unexpected")
             return self.render_to_response(self.get_context_data(form=form))

mytests.py

def test_SomeView(self):

    # test: everything must works just fine
    form_data = some_valid_data
    response = self.client.post("some_url", form_data)
    self.assertEqual(response.status_code, 200)

    # test: Exception1 should be raised in SomeView
    form_data = some_NOT_valid_data # on purpose
    response = self.client.post(reverse("some_url"), form_data)
    for message in list(response.context['messages']): print(message) 
    # UPDATE1 - no message is print, 
    # even if they are displayed as expected on my real web page 
    # when doing willingly bad stuff

非常感谢您的帮助!

UPDATE1

最初,我想这样做:

form_data = some_NOT_valid_data
self.assertRaises(
        Exception1,
        lambda: self.client.post("some_url"), form_data)
    )

然后我改变了关于jonrsharpe评论的想法(测试行为比实现更多)。因此,当Exceptions被捕获时,我试图获取生成的错误消息,因此我可以检查它们是否符合我的预期。但是我显然无法获得任何context message,即使它们在我的模板中按预期显示!然而context并非空洞。

UPDATE2 - 回答我自己的问题

我终于找到了如何管理它。我试图添加follow=True(如response = self.client.post("some_url", form_data, follow=True)中所示,但这不正确,因为它不是重定向,但是使用不同的上下文再次呈现相同的视图。所以,我做的是以下内容:

对于我视图中的每个except语句:

except Exception1 as e:
    # some other stuff here
    print("Exception1 occured: ", e)
    messages.error(self.request, "Message1")
    return self.render_to_response(context=self.get_context_data(form=form, messages=list(messages.get_messages(self.request))))

另外,在我看来,我覆盖了我的get_context_data方法:

def get_context_data(self, **kwargs):
    context = super(SomeView, self).get_context_data(**kwargs)
    if 'messages' in kwargs: context['messages'] = kwargs['messages']
    return context

然后,最后,在我的测试中,我能够获得例外消息:

response = self.client.post("some_url", form_data)
self.assertTrue("part_of_message" in str(response.context.get('messages')[0]))

它有效!

1 个答案:

答案 0 :(得分:0)

我终于成功了(参见我自己的更新2 ),并注明了@jonrsharpe。测试行为比完全测试我想要的更好(如果exception被引发,然后被捕获)。所以我努力测试错误messages是否正确生成,并且在纠正了我自己的一些错误后,我终于得到了一些有用的东西!或者没有工作,但正如预期的那样:你明白了。