Question

最近几天我一直在尝试编写Django TestCase，但未能为多个模型编写测试用例

这是我的models.py

from django.db import models
from django.contrib.auth.models import User

class Author(models.Model):
    name = models.TextField(max_length=50)

class Category(models.Model):
    name = models.CharField(max_length=100)

class Article(models.Model):
    author = models.ForeignKey(Author, on_delete=models.CASCADE)
    title = models.CharField(max_length=200)
    body = models.TextField()
    category = models.ForeignKey(Category, on_delete=models.CASCADE)

我试图这样写TestCase。

这是我的tests.py

from django.test import TestCase
from blog.models import Article, Author, Category


class TestContactModel(TestCase):
    def setUp(self):
        self.article = Article(author='jhon', title='how to test', body='this is body', category='djangooo')
        self.article.save()

    def test_contact_creation(self):
        self.assertEqual(article.objects.count(), 1)

    def test_contact_representation(self):
        self.assertEqual(self.article.title, str(self.article))

谁能告诉我该如何进行这项测试？感谢您的时间和关怀

Answer 1

author是ForeignKey，因此您应该首先创建一个Author，然后将引用传递给该Author对象。 category外键也是如此：

class TestContactModel(TestCase):

    def setUp(self):
        self.author = author = Author.objects.create(name='Douglas Adams')
        self.category = category = Category.objects.create(name='sci-fi')
        self.article = Article.objects.create(
            author=author,
            title="The Hitchhiker's Guide to the Galaxy",
            body='42',
            category=category
        )

用于多种模型的Django测试用例编写

1 个答案: