69 lines
2.6 KiB
Python
69 lines
2.6 KiB
Python
from django.test import TestCase, RequestFactory
|
|
from graphene.test import Client
|
|
from graphql_relay import to_global_id
|
|
|
|
from api.schema import schema
|
|
from books.factories import ModuleFactory
|
|
from books.models import ContentBlock, Chapter
|
|
from users.factories import SchoolClassFactory
|
|
from users.models import User
|
|
from users.services import create_users
|
|
|
|
|
|
class ContentBlocksVisibleForTeachers(TestCase):
|
|
def setUp(self):
|
|
module = ModuleFactory()
|
|
chapter = Chapter(title='Some Chapter')
|
|
module.add_child(instance=chapter)
|
|
create_users()
|
|
teacher = User.objects.get(username='teacher')
|
|
teacher2 = User.objects.get(username='teacher2')
|
|
student = User.objects.get(username='student1')
|
|
school_class = SchoolClassFactory(users=[teacher, teacher2, student])
|
|
|
|
default_content_block = ContentBlock(title='default block', slug='default')
|
|
user_content_block = ContentBlock(title='user created', slug='user', user_created=True, owner=teacher)
|
|
chapter.specific.add_child(instance=default_content_block)
|
|
chapter.specific.add_child(instance=user_content_block)
|
|
|
|
user_content_block.visible_for.add(school_class)
|
|
user_content_block.save()
|
|
|
|
teacher_request = RequestFactory().get('/')
|
|
teacher_request.user = teacher
|
|
self.teacher_client = Client(schema=schema, context_value=teacher_request)
|
|
|
|
teacher2_request = RequestFactory().get('/')
|
|
teacher2_request.user = teacher2
|
|
self.teacher2_client = Client(schema=schema, context_value=teacher2_request)
|
|
|
|
student_request = RequestFactory().get('/')
|
|
student_request.user = student
|
|
self.student_client = Client(schema=schema, context_value=student_request)
|
|
|
|
self.chapter = to_global_id('ChapterNode', chapter.pk)
|
|
|
|
def _assert_school_classes(self, query, client):
|
|
result = client.execute(query, variables={
|
|
'id': self.chapter
|
|
})
|
|
self.assertIsNone(result.get('errors'))
|
|
self.assertEqual(len(result.get('data').get('chapter').get('contentBlocks')), 2)
|
|
|
|
def test_teacher_created_content_block(self):
|
|
self.assertEqual(ContentBlock.objects.count(), 2)
|
|
|
|
query = """
|
|
query ChapterQuery($id: ID!) {
|
|
chapter(id: $id) {
|
|
contentBlocks {
|
|
id
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
|
|
self._assert_school_classes(query, self.student_client)
|
|
self._assert_school_classes(query, self.teacher_client)
|
|
self._assert_school_classes(query, self.teacher2_client)
|