mdblog/markdownblog/blog/factories.py

24 lines
564 B
Python

import factory
from blog.models import Topic, Blogpost, Tag
class TopicFactory(factory.Factory):
class Meta:
model = Topic
name = factory.Faker("word")
numbered = False
class PostFactory(factory.Factory):
class Meta:
model = Blogpost
title = factory.Faker("word")
tags = Tag.objects.all()[0] if len(Tag.objects.all()) != 0 else None
if len(Topic.objects.all()) == 0:
TopicFactory.create_batch(10)
topics = Topic.objects.all()[0] if len(Topic.objects.all()) != 0 else None
mdfile = "/tmp/test.md"