时间戳

首页 > 解决方案 > 为多个模型编写 Django 测试用例

django - 为多个模型编写 Django 测试用例

问题描述

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

这是我的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))

谁能告诉我如何制作这个测试？非常感谢您的时间和关怀

标签： djangodjango-testingdjango-tests

解决方案

author是 a ，因此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
        )

推荐阅读