From 84b646ed0a8abc1f6f1408f50043bfade9dcf9f8 Mon Sep 17 00:00:00 2001
From: Ramon Wenger <ramon.wenger@iterativ.ch>
Date: Mon, 30 Aug 2021 16:54:06 +0200
Subject: [PATCH] Add unit test for restricted rooms

---
 server/rooms/schema.py                        |  9 +++
 .../rooms/tests/test_room_visibility_query.py | 73 +++++++++++++++++++
 2 files changed, 82 insertions(+)
 create mode 100644 server/rooms/tests/test_room_visibility_query.py

diff --git a/server/rooms/schema.py b/server/rooms/schema.py
index 7d202f3a..aef280cf 100644
--- a/server/rooms/schema.py
+++ b/server/rooms/schema.py
@@ -52,6 +52,15 @@ class RoomNode(DjangoObjectType):
     def resolve_entry_count(self, *args, **kwargs):
         return self.room_entries.count()
 
+    @staticmethod
+    def resolve_room_entries(root: Room, info, **kwargs):
+        user = info.context.user
+        if root.restricted and not user.is_teacher():
+            return root.room_entries.filter(author=user)
+        else:
+            return root.room_entries.all()
+
+
 
 class RoomsQuery(object):
     # room = relay.Node.Field(RoomNode)
diff --git a/server/rooms/tests/test_room_visibility_query.py b/server/rooms/tests/test_room_visibility_query.py
new file mode 100644
index 00000000..97b7be0d
--- /dev/null
+++ b/server/rooms/tests/test_room_visibility_query.py
@@ -0,0 +1,73 @@
+from graphql_relay import to_global_id
+
+from core.tests.base_test import SkillboxTestCase
+from core.tests.helpers import GQLResult
+from rooms.factories import RoomEntryFactory, RoomFactory
+
+class GQLRoom:
+    def __init__(self, result):
+        self.restricted = result.get('restricted')
+        self.room_entries = list(map(lambda x: x['node'], result.get('roomEntries').get('edges')))
+
+
+class TestRoomVisibilityQueryTestCase(SkillboxTestCase):
+    def setUp(self) -> None:
+        self.createDefault()
+        self.room = RoomFactory(school_class=self.school_class, restricted=True)
+        self.room_id = to_global_id('RoomNode', self.room.id)
+        RoomEntryFactory(room=self.room, author=self.student1)
+        RoomEntryFactory(room=self.room, author=self.student2)
+
+        self.query = """
+query RoomQuery ($slug: String!) {
+    room(slug: $slug) {
+        restricted
+        roomEntries {
+          edges {
+            node {
+              id
+            }
+          }
+        }
+    }
+}
+        """
+
+    def _test_room(self, user, length, restricted=True):
+        res = self.get_client(user).execute(self.query, variables={
+            'slug': self.room.slug
+        })
+        result = GQLResult(res)
+        self.assertIsNone(result.errors)
+        room = GQLRoom(result.data.get('room'))
+        self.assertEqual(room.restricted, restricted)
+        self.assertEqual(len(room.room_entries), length)
+
+
+    def test_restricted_query(self):
+        self._test_room(self.teacher, 2)
+        self._test_room(self.student1, 1)
+        self._test_room(self.student2, 1)
+
+    def test_unrestricted_query(self):
+        self.room.restricted = False
+        self.room.save()
+
+        self._test_room(self.teacher, 2, False)
+        self._test_room(self.student1, 2, False)
+        self._test_room(self.student2, 2, False)
+
+        # res = self.get_client().execute(self.query, variables={
+        #     'slug': self.room.slug
+        # })
+        # result = GQLResult(res)
+        # self.assertIsNone(result.errors)
+        # room = GQLRoom(result.data.get('room'))
+        # self.assertTrue(room.restricted)
+        # self.assertEqual(len(room.room_entries), 2)
+        #
+        # res = self.get_client(self.student1).execute(self.query, variables={
+        #     'slug': self.room.slug
+        # })
+
+