Fix Unit Tests

server/books/tests/test_module_mutations.py

@@ -1,4 +1,4 @@
-from django.test import TestCase
+from django.test import TestCase, RequestFactory
 from graphene.test import Client
 from graphql_relay import to_global_id
 
@@ -6,24 +6,29 @@ from api.schema import schema
 from api.utils import get_graphql_mutation, get_object
 from books.factories import ContentBlockFactory, ModuleFactory
 from books.models import ContentBlock
+from core.factories import UserFactory
 
 
 class NewContentBlockMutationTest(TestCase):
     def setUp(self):
         module = ModuleFactory()
+        user = UserFactory(username='aschi')
         content_block = ContentBlockFactory(module=module)
 
+        request = RequestFactory().get('/')
+        request.user = user
+        self.client = Client(schema=schema, context_value=request)
+
         self.sibling_id = to_global_id('ContentBlock', content_block.pk)
 
     def test_add_new_content_block(self):
         self.assertEqual(ContentBlock.objects.count(), 1)
-        client = Client(schema=schema)
 
         mutation = get_graphql_mutation('addContentBlock.gql')
 
         title = "Hello World"
 
-        result = client.execute(mutation, variables={
+        result = self.client.execute(mutation, variables={
             'input': {
                 "contentBlock": {
                     "title": title,
@@ -53,13 +58,12 @@ class NewContentBlockMutationTest(TestCase):
 
     def test_addNewContentBlock_withImageUrlBlock(self):
         self.assertEqual(ContentBlock.objects.count(), 1)
-        client = Client(schema=schema)
 
         mutation = get_graphql_mutation('addContentBlock.gql')
 
         title = "Hello World"
 
-        result = client.execute(mutation, variables={
+        result = self.client.execute(mutation, variables={
             'input': {
                 "contentBlock": {
                     "title": title,