From dc6042c2bdf4e899ad153f57a4873a68df4e4b1c Mon Sep 17 00:00:00 2001 From: Thomas Leonard Date: Thu, 25 Feb 2021 13:49:10 +0100 Subject: [PATCH] Fix flaky test --- .../filter/tests/test_enum_filtering.py | 22 +++++++++++++------ 1 file changed, 15 insertions(+), 7 deletions(-) diff --git a/graphene_django/filter/tests/test_enum_filtering.py b/graphene_django/filter/tests/test_enum_filtering.py index 650c55e..befc67f 100644 --- a/graphene_django/filter/tests/test_enum_filtering.py +++ b/graphene_django/filter/tests/test_enum_filtering.py @@ -135,10 +135,18 @@ def test_filter_enum_field_schema_type(schema): in schema_str ) - assert ( - """type Query { - allReporters(offset: Int, before: String, after: String, first: Int, last: Int): ReporterTypeConnection - allArticles(offset: Int, before: String, after: String, first: Int, last: Int, lang: ArticleLang, lang_In: [ArticleLang], reporter_AChoice: ReporterAChoice, reporter_AChoice_In: [ReporterAChoice]): ArticleTypeConnection -}""" - in schema_str - ) + filters = { + "offset": "Int", + "before": "String", + "after": "String", + "first": "Int", + "last": "Int", + "lang": "ArticleLang", + "lang_In": "[ArticleLang]", + "reporter_AChoice": "ReporterAChoice", + "reporter_AChoice_In": "[ReporterAChoice]", + } + + all_articles_filters = schema_str.split(" allArticles(")[1].split("): ArticleTypeConnection\n")[0].split(", ") + for filter_field, gql_type in filters.items(): + assert "{}: {}".format(filter_field, gql_type) in all_articles_filters