skillbox/server/core/tests/test_api.py

51 lines
1.6 KiB
Python

import json
from django.test import TestCase, Client
from core.factories import UserFactory
class ApiAccessTestCase(TestCase):
def setUp(self):
self.query = json.dumps({
'operationName': 'ModulesQuery',
'query': 'query ModulesQuery { modules { edges { node { id }}}}',
'variables': None,
})
def test_graphqlEndpoint_shouldNotBeAccessibleWithoutLogin(self):
c = Client()
response = c.post('/api/graphql/', data=self.query, content_type='application/json')
self.assertEqual(response.status_code, 402)
def test_graphqlEndpoint_shouldBeAccessibleForSuperUser(self):
UserFactory(username='admin', is_staff=True, is_active=True, is_superuser=True)
c = Client()
c.login(username='admin', password='test')
response = c.post('/api/graphql/', data=self.query, content_type='application/json')
self.assertEqual(200, response.status_code)
def test_publicGraphqlEndpoint_shouldBeAccessibleWithoutLogin(self):
query = json.dumps({
'operationName': 'BetaLogin',
'query': '''
mutation BetaLogin($input: BetaLoginInput!){
betaLogin(input: $input) {
success
}
}
''',
'variables': {
'input': {
'usernameInput': 'test',
'passwordInput': 'test'
}
},
})
c = Client()
response = c.post('/api/graphql-public/', data=query, content_type='application/json')
self.assertEqual(response.status_code, 200)