Ошибка в тесте Django с использованием Factory Boy

Я использую Factory Boy в своем проекте Django. Не могли бы вы объяснить мне мою ошибку: почему у меня возникает ошибка при запуске 'tests.py' - 'ValueError: "" должно иметь значение для поля "post", прежде чем можно будет использовать это отношение "многие ко многим" .'

Вот мой код:

import factory

from . import models

# factories


class TagFactory(factory.Factory):
    class Meta:
        model = models.Tag

    name = factory.Sequence(lambda n: 'tag-%s' % n)
    slug = factory.Sequence(lambda n: 'slug-%s' % n)


class CategoryFactory(factory.Factory):
    class Meta:
        model = models.Category

    name = factory.Sequence(lambda n: 'cat-%s' % n)
    slug = factory.Sequence(lambda n: 'cat-slug-%s' % n)


class PostFactory(factory.Factory):
    class Meta:
        model = models.Post

    title = factory.Sequence(lambda n: 'postik-%s' % n)
    text = 'some text'
    slug = factory.Sequence(lambda n: 'post_slug_%s' % n)
    category = factory.SubFactory(CategoryFactory)

    @factory.post_generation
    def tags(self, create, extracted, **kwargs):
        if not create:
            # Simple build, do nothing.
            return
        if extracted:
            for tag in extracted:
                self.tags.add(tag)

тесты.py

from django.test import TestCase

from . import factories
from .models import Post, Category, Tag
from django.core.urlresolvers import reverse


class PostTests(TestCase):
    """
    display_post view test.
    """
    def setUp(self):
        self.tag = factories.TagFactory()
        self.category = factories.CategoryFactory()
        self.poster = factories.PostFactory.create(tags=('tag-1'))

 ...

модели.py

from django.db import models
from django.core.urlresolvers import reverse
from django.utils import timezone
from django.utils.translation import ugettext_lazy as _

from ckeditor.fields import RichTextField


class Category(models.Model):
    """
    Category.
    """

    name = models.CharField(max_length=20)
    slug = models.SlugField(max_length=20)

    def __str__(self):
        return self.name

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


class Tag(models.Model):
    """
    Tag.
    """

    name = models.CharField(max_length=100, unique=True)
    slug = models.SlugField(max_length=100, unique=True)

    def __str__(self):
        return self.name

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

    class Meta:
        verbose_name = _('Tag')


class Post(models.Model):
    """
    Post model.
    """

    title = models.CharField(max_length=150)
    created_date = models.DateTimeField(auto_now_add=True)
    image = models.ImageField(upload_to="pictures/%Y/%m/%d",
                                        blank=True, null=True)
    text = RichTextField(max_length=10000)
    slug = models.SlugField(max_length=150, unique=True)
    tags = models.ManyToManyField(Tag, null=True, blank=True)
    category = models.ForeignKey(Category)

    def __str__(self):
        return self.title

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

    class Meta:
        ordering = ['-created_date']
        verbose_name = _('Eco Post')
        verbose_name_plural = _('Eco Posts')

Проследить:

ERROR: test_display_post (posts.tests.PostTests)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/boshepelinski/Documents/Projects/Django/ecohata/ecohata/posts/tests.py", line 17, in setUp
    self.poster = factories.PostFactory.create(tags=('tag-1'))
  File "/Users/boshepelinski/Documents/Projects/Django/ecohata/lib/python3.4/site-packages/factory/base.py", line 585, in create
    return cls._generate(True, attrs)
  File "/Users/boshepelinski/Documents/Projects/Django/ecohata/lib/python3.4/site-packages/factory/base.py", line 516, in _generate
    results[name] = decl.call(obj, create, extraction_context)
  File "/Users/boshepelinski/Documents/Projects/Django/ecohata/lib/python3.4/site-packages/factory/declarations.py", line 490, in call
    extraction_context.value, **extraction_context.extra)
  File "/Users/boshepelinski/Documents/Projects/Django/ecohata/ecohata/posts/factories.py", line 41, in tags
    self.tags.add(tag)
  File "/Users/boshepelinski/Documents/Projects/Django/ecohata/lib/python3.4/site-packages/django/db/models/fields/related.py", line 1175, in __get__
    through=self.field.rel.through,
  File "/Users/boshepelinski/Documents/Projects/Django/ecohata/lib/python3.4/site-packages/django/db/models/fields/related.py", line 831, in __init__
    (instance, source_field_name))
ValueError: "<Post: postik-0>" needs to have a value for field "post" before this many-to-many relationship can be used.

person jwshadow    schedule 01.03.2015    source источник


Ответы (1)


Исправление довольно простое: ваши фабрики наследуются от factory.Factory, но вы используете модели Django. Вместо этого вы должны унаследовать factory.django.DjangoModelFactory.

В противном случае factory_boy не знает, что он должен save() сгенерировать ваши объекты, и впоследствии вы получите ошибку.

У вас также есть еще одна проблема в вашем коде: когда вы пишете factories.PostFactory.create(tags=('tag-1')), это фактически то же самое, что писать factories.PostFactory.create(tags='tag-1).

Однако:

  • Определение вашего объявления @post_generation tags предполагает список тегов, поэтому вы должны использовать factories.PostFactory.create(tags=['tag-1'])
  • На самом деле, поскольку он вызывает self.tags.add(tag), он ожидает итерации из Tag объектов; его надо вызывать через factories.PostFactory.create(tags=[TagFactory()]).
person Xelnor    schedule 02.03.2015
comment
Большое спасибо, Кселнор! - person jwshadow; 02.03.2015
comment
Я столкнулся с той же проблемой, что и @jwshadow, потому что я полагаю, что мы оба следовали одному и тому же рецепт на странице Factory Boy. Должны ли мы представить исправление в документации? - person luizs81; 22.08.2016