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, ChapterFactory, ContentBlockFactory
from users.models import User, SchoolClass
from users.services import create_users

MODULE_QUERY = """
query ModulesQuery($slug: String!) {
  module(slug: $slug) {
      id
      title
      chapters {
        edges {
            node {
                id
                contentBlocks {
                    edges {
                        node {
                            id
                            visibleFor {
                                edges {
                                    node {
                                        name
                                    }
                                }
                            }
                            hiddenFor {
                                edges {
                                    node {
                                        name
                                    }   
                                }
                            }
                        }
                    }
                }
            }
        }
      }
  }  
}
"""


class CreateSnapshotTestCase(TestCase):
    def setUp(self):
        create_users()
        skillbox_class = SchoolClass.objects.get(name='skillbox')
        second_class = SchoolClass.objects.get(name='second_class')
        # teacher will create snapshot
        self.teacher = User.objects.get(username='teacher')
        self.module = ModuleFactory(slug='some-module')
        # module M has a chapter
        self.chapter = ChapterFactory(parent=self.module, slug='some-chapter')
        # chapter has some content blocks a, b, c
        self.a = ContentBlockFactory(parent=self.chapter, module=self.module, slug='cb-a')
        self.b = ContentBlockFactory(parent=self.chapter, module=self.module, slug='cb-b')
        # content block c is user created
        self.c = ContentBlockFactory(parent=self.chapter, owner=self.teacher, user_created=True, module=self.module, slug='cb-c')
        # content block a and c are visible to school class X
        self.b.hidden_for.add(skillbox_class)
        self.c.visible_for.add(skillbox_class)

        request = RequestFactory().get('/')
        request.user = self.teacher
        self.client = Client(schema=schema, context_value=request)


        # we make a snapshot S of the module M
        # snapshot S looks like module M for school class X

    def test_visible_and_hidden(self):
        result = self.client.execute(MODULE_QUERY, variables={
            'slug': self.module.slug
        })
        self.assertIsNone(result.get('errors'))
        module = result.get('data').get('module')
        chapter = module.get('chapters').get('edges')[0]['node']
        self.assertIsNotNone(chapter)
        content_blocks = [edge['node'] for edge in chapter.get('contentBlocks').get('edges')]
        content_block_ids = [node['id'] for node in content_blocks]
        self.assertTrue(to_global_id('ContentBlockNode', self.a.id) in content_block_ids)
        self.assertTrue(to_global_id('ContentBlockNode', self.b.id) in content_block_ids)
        self.assertTrue(to_global_id('ContentBlockNode', self.c.id) in content_block_ids)
        b = [node for node in content_blocks if node['id'] == to_global_id('ContentBlockNode', self.b.id)][0]
        c = [node for node in content_blocks if node['id'] == to_global_id('ContentBlockNode', self.c.id)][0]
        self.assertTrue('skillbox' in [edge['node']['name'] for edge in b.get('hiddenFor').get('edges')])
        self.assertTrue('skillbox' in [edge['node']['name'] for edge in c.get('visibleFor').get('edges')])