skillbox/server/rooms/tests/test_comments.py

37 lines
1.0 KiB
Python

from graphql_relay import to_global_id
from core.tests.base_test import SkillboxTestCase
from rooms.factories import RoomEntryFactory
class CommentTestCase(SkillboxTestCase):
def setUp(self) -> None:
self.createDefault()
def test_add_comment(self):
room_entry = RoomEntryFactory()
text = 'First!!!'
mutation = """
mutation AddComment($input: AddCommentInput!) {
addComment(input: $input) {
success
}
}
"""
self.assertEqual(room_entry.comments.count(), 0)
room_entry_id = to_global_id('RoomEntryNode', room_entry.id)
result = self.get_client().execute(mutation, variables={
'input': {
'roomEntry': room_entry_id,
'comment': text
}
})
self.assertIsNone(result.get('errors'))
self.assertEqual(room_entry.comments.count(), 1)
comment = room_entry.comments.first()
self.assertEqual(comment.text, text)
self.assertEqual(comment.owner.id, self.teacher.id)