graphene-django/graphene_django/tests/issues/test_520.py

# https://github.com/graphql-python/graphene-django/issues/520

import datetime

from django import forms

import graphene

from graphene import Field, ResolveInfo
from graphene.types.inputobjecttype import InputObjectType
from py.test import raises
from py.test import mark
from rest_framework import serializers

from ...types import DjangoObjectType
from ...rest_framework.models import MyFakeModel
from ...rest_framework.mutation import SerializerMutation
from ...forms.mutation import DjangoFormMutation


class MyModelSerializer(serializers.ModelSerializer):
    class Meta:
        model = MyFakeModel
        fields = "__all__"


class MyForm(forms.Form):
    text = forms.CharField()


def test_can_use_form_and_serializer_mutations():
    class MyMutation(SerializerMutation):
        class Meta:
            serializer_class = MyModelSerializer

    class MyFormMutation(DjangoFormMutation):
        class Meta:
            form_class = MyForm

    class Mutation(graphene.ObjectType):
        my_mutation = MyMutation.Field()
        my_form_mutation = MyFormMutation.Field()

    graphene.Schema(mutation=Mutation)