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.factories import UserFactory
from rooms.factories import RoomEntryFactory, RoomFactory
from rooms.models import RoomEntry
from users.factories import SchoolClassFactory
from users.services import create_users


class RoomEntryMutationsTestCase(TestCase):
    def setUp(self):
        self.user = UserFactory(username='aschi')
        self.another_user = UserFactory(username='pesche')
        s = SchoolClassFactory(users=[self.user, self.another_user])
        self.room_entry = RoomEntryFactory(author=self.user, room=RoomFactory(school_class=s))

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

    def test_delete_room_entry(self):
        self.assertEqual(RoomEntry.objects.count(), 1)

        mutation = '''
        mutation DeleteRoomEntry($input: DeleteRoomEntryInput!) {
              deleteRoomEntry(input: $input) {
                success
                errors
              }
            }
        '''

        result = self.client.execute(mutation, variables={
            'input': {
                'id': to_global_id('RoomEntryNode', self.room_entry.pk)
            }
        })
        self.assertIsNone(result.get('errors'))
        self.assertEqual(RoomEntry.objects.count(), 0)

    def test_delete_room_not_owner(self):
        self.assertEqual(RoomEntry.objects.count(), 1)
        mutation = '''
                mutation DeleteRoomEntry($input: DeleteRoomEntryInput!) {
                      deleteRoomEntry(input: $input) {
                        success
                        errors
                      }
                    }
                '''

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

        result = client.execute(mutation, variables={
            'input': {
                'id': to_global_id('RoomEntryNode', self.room_entry.pk)
            }
        })

        self.assertIsNotNone(result.get('errors'))
        self.assertEqual(RoomEntry.objects.count(), 1)