3

这是我项目中的一个简单结构：

MyAPP---
        note---
               __init__.py
               views.py
               urls.py
               test.py
               models.py
        auth--
              ...
        template---
                   auth---
                          login.html
                          register.html
                   note---
                          noteshow.html
                   media---
                           css---
                                 ...
                           js---
                                 ...
        settings.py
        urls.py
        __init__.py
        manage.py

我想做一个单元测试，它可以测试noteshow页面是否正常工作。

编码：

from django.test import TestCase

class Note(TestCase):
    def test_noteshow(self):
        response = self.client.get('/note/')
        self.assertEqual(response.status_code, 200)
        self.assertTemplateUsed(response, '/note/noteshow.html')

问题是我的项目包含一个身份验证模块，当他们访问 noteshow.html 时，它将强制取消登录用户重定向到 login.html 页面。

因此，当我运行我的单元测试时，在 bash 中会引发 response.status_code 始终为 302 而不是 200 的故障。

好吧，虽然通过这个结果我可以检查 auth mod 运行良好，但它不像我想要的那样。

好的，问题是我怎样才能进行另一个单元测试来检查我的noteshow.template是否被使用？

谢谢大家。

django 版本：1.1.1

蟒蛇版本：2.6.4

将 Eclipse 用于 MAC 操作系统

4

1 回答 1

6

只需为每个测试登录一个用户。最好的方法是使用setUp()创建客户端、创建用户然后登录用户的tearDown()方法。也使用相反的方法（注销用户并删除用户）。

方法setUp()和tearDown()为测试集中的每个测试自动运行。

它看起来像这样：

class Note(TestCase):
    def setUp(self):
        self.client = Client()
        self.new_user = User.objects.create_user('blah', 'blah@blah.com', 'blah')
        self.new_user.save()
        self.client.login(username='blah', password='blah')

    def tearDown(self):
        self.client.logout()
        self.new_user.delete()

    def test_noteshow(self):
        response = self.client.get('/note/')
        self.assertEqual(response.status_code, 200)
        self.assertTemplateUsed(response, '/note/noteshow.html')

希望这可以帮助。

于 2010-05-10T04:25:49.583 回答