Fix tests

This commit is contained in:
Jonathan Kim 2020-01-30 10:11:05 +00:00
parent 82fcb37d8b
commit e2083594d9
2 changed files with 8 additions and 8 deletions

View File

@ -340,7 +340,7 @@ def test_generate_enum_name():
# Simple case # Simple case
field = graphene.Field(graphene.String, name="type") field = graphene.Field(graphene.String, name="type")
model_meta = MockDjangoModelMeta(app_label="users", object_name="User") model_meta = MockDjangoModelMeta(app_label="users", object_name="User")
assert generate_enum_name(model_meta, field) == "DjangoModelUsersUserTypeChoices" assert generate_enum_name(model_meta, field) == "UsersUserTypeChoices"
# More complicated multiple work case # More complicated multiple work case
field = graphene.Field(graphene.String, name="fizz_buzz") field = graphene.Field(graphene.String, name="fizz_buzz")
@ -349,7 +349,7 @@ def test_generate_enum_name():
) )
assert ( assert (
generate_enum_name(model_meta, field) generate_enum_name(model_meta, field)
== "DjangoModelSomeLongAppNameSomeObjectFizzBuzzChoices" == "SomeLongAppNameSomeObjectFizzBuzzChoices"
) )
graphene_settings.DJANGO_CHOICE_FIELD_ENUM_V3_NAMING = False graphene_settings.DJANGO_CHOICE_FIELD_ENUM_V3_NAMING = False

View File

@ -514,19 +514,19 @@ class TestDjangoObjectType:
query: Query query: Query
} }
enum CustomEnumKind {
CAT
DOG
}
type PetModelKind { type PetModelKind {
id: ID! id: ID!
kind: CustomEnumKind! kind: TestsPetModelKindChoices!
} }
type Query { type Query {
pet: PetModelKind pet: PetModelKind
} }
enum TestsPetModelKindChoices {
CAT
DOG
}
""" """
) )
graphene_settings.DJANGO_CHOICE_FIELD_ENUM_V3_NAMING = False graphene_settings.DJANGO_CHOICE_FIELD_ENUM_V3_NAMING = False