Make python2.7 compatible and run black

This commit is contained in:
Kike Isidoro 2019-08-02 17:06:49 +02:00
parent 0425985dab
commit 1d0473cda6

View File

@ -821,17 +821,17 @@ def test_integer_field_filter_type():
def test_filter_filterset_based_on_mixin(): def test_filter_filterset_based_on_mixin():
class ArticleFilterMixin: class ArticleFilterMixin(FilterSet):
@classmethod @classmethod
def get_filters(cls): def get_filters(cls):
filters = super().get_filters() filters = super(FilterSet, cls).get_filters()
filters.update({ filters.update(
'viewer__email__in': django_filters.CharFilter( {
method='filter_email_in', "viewer__email__in": django_filters.CharFilter(
field_name='reporter__email__in', method="filter_email_in", field_name="reporter__email__in"
), )
}) }
)
return filters return filters
@ -858,14 +858,16 @@ def test_filter_filterset_based_on_mixin():
all_articles = DjangoFilterConnectionField(NewArticleFilterNode) all_articles = DjangoFilterConnectionField(NewArticleFilterNode)
reporter = Reporter.objects.create( reporter = Reporter.objects.create(
first_name="John", last_name="Doe", email="john@doe.com") first_name="John", last_name="Doe", email="john@doe.com"
)
article = Article.objects.create( article = Article.objects.create(
headline="Hello", headline="Hello",
reporter=reporter, reporter=reporter,
editor=reporter, editor=reporter,
pub_date=datetime.now(), pub_date=datetime.now(),
pub_date_time=datetime.now()) pub_date_time=datetime.now(),
)
schema = Schema(query=Query) schema = Schema(query=Query)
@ -884,17 +886,7 @@ def test_filter_filterset_based_on_mixin():
""" """
expected = { expected = {
"allArticles": { "allArticles": {"edges": [{"node": {"viewer": {"email": reporter.email}}}]}
"edges": [
{
"node": {
"viewer": {
"email": reporter.email,
}
}
}
]
}
} }
result = schema.execute(query) result = schema.execute(query)