使用假用户的 Django 测试模型

Question

我有一个想要测试的 django 模型。该模型包括“用户”字段。有没有办法创建临时用户以进行测试？

测试.py

from django.test import TestCase
from .models import Task
from django.utils import timezone
from .forms import Taskform

# models test
class TaskTest(TestCase):

    def create_task(self, title="Title", description="A description"):
        return Task.objects.create(title=title, pub_date=timezone.datetime.now(), completed=False, description=description, user='user')

    def test_task_creation(self):
        w = self.create_task()
        self.assertTrue(isinstance(t, Task))
        self.assertEqual(t.__unicode__(), t.title)

Answer 1

嗯，User实际上只是另一个Django模型(当然它在Django中有一些额外的绑定(bind)，这使得它成为一个“流行”和“特殊”模型，但是它具有相同的接口(interface)，您可以像其他模型一样构造实例)。因此，您可以导入该模型，并创建一个用户。

例如:

from django.contrib.auth.models import <b>User</b>

my_user = User.objects.create(username='Testuser')

通常在测试中，人们会使用工厂(例如使用 Factory Boy )来轻松使用某些数据构造对象。例如:

import factory
from factory.django import <b>DjangoModelFactory</b>

class <b>UserFactory</b>(DjangoModelFactory):

    username = factory.Sequence('testuser{}'.format)
    email = factory.Sequence('testuser{}@company.com'.format)

    class Meta:
        model = User

然后您可以使用以下命令创建用户:

my_user2 = UserFactory()
my_user3 = UserFactory(username='alice')
my_user4 = UserFactory(username='bob', email='bob@work.com')