diff --git a/graphene_django/utils/testing.py b/graphene_django/utils/testing.py index 584a08b..d923917 100644 --- a/graphene_django/utils/testing.py +++ b/graphene_django/utils/testing.py @@ -61,7 +61,7 @@ def graphql_query( return resp -class GraphQLTestCase(TestCase): +class GraphQLTestMixin(object): """ Based on: https://www.sam.today/blog/testing-graphql-with-graphene-django/ """ @@ -139,3 +139,12 @@ class GraphQLTestCase(TestCase): """ content = json.loads(resp.content) self.assertIn("errors", list(content.keys()), msg or content) + + +class GraphQLTestCase(GraphQLTestMixin, TestCase): + pass + + +class GraphQLTransactionTestCase(GraphQLTestMixin, TransactionTestCase) + pass +