首页 > 代码库 > 实战Django：官方实例Part5

实战Django：官方实例Part5

2024-08-06 16:00:17   218人阅读

俗话说，人非圣贤，孰能无过。在堆代码的过程中，即便是老攻城狮，也会写下一些错误的内容。俗话又说，过而能改，善莫大焉。要改，首先要知道哪里存在错误，这便是我们要对投票应用进行测试的原因。

 

21.撰写第一个测试

在我们这个项目中，还真有一个bug存在。这个bug位于Question.was_published_recently() 方法中。当Question提交的日期是正确的，那没问题，但若提交的日期是错误的——比如日期是几天之后，问题就来了。

你可以在管理页面中增加一个投票，把日期设置在几天之后，你会发现你刚增加的投票被程序认为是“最近发布”的。

我们可以编写一段测试程序来界定问题。

编辑polls/tests.py 文件，添加下面的内容：

polls/tests.py ：

import datetimefrom django.utils import timezonefrom django.test import TestCasefrom polls.models import Questionclass QuestionMethodTests(TestCase):    def test_was_published_recently_with_future_question(self):        """        如果question的发布日期是在将来，那么was_published_recently()应该        返回一个False值。        """        time = timezone.now() + datetime.timedelta(days=30)        future_question = Question(pub_date=time)        self.assertEqual(future_question.was_published_recently(), False)

我们来运行一下测试，在Dos命令提示符下（注意，检查一下是否位于项目文件夹mysite下，，就象我们在Part1中所做的那样），输入：

python manage.py test polls

你会看到象这样的运行结果：

Creating test database for alias ‘default‘...F======================================================================FAIL: test_was_published_recently_with_future_question (polls.tests.QuestionMethodTests)----------------------------------------------------------------------Traceback (most recent call last):  File "/path/to/mysite/polls/tests.py", line 16, in test_was_published_recently_with_future_question    self.assertEqual(future_question.was_published_recently(), False)AssertionError: True != False----------------------------------------------------------------------Ran 1 test in 0.001sFAILED (failures=1)Destroying test database for alias ‘default‘...

我们来看一下整个运行过程：

  • 执行python manage.py test polls后，程序会自动检索polls应用下面的tests.py文档。
  • 然后它会发现我们的测试类：QuestionMethodTests
  • 程序会根据测试类创建一个临时数据库；
  • 在test_was_published_recently_with_future_question中，程序会创建一个Question实例，它的发布日期在30天之后；
  • 最后它使用了assertEqual() 方法，它发现was_published_recently() 返回的值是True，而实际上我们希望它返回的是False。

所以我们看到最终的结果是FAILED，说明我们的程序存在问题。

 

22.修复Bug

既然找到了问题所在，我们来修复它。

编辑polls/models.py 文件，作如下改动：

polls/models.py ：

def was_published_recently(self):    now = timezone.now()    return now - datetime.timedelta(days=1) <= self.pub_date <= now

然后再运行一次测试，可看到如下结果：

Creating test database for alias ‘default‘....----------------------------------------------------------------------Ran 1 test in 0.001sOKDestroying test database for alias ‘default‘...

这回正常了。

23.更多综合性的测试

有时项目中不止一个bug,下面，我们再编写两个测试。

编辑polls/tests.py 文件，在QuestionMethodTests类下添加下面的内容：

polls/tests.py ：

def test_was_published_recently_with_old_question(self):    """    was_published_recently() should return False for questions whose    pub_date is older than 1 day    """    time = timezone.now() - datetime.timedelta(days=30)    old_question = Question(pub_date=time)    self.assertEqual(old_question.was_published_recently(), False)def test_was_published_recently_with_recent_question(self):    """    was_published_recently() should return True for questions whose    pub_date is within the last day    """    time = timezone.now() - datetime.timedelta(hours=1)    recent_question = Question(pub_date=time)    self.assertEqual(recent_question.was_published_recently(), True)

现在，我们一共有三个测试来确认Question.was_published_recently() 在过去、最近、将来三个时间点上创建问题时返回正确的值。

当然。投票应用还只是一个非常简单的例子，相应的bug也不会很多。但在以后我们开发项目的过程中，我们会碰到一些复杂的应用，这时，测试就变得更加重要了。

24.测试视图

前面我们只是对应用的内部业务逻辑进行测试，接下来，我们要模拟用户操作，来测试我们的视图。

在Part4中，我们的Index视图使用了Django通用视图中的ListView，它的内容是这样的：

polls/views.py ：

class IndexView(generic.ListView):    template_name = ‘polls/index.html‘    context_object_name = ‘latest_question_list‘    def get_queryset(self):        """Return the last five published questions."""        return Question.objects.order_by(‘-pub_date‘)[:5]

我们需要修正get_queryset这个方法，让它在提取数据时检查发布时间，与当前时间进行比对。我们编辑polls/views.py，在文件头部先加入：

polls/views.py：

from django.utils import timezone

然后修正get_queryset方法 ：

polls/views.py：

def get_queryset(self):    """    Return the last five published questions (not including those set to be    published in the future).    """    return Question.objects.filter(        pub_date__lte=timezone.now()    ).order_by(‘-pub_date‘)[:5]

Question.objects.filter(pub_date__lte=timezone.now())可确保返回的结果集中的Question对象的发布日期早于或等于当前的时间。

现在我们来测试一下这个新的视图。

编辑polls/tests.py，先在文件头部加入：

polls/tests.py： 

from django.core.urlresolvers import reverse

然后再加上以下内容：

polls/tests.py： 
def create_question(question_text, days):    """    Creates a question with the given `question_text` published the given    number of `days` offset to now (negative for questions published    in the past, positive for questions that have yet to be published).    """    time = timezone.now() + datetime.timedelta(days=days)    return Question.objects.create(question_text=question_text,                                   pub_date=time)class QuestionViewTests(TestCase):    def test_index_view_with_no_questions(self):        """        If no questions exist, an appropriate message should be displayed.        """        response = self.client.get(reverse(‘polls:index‘))        self.assertEqual(response.status_code, 200)        self.assertContains(response, "No polls are available.")        self.assertQuerysetEqual(response.context[‘latest_question_list‘], [])    def test_index_view_with_a_past_question(self):        """        Questions with a pub_date in the past should be displayed on the        index page        """        create_question(question_text="Past question.", days=-30)        response = self.client.get(reverse(‘polls:index‘))        self.assertQuerysetEqual(            response.context[‘latest_question_list‘],            [‘<Question: Past question.>‘]        )    def test_index_view_with_a_future_question(self):        """        Questions with a pub_date in the future should not be displayed on        the index page.        """        create_question(question_text="Future question.", days=30)        response = self.client.get(reverse(‘polls:index‘))        self.assertContains(response, "No polls are available.",                            status_code=200)        self.assertQuerysetEqual(response.context[‘latest_question_list‘], [])    def test_index_view_with_future_question_and_past_question(self):        """        Even if both past and future questions exist, only past questions        should be displayed.        """        create_question(question_text="Past question.", days=-30)        create_question(question_text="Future question.", days=30)        response = self.client.get(reverse(‘polls:index‘))        self.assertQuerysetEqual(            response.context[‘latest_question_list‘],            [‘<Question: Past question.>‘]        )    def test_index_view_with_two_past_questions(self):        """        The questions index page may display multiple questions.        """        create_question(question_text="Past question 1.", days=-30)        create_question(question_text="Past question 2.", days=-5)        response = self.client.get(reverse(‘polls:index‘))        self.assertQuerysetEqual(            response.context[‘latest_question_list‘],            [‘<Question: Past question 2.>‘, ‘<Question: Past question 1.>‘]        )

我们来看一下这部分代码：

  • 首先我们使用了一个叫create_question的函数，它是用来快速创建问题的，因为随后的测试中都会用到，所以有这个函数，可减少一些重复劳动。
  • test_index_view_with_no_questions不创建任何问题，只是检查当没有任何投票问题的时候，首页是否能返回“No polls are available.”这个信息，同时检查latest_question_list是不是空的。
  • 在test_index_view_with_a_past_question中，我们创建了一个问题，把它的发布时间设在了30天前，然后检查它是不是出现在首页的列表中；
  • 在test_index_view_with_a_future_question中，我们创建了一个问题，把它的发布时间设在了30天后，这里的每一个测试方法在执行的时候，数据库都会重置。所以我们在上一步测试中创建的那个问题不再存在，这样，首页的列表就应该是空的；

即使我们在首页视图中不再显示那些发布在将来时段的问题，但还会有用户通过合适的链接来访问到这些内容。这就意味着，我们要调整内容页的视图。

编辑polls/views.py，在DetailView中加入get_queryset方法：

polls/views.py：

class DetailView(generic.DetailView):    ...    def get_queryset(self):        """        Excludes any questions that aren‘t published yet.        """        return Question.objects.filter(pub_date__lte=timezone.now())

我们同样要编写一些测试来检查这个视图是否起作用了。

编辑polls/tests.py，加入下列内容：

polls/tests.py： 

class QuestionIndexDetailTests(TestCase):    def test_detail_view_with_a_future_question(self):        """        The detail view of a question with a pub_date in the future should        return a 404 not found.        """        future_question = create_question(question_text=‘Future question.‘,                                          days=5)        response = self.client.get(reverse(‘polls:detail‘,                                   args=(future_question.id,)))        self.assertEqual(response.status_code, 404)    def test_detail_view_with_a_past_question(self):        """        The detail view of a question with a pub_date in the past should        display the question‘s text.        """        past_question = create_question(question_text=‘Past Question.‘,                                        days=-5)        response = self.client.get(reverse(‘polls:detail‘,                                   args=(past_question.id,)))        self.assertContains(response, past_question.question_text,                            status_code=200)

我们来简单分析一下这段测试：

  • 在test_detail_view_with_a_future_question中，我们创建了一个问题，把它的发布时间设置在5天后，然后模拟用户去访问，如果我们新的DetailView起作用的话，这个链接应该是空的，换句话说，访问这个链接时，用户会得到一个404的状态码。
  • 在test_detail_view_with_a_past_question中，我们创建了一个问题，把它的发布时间设置在5天前，同样模拟用户去访问，这种情况下，用户会得到的状态码应该是200，也就是说，链接是有效的。

我们还可以就更多的问题进行测试，同时根据测试优化我们的应用。

举个例子，用户在使用这个应用的过程中，发布投票时没有带任何的投票项，此时，我们的视图需要具备相应的检测功能，来防止这类事情的发生。我们可以编写一个测试，创建一个问题，让它不再任何投票项，通过这样的测试来界定问题，并根据测试结果对视图进行调整。

在测试中，我们奉行一个理念：测试越多越好。测试越多，说明我们的应用越可靠。或许有一天，我们的测试代码的数量甚至超过了正式代码，不必在意这些。测试只会让我们的代码愈来愈成熟。

 

【未完待续】

本文版权归舍得学苑所有，欢迎转载，转载请注明作者和出处。谢谢！
作者：舍得
首发：舍得学苑@博客园

实战Django：官方实例Part5


联系
我们