Fix unit tests
This commit is contained in:
parent
e1a87739df
commit
812021cb54
|
|
@ -9,10 +9,10 @@ from wagtail.core.models import Page, Site
|
|||
from wagtail.core.rich_text import RichText
|
||||
|
||||
from assignments.models import Assignment
|
||||
from basicknowledge.models import BasicKnowledge
|
||||
from books.blocks import BasicKnowledgeBlock, ImageUrlBlock, LinkBlock, AssignmentBlock, VideoBlock
|
||||
from books.models import Book, Topic, Module, Chapter, ContentBlock, TextBlock
|
||||
from core.factories import BasePageFactory, fake, DummyImageFactory, fake_paragraph, fake_title
|
||||
from basicknowledge.models import BasicKnowledge, INTERDISCIPLINARY, InstrumentType, LANGUAGE_COMMUNICATION, SOCIETY
|
||||
from books.blocks import AssignmentBlock, BasicKnowledgeBlock, ImageUrlBlock, LinkBlock, VideoBlock
|
||||
from books.models import Book, Chapter, ContentBlock, Module, TextBlock, Topic
|
||||
from core.factories import BasePageFactory, DummyImageFactory, fake, fake_paragraph, fake_title
|
||||
|
||||
|
||||
class BookFactory(BasePageFactory):
|
||||
|
|
@ -70,9 +70,18 @@ class TextBlockFactory(wagtail_factories.StructBlockFactory):
|
|||
model = TextBlock
|
||||
|
||||
|
||||
class InstrumentTypeFactory(factory.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = InstrumentType
|
||||
|
||||
category = factory.Iterator([LANGUAGE_COMMUNICATION, SOCIETY, INTERDISCIPLINARY])
|
||||
name = factory.LazyAttribute(lambda x: fake.text(max_nb_chars=20))
|
||||
|
||||
|
||||
class InstrumentFactory(BasePageFactory):
|
||||
title = factory.LazyAttribute(fake_title)
|
||||
type = factory.Iterator([BasicKnowledge.LANGUAGE_COMMUNICATION, BasicKnowledge.SOCIETY, BasicKnowledge.INTERDISCIPLINARY])
|
||||
old_type = factory.Iterator([LANGUAGE_COMMUNICATION, SOCIETY, INTERDISCIPLINARY])
|
||||
new_type = factory.SubFactory(InstrumentTypeFactory)
|
||||
|
||||
class Meta:
|
||||
model = BasicKnowledge
|
||||
|
|
@ -83,7 +92,6 @@ class InstrumentFactory(BasePageFactory):
|
|||
return super()._create(model_class, *args, **kwargs)
|
||||
|
||||
|
||||
|
||||
class BasicKnowledgeBlockFactory(wagtail_factories.StructBlockFactory):
|
||||
description = factory.LazyAttribute(fake_paragraph)
|
||||
basic_knowledge = factory.SubFactory(InstrumentFactory)
|
||||
|
|
|
|||
Loading…
Reference in New Issue