Fix unit test

This commit is contained in:
Ramon Wenger 2021-04-14 23:25:08 +02:00
parent be9a829b01
commit 0bef8fe570
2 changed files with 8 additions and 9 deletions

View File

@ -37,12 +37,12 @@ class SnapshotManager(models.Manager):
title_hidden=chapter.title_hidden_for.filter(id=school_class.id).exists(),
description_hidden=chapter.description_hidden_for.filter(id=school_class.id).exists()
)
base_qs = ContentBlock.get_by_parent(chapter).filter(snapshotcontentblock__isnull=True)
base_qs = ContentBlock.get_by_parent(chapter).filter(contentblocksnapshot__isnull=True)
for content_block in base_qs.filter(user_created=False):
if content_block.hidden_for.filter(id=school_class.id).exists():
snapshot.hidden_content_blocks.add(content_block)
for content_block in base_qs.filter(user_created=True).filter(owner=user):
new_content_block = SnapshotContentBlock(
new_content_block = ContentBlockSnapshot(
hidden=False,
snapshot=snapshot,
contents=content_block.contents,

View File

@ -66,7 +66,11 @@ class SnapshotChapterNode(DjangoObjectType):
@staticmethod
def resolve_title_hidden(parent, info):
return parent
return parent.title_hidden
@staticmethod
def resolve_description_hidden(parent, info):
return parent.description_hidden
@staticmethod
def resolve_title(parent, info):
@ -82,18 +86,13 @@ class SnapshotChapterNode(DjangoObjectType):
user_created = Q(user_created=True)
hidden_for_snapshot = Q(hidden_for_snapshots=snapshot)
custom_hidden = Q(snapshotcontentblock__hidden=True)
custom_hidden = Q(contentblocksnapshot__hidden=True)
qs = ContentBlock.get_by_parent(parent.chapter) \
.exclude(user_created) \
.exclude(hidden_for_snapshot) \
.exclude(custom_hidden)
# exclude hidden for snapshot
# exclude with owner
# include visible snapshot content blocks
# todo
return qs
def resolve_id(self, *args):