10

factory_boy現在、テストでフィクスチャを作成するために 使用しています。Factory_boyドキュメントでは、モデル内のフィールドのSubFactoryように機能する可能性があることについてのみ言及されています。ForeignKeyただし、ManyToMany アソシエーションには何もありませんでした。次の Post モデルがある場合、そのファクトリーを作成するにはどうすればよいでしょうか?

class Post(models.Model):
    title = models.CharField(max_length=100)
    tags = models.ManyToManyField('tags.Tag')

class PostFactory(factory.Factory):
    FACTORY_FOR = Post

    title = 'My title'
    tags = ???
4

3 に答える 3

11

post_generation フックについてはどうですか- factory_boyの新しいバージョンを使用していると思いますか?

import random
import factory

class PostFactory(factory.Factory):
    FACTORY_FOR = Post
    title = factory.Sequence(lambda n: "This is test title number" + n)
    @factory.post_generation(extract_prefix='tags')
    def add_tags(self, create, extracted, **kwargs):
        # allow something like PostFactory(tags = Tag.objects.filter())
        if extracted and type(extracted) == type(Tag.objects.all()):
            self.tags = extracted
            self.save()
        else:
            if Tag.objects.all().count() < 5:
                TagFactory.create_batch(5, **kwargs)
            for tag in Tag.objects.all().order_by('?')[:random.randint(1, 5)]:
                self.tags.add(tag)

を使用できますが、シーケンスでPostFactory(tags__field = 'some fancy default text')適切に作成することをお勧めします...TagFactory

バインドできるはずですPostFactory(tags = Tag.objects.filter())が、この部分はテストされていません...

于 2012-09-11T08:05:23.953 に答える
7

クラスメソッドをオーバーライドできます_prepare:

class PostFactory(Factory):
    FACTORY_FOR = Post

    title = 'My title'

    @classmethod
    def _prepare(cls, create, **kwargs):
        post = super(PostFactory, cls)._prepare(create, **kwargs)
        if post.id:
            post.tags = Tag.objects.all()
        return post

投稿に ID がない場合、投稿にタグを追加できないことに注意してください。

于 2012-07-19T05:29:27.087 に答える
2

私はそれをテストしませんでしたが、問題は何ですか:

class PostFactory(factory.Factory):
    FACTORY_FOR = Post
    title = 'My title'

class TagFactory(factory.Factory):
    FACTORY_FOR = Tag

post = PostFactory()
tag = TagFactory()
post.tags.add(tag)
于 2012-05-19T01:28:33.810 に答える