From e524fe610743fef50ebff4a1b00aa01536ea93d9 Mon Sep 17 00:00:00 2001 From: Ramon Wenger Date: Thu, 4 Oct 2018 13:51:18 +0200 Subject: [PATCH] Update factory to include image URL block generation --- server/books/factories.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/server/books/factories.py b/server/books/factories.py index d30515db..913bacbf 100644 --- a/server/books/factories.py +++ b/server/books/factories.py @@ -85,7 +85,7 @@ class VideoBlockFactory(wagtail_factories.StructBlockFactory): model = VideoBlock -block_types = ['text_block', 'basic_knowledge', 'student_entry', 'image_block', 'task'] +block_types = ['text_block', 'basic_knowledge', 'student_entry', 'image_url_block', 'task'] class ContentBlockFactory(BasePageFactory): @@ -176,9 +176,12 @@ class ContentBlockFactory(BasePageFactory): elif block_type == 'assignment': kwargs['{}__{}__{}__{}'.format(stream_field_name, i, 'assignment', 'task_text')] = RichText( fake_paragraph()) - elif block_type == 'image_block': + elif block_type == 'image_url_block': kwargs[ - '{}__{}__{}__{}'.format(stream_field_name, i, 'image_block', 'image__title')] = fake_paragraph() + '{}__{}__{}__{}'.format(stream_field_name, i, 'image_url_block', 'title')] = fake_paragraph() + kwargs[ + '{}__{}__{}__{}'.format(stream_field_name, i, 'image_url_block', + 'url')] = 'https://picsum.photos/400/?random={}'.format(''.join(random.choice('abcdefghiklmn') for _ in range(6))) elif block_type == 'task': kwargs['{}__{}__{}__{}'.format(stream_field_name, i, 'task', 'text')] = RichText(fake_paragraph())