ramon
/
skillbox
1
0
Fork 0
Code Issues 1 Pull Requests 1 Packages Projects 1 Releases 1 Wiki Activity

Fix dummy data

This commit is contained in:
Ramon Wenger 2020-05-26 11:19:06 +02:00
parent 74c4c4e8fe
commit 09153782d5
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/books/factories.py
View File

@ -5,7 +5,7 @@ import wagtail_factories
from django.contrib.auth import get_user_model from django.contrib.auth import get_user_model
from factory import CREATE_STRATEGY from factory import CREATE_STRATEGY
from wagtail.core import blocks from wagtail.core import blocks
from wagtail.core.models import Page from wagtail.core.models import Page, Site
from wagtail.core.rich_text import RichText from wagtail.core.rich_text import RichText
from assignments.models import Assignment from assignments.models import Assignment
@ -77,6 +77,12 @@ class InstrumentFactory(BasePageFactory):
class Meta: class Meta:
model = BasicKnowledge model = BasicKnowledge
@classmethod
def _create(cls, model_class, *args, **kwargs):
kwargs['parent'] = Site.objects.get(is_default_site=True).root_page
return super()._create(model_class, *args, **kwargs)
class BasicKnowledgeBlockFactory(wagtail_factories.StructBlockFactory): class BasicKnowledgeBlockFactory(wagtail_factories.StructBlockFactory):
description = factory.LazyAttribute(fake_paragraph) description = factory.LazyAttribute(fake_paragraph)
@ -166,8 +172,6 @@ class ContentBlockFactory(BasePageFactory):
module=module module=module
) )
kwargs['{}__{}__{}__{}'.format(stream_field_name, idx, block_type, 'assignment')] = assignment kwargs['{}__{}__{}__{}'.format(stream_field_name, idx, block_type, 'assignment')] = assignment
else: else:
for jdx, field in enumerate(value): for jdx, field in enumerate(value):