Django单元测试不会更新BooleanField

时间:2018-07-13 17:19:15

标签: django django-forms django-views django-testing

我遇到一种情况,我想为文章发布功能,而我已经拥有了。问题是我无法使单元测试正常工作,试图将True张贴到已发布的字段中。

这是我的测试结果。

class ArticleDetailViewTest(TestCase):

    def setUp(self):
        email = 'testuser@gmail.com'
        first_name = 'test'
        last_name = 'user'
        password = 'test15146'
        user = User.objects.create_user(email, first_name, last_name, password)
        self.client = Client()
        self.client.login(username='testuser@gmail.com', password='test15146')

    def test_can_publish_article_from_POST(self):
        other_article_two = Article.objects.create(name='Test Name One', author=User.objects.get(email='testuser@gmail.com'))
        correct_article_two = Article.objects.create(name='Test Name Two', author=User.objects.get(email='testuser@gmail.com'))

        response = self.client.post(reverse('publish_article', kwargs={'pk' : correct_article_two.pk}))

        self.assertEqual(response.status_code, 302)
        self.assertRedirects(response, f'/articles/{correct_article_two.pk}/')
        self.assertEqual(correct_article_two.published, True)

这就是我的发布文章视图。

def publish_article(request, pk):
    article = get_object_or_404(models.Article, pk=pk)
    if request.method == "POST":
        article.published = True
        article.save(update_fields=['published'])
        return redirect('article_detail_view', pk=article.pk)
    else:
        raise Http404

这是我的网址。

from django.urls import path

from . import views

urlpatterns = [ 
    path('', views.HomePageView.as_view(), name='home'),
    path('new/', views.ArticleCreateView.as_view(), name='article_new'),
    path('<int:pk>/', views.ArticleDetailView.as_view(), name='article_detail_view'),
    path('<int:pk>/publish/', views.publish_article, name='publish_article'),
    path('<int:pk>/unpublish/', views.unpublish_article, name='unpublish_article'),]

这是我的模特。

from django.conf import settings
from django.db import models
from django.urls import reverse
from django import forms

class Article(models.Model):
    name = models.CharField(max_length=255, blank=False)
    author = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.CASCADE, 
    )
    published = models.BooleanField(default=False)

    def __str__(self):
        return self.name

    def get_absolute_url(self):
        return reverse('article_detail_view', args=[str(self.id)])

class ArticleForm(forms.ModelForm):
    class Meta:
        model = Article
        fields = ['name',]

最后,这是模板。

{% extends 'base.html' %}

{% block title %}{{ article.name }}{% endblock %}

{% block content %}

{% if article.author_id == user.pk %}
<div class="row">
    <div class="col-6 col-centered">
        <h2 class="text-center">{{ article.name }}</h2>
        <p class="text-center">by {{ article.author.first_name }} {{ article.author.last_name }}</p>
        {% if article.published %}
            <small class="form-text text-muted mb-1">This article is public.</small>
            <form method="post" action="/articles/{{ article.pk }}/unpublish/">
                {% csrf_token %}
                <input type="submit" name="unpublish" value="Unpublish Article" id="id_unpublish_button" class="btn btn-primary btn-md"/>
            </form>
        {% else %}
            <small class="form-text text-muted mb-1">This article is private.</small>
            <form method="post" action="/articles/{{ article.pk }}/publish/">
                {% csrf_token %}
                <input type="submit" name="publish" value="Publish Article" id="id_publish_button" class="btn btn-primary btn-md"/>
            </form>
        {% endif %}
    </div>
</div>
{% elif article.published %}
<div class="row">
    <div class="col-6 col-centered">
        <h2 class="text-center">{{ article.name }}</h2>
        <p class="text-center">by {{ article.author.first_name }} {{ article.author.last_name }}</p>
    </div>
</div>
{% else %}
<div class="row">
    <div class="col-6 col-centered">
        <p class="text-center">This article is private.</p>
    </div>
</div>
{% endif %}

{% endblock %}

这是我从测试中得到的错误消息。问题似乎是我可以使用self.client.post将其发布到URL。任何帮助将不胜感激。

失败:test_can_publish_article_from_POST(articles.tests.ArticleDetailViewTest)

回溯(最近通话最近):   在test_can_publish_article_from_POST中的文件“ /Users/christopher.chough/article_directory/articles/tests.py”,第126行     self.assertEqual(correct_article_two.published,True) AssertionError:False!= True


在2.340秒内进行了17次测试

失败(失败= 1)

1 个答案:

答案 0 :(得分:0)

您的测试方法中的对象未更新。您可以使用refresh_from_db方法在更改后对其进行更新:

def test_can_publish_article_from_POST(self):
    other_article_two = Article.objects.create(name='Test Name One', author=User.objects.get(email='testuser@gmail.com'))
    correct_article_two = Article.objects.create(name='Test Name Two', author=User.objects.get(email='testuser@gmail.com'))

    response = self.client.post(reverse('publish_article', kwargs={'pk' : correct_article_two.pk}))

    correct_article_two.refresh_from_db() # Add this line

    self.assertEqual(response.status_code, 302)
    self.assertRedirects(response, f'/articles/{correct_article_two.pk}/')
    self.assertEqual(correct_article_two.published, True)