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

View File

@ -5,7 +5,7 @@ import wagtail_factories
from django.contrib.auth import get_user_model
from factory import CREATE_STRATEGY
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 assignments.models import Assignment
@ -77,6 +77,12 @@ class InstrumentFactory(BasePageFactory):
class Meta:
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):
description = factory.LazyAttribute(fake_paragraph)
@ -166,8 +172,6 @@ class ContentBlockFactory(BasePageFactory):
module=module
)
kwargs['{}__{}__{}__{}'.format(stream_field_name, idx, block_type, 'assignment')] = assignment
else:
for jdx, field in enumerate(value):