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
field = graphene.Field(graphene.String, name="type")
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
field = graphene.Field(graphene.String, name="fizz_buzz")
@ -349,7 +349,7 @@ def test_generate_enum_name():
)
assert (
generate_enum_name(model_meta, field)
== "DjangoModelSomeLongAppNameSomeObjectFizzBuzzChoices"
== "SomeLongAppNameSomeObjectFizzBuzzChoices"
)
graphene_settings.DJANGO_CHOICE_FIELD_ENUM_V3_NAMING = False

View File

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