from django.test.testcases import TestCase
import graphene
import json
from schema import Query
from waveforms.models import Annotation
class TestGraphQL(TestCase):
"""
Test the GraphQL API queries.
"""
def test_all_annotations(self):
"""
Test querying for annotations and their attributes.
Parameters
----------
N/A
Returns
-------
N/A
"""
schema = graphene.Schema(query=Query, auto_camelcase=False)
query_correct = """
query {
all_annotations {
user
project
record
event
decision
comments
decision_date
}
}
"""
query_incorrect = """
query {
all_annotations {
user
project
record
event
decision
comments
}
}
"""
correct_output = []
for p in Annotation.objects.all():
correct_output.append({
'user': p.user,
'project': p.project,
'record': p.record,
'event': p.event,
'slug': p.slug,
'decision': p.decision,
'comments': p.comments,
'decision_date': p.decision_date,
})
result_correct = schema.execute(query_correct)
self.assertIsNone(result_correct.errors)
result_incorrect = schema.execute(query_incorrect)
self.assertIsNotNone(result_incorrect.errors)
result_correct = json.loads(json.dumps(result_correct.to_dict()))['data']['all_annotations']
matches = [i for i in result_correct if i not in correct_output]
self.assertEqual(matches, [])