from django.test import TestCase, RequestFactory
from graphene.test import Client
from graphql_relay import to_global_id

from api.schema import schema
from core.tests.base_test import SkillboxTestCase
from portfolio.factories import ProjectFactory
from portfolio.models import Project
from rooms.models import Room
from users.factories import SchoolClassFactory
from users.models import User, SchoolClass
from users.services import create_users


class ProjectQuery(SkillboxTestCase):
    def setUp(self):
        self.createDefault()
        school_class1 = SchoolClassFactory(users=[self.teacher, self.student1])
        school_class2 = SchoolClassFactory(users=[self.teacher2, self.student2])

        self.project1 = ProjectFactory(student=self.student1)
        self.query = '''
            query ProjectsQuery {
              projects {
                    ...ProjectParts                    
              }
            }
            
            fragment ProjectParts on ProjectNode {
              id
              title
              appearance
              description
              slug
              objectives
              __typename
            }
            
        '''

    def test_should_see_own_projects(self):
        self.assertEqual(Project.objects.count(), 1)

        result = self.get_client(self.student1).execute(self.query)

        self.assertIsNone(result.get('errors'))
        self.assertEqual(result.get('data').get('projects')[0].get('title'), self.project1.title)

    def test_should_not_see_other_projects(self):
        self.assertEqual(Project.objects.count(), 1)

        result = self.get_client(self.student2).execute(self.query)

        self.assertIsNone(result.get('errors'))
        self.assertEqual(len(result.get('data').get('projects')), 0)

    def test_teacher_should_not_see_unfinished_projects(self):
        result = self.get_client().execute(self.query)

        self.assertIsNone(result.get('errors'))
        self.assertEqual(len(result.get('data').get('projects')), 0)

    def test_teacher_should_only_see_finished_projects(self):
        self.project1.final = True
        self.project1.save()
        self.assertEqual(Project.objects.count(), 1)

        result = self.get_client().execute(self.query)

        self.assertIsNone(result.get('errors'))
        self.assertEqual(result.get('data').get('projects')[0].get('title'),
                         self.project1.title)


    def test_other_teacher_should_not_see_projects(self):
        self.project1.final = True
        self.project1.save()
        self.assertEqual(Project.objects.count(), 1)


        result = self.get_client(self.teacher2).execute(self.query)

        self.assertIsNone(result.get('errors'))
        self.assertEqual(len(result.get('data').get('projects')), 0)