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

Enrich test data generation with ImageFields

This commit is contained in:
Pawel Kowalski 2018-08-16 15:23:41 +02:00
parent a24acad07a
commit 80719e9a65
3 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/book/blocks.py
View File

@ -5,7 +5,7 @@ DEFAULT_RICH_TEXT_FEATURES = ['bold', 'italic', 'link', 'ol', 'ul']
# 'text_block' 'task' # 'text_block' 'task'
class TextBlock(blocks.StructBlock): class TextBlock(blocks.StructBlock):
text = blocks.CharBlock() text = blocks.RichTextBlock()
# 'modal_text' # 'modal_text'

16
server/book/factories.py
View File

@ -4,6 +4,7 @@ import factory
import wagtail_factories import wagtail_factories
from factory import CREATE_STRATEGY from factory import CREATE_STRATEGY
from wagtail.core.rich_text import RichText
from book.blocks import ModalTextBlock from book.blocks import ModalTextBlock
from book.models import Book, Topic, Module, Chapter, ContentBlock, TextBlock from book.models import Book, Topic, Module, Chapter, ContentBlock, TextBlock
@ -56,6 +57,9 @@ class ModalTextBlockFactory(wagtail_factories.StructBlockFactory):
model = ModalTextBlock model = ModalTextBlock
block_types = ['text_block', 'modal_text', 'student_entry', 'image_block', 'task']
class ContentBlockFactory(BasePageFactory): class ContentBlockFactory(BasePageFactory):
class Meta: class Meta:
model = ContentBlock model = ContentBlock
@ -65,6 +69,7 @@ class ContentBlockFactory(BasePageFactory):
contents = wagtail_factories.StreamFieldFactory({ contents = wagtail_factories.StreamFieldFactory({
'text_block': TextBlockFactory, 'text_block': TextBlockFactory,
'modal_text': ModalTextBlockFactory, 'modal_text': ModalTextBlockFactory,
'image_block': wagtail_factories.ImageChooserBlockFactory
}) })
@classmethod @classmethod
@ -73,11 +78,18 @@ class ContentBlockFactory(BasePageFactory):
for idx, resource in enumerate(kwargs[stream_field_name]): for idx, resource in enumerate(kwargs[stream_field_name]):
value = resource['value'] value = resource['value']
for jdx, field in enumerate(value): for jdx, field in enumerate(value):
kwargs['{}__{}__{}__{}'.format(stream_field_name, idx, resource['type'], field)] = value[field] kwargs['{}__{}__{}__{}'.format(stream_field_name, idx, resource['type'], field)] = RichText(value[field])
del kwargs[stream_field_name] del kwargs[stream_field_name]
else: else:
for i in range(0, random.randint(3, 7)): for i in range(0, random.randint(3, 7)):
kwargs['{}__{}__{}__b'.format(stream_field_name, i, 'text_block', 'text')] = fake_title_noparam() block_type = random.choice(block_types)
if block_type == 'text_block':
kwargs['{}__{}__{}__{}'.format(stream_field_name, i, 'text_block', 'text')] = RichText(fake_title_noparam())
elif block_type == 'image_block':
kwargs['{}__{}__{}__{}'.format(stream_field_name, i, 'image_block', 'image__title')] = fake_title_noparam()
# for i in range(4, 6):
# kwargs['{}__{}__{}__{}'.format(stream_field_name, i, 'image_block', 'image')] = ''
@classmethod @classmethod
def create(cls, **kwargs): def create(cls, **kwargs):

4
server/core/management/commands/dummy_data.py
View File

@ -69,14 +69,14 @@ data = [
'type': 'text_block', 'type': 'text_block',
'value': { 'value': {
'type': 'text_block', 'type': 'text_block',
'text': 'Sie haben diesen Sommer ihre Lehre begonnen. Was bedeutet dieser neue Abschnitt für Sie?\nHalten Sie Ihre Erfahrungen im Bereich fest und stellen Sie diese anschliessend der Klasse vor.' 'text': '<p>Sie haben diesen Sommer ihre Lehre begonnen. Was bedeutet dieser neue Abschnitt für Sie?<br/>Halten Sie Ihre Erfahrungen im Bereich fest und stellen Sie diese anschliessend der Klasse vor.</p>'
} }
}, },
{ {
'type': 'text_block', 'type': 'text_block',
'value': { 'value': {
'type': 'text_block', 'type': 'text_block',
'text': 'Das folgende Interview bezieht sich auf Jugendliche, die Ihre Lehre im Sommer begonnen haben. Lesen Sie das Interview durch und bearbeiten Sie anschliessend die Aufgaben.' 'text': '<p>Das folgende Interview bezieht sich auf Jugendliche, die Ihre Lehre im Sommer begonnen haben.<p></p>Lesen Sie das Interview durch und bearbeiten Sie anschliessend die Aufgaben.</p>'
} }
}, },
# { # {