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

Fix unit test

This commit is contained in:
Ramon Wenger 2021-09-20 20:33:59 +02:00
parent 51beefe293
commit 3951b3f90c
1 changed files with 19 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
server/portfolio/tests/test_project_query.py
View File

@ -3,6 +3,7 @@ from graphene.test import Client
from graphql_relay import to_global_id from graphql_relay import to_global_id
from api.schema import schema from api.schema import schema
from core.tests.base_test import SkillboxTestCase
from portfolio.factories import ProjectFactory from portfolio.factories import ProjectFactory
from portfolio.models import Project from portfolio.models import Project
from rooms.models import Room from rooms.models import Room
@ -11,27 +12,17 @@ from users.models import User, SchoolClass
from users.services import create_users from users.services import create_users
class ProjectQuery(TestCase): class ProjectQuery(SkillboxTestCase):
def setUp(self): def setUp(self):
create_users() self.createDefault()
self.teacher = User.objects.get(username='teacher') school_class1 = SchoolClassFactory(users=[self.teacher, self.student1])
self.teacher2 = User.objects.get(username='teacher2')
self.student = User.objects.get(username='student1')
self.student2 = User.objects.get(username='student2')
school_class1 = SchoolClassFactory(users=[self.teacher, self.student])
school_class2 = SchoolClassFactory(users=[self.teacher2, self.student2]) school_class2 = SchoolClassFactory(users=[self.teacher2, self.student2])
self.project1 = ProjectFactory(student=self.student)
self.project1 = ProjectFactory(student=self.student1)
self.query = ''' self.query = '''
query ProjectsQuery { query ProjectsQuery {
projects { projects {
edges {
node {
...ProjectParts ...ProjectParts
__typename
}
__typename
}
__typename
} }
} }
@ -49,72 +40,45 @@ class ProjectQuery(TestCase):
def test_should_see_own_projects(self): def test_should_see_own_projects(self):
self.assertEqual(Project.objects.count(), 1) self.assertEqual(Project.objects.count(), 1)
request = RequestFactory().get('/')
request.user = self.student
self.client = Client(schema=schema, context_value=request)
result = self.client.execute(self.query) result = self.get_client(self.student1).execute(self.query)
self.assertIsNone(result.get('errors')) self.assertIsNone(result.get('errors'))
self.assertEqual(result.get('data').get('projects').get('edges')[0].get('node').get('title'), self.project1.title) self.assertEqual(result.get('data').get('projects')[0].get('title'), self.project1.title)
def test_should_not_see_other_projects(self): def test_should_not_see_other_projects(self):
self.assertEqual(Project.objects.count(), 1) self.assertEqual(Project.objects.count(), 1)
request = RequestFactory().get('/')
request.user = self.student2
self.client = Client(schema=schema, context_value=request)
result = self.client.execute(self.query) result = self.get_client(self.student2).execute(self.query)
self.assertIsNone(result.get('errors')) self.assertIsNone(result.get('errors'))
self.assertEqual(len(result.get('data').get('projects').get('edges')), 0) self.assertEqual(len(result.get('data').get('projects')), 0)
def test_teacher_should_not_see_unfinished_projects(self): def test_teacher_should_not_see_unfinished_projects(self):
request = RequestFactory().get('/') result = self.get_client().execute(self.query)
request.user = self.teacher
self.client = Client(schema=schema, context_value=request)
result = self.client.execute(self.query)
self.assertIsNone(result.get('errors')) self.assertIsNone(result.get('errors'))
self.assertEqual(len(result.get('data').get('projects').get('edges')), 0) self.assertEqual(len(result.get('data').get('projects')), 0)
def test_teacher_should_only_see_finished_projects(self):
self.project1.final = True
self.assertEqual(Project.objects.count(), 1)
request = RequestFactory().get('/')
request.user = self.teacher
self.client = Client(schema=schema, context_value=request)
result = self.client.execute(self.query)
self.assertIsNone(result.get('errors'))
self.assertEqual(result.get('data').get('projects').get('edges')[0].get('node').get('title'),
self.project1.title)
def test_teacher_should_only_see_finished_projects(self): def test_teacher_should_only_see_finished_projects(self):
self.project1.final = True self.project1.final = True
self.project1.save() self.project1.save()
self.assertEqual(Project.objects.count(), 1) self.assertEqual(Project.objects.count(), 1)
request = RequestFactory().get('/')
request.user = self.teacher
self.client = Client(schema=schema, context_value=request)
result = self.client.execute(self.query) result = self.get_client().execute(self.query)
self.assertIsNone(result.get('errors')) self.assertIsNone(result.get('errors'))
self.assertEqual(result.get('data').get('projects').get('edges')[0].get('node').get('title'), self.assertEqual(result.get('data').get('projects')[0].get('title'),
self.project1.title) self.project1.title)
def test_other_teacher_should_not_see_projects(self): def test_other_teacher_should_not_see_projects(self):
self.project1.final = True self.project1.final = True
self.project1.save() self.project1.save()
self.assertEqual(Project.objects.count(), 1) self.assertEqual(Project.objects.count(), 1)
request = RequestFactory().get('/')
request.user = self.teacher2
self.client = Client(schema=schema, context_value=request)
result = self.client.execute(self.query)
result = self.get_client(self.teacher2).execute(self.query)
self.assertIsNone(result.get('errors')) self.assertIsNone(result.get('errors'))
self.assertEqual(len(result.get('data').get('projects').get('edges')), 0) self.assertEqual(len(result.get('data').get('projects')), 0)