From 933c6af14922a368b0172ebcfb634eac2c8a4803 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lucas=20Br=C3=A9mond?= Date: Fri, 1 Jan 2021 23:11:07 -0800 Subject: [PATCH] Fix test_should_query_postgres_fields --- graphene_django/compat.py | 5 ++++- graphene_django/tests/test_query.py | 4 ++-- setup.py | 1 - 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/graphene_django/compat.py b/graphene_django/compat.py index 8a2b933..537fd1d 100644 --- a/graphene_django/compat.py +++ b/graphene_django/compat.py @@ -6,13 +6,16 @@ try: # Postgres fields are only available in Django with psycopg2 installed # and we cannot have psycopg2 on PyPy from django.contrib.postgres.fields import ( + IntegerRangeField, ArrayField, HStoreField, JSONField as PGJSONField, RangeField, ) except ImportError: - ArrayField, HStoreField, PGJSONField, RangeField = (MissingType,) * 4 + IntegerRangeField, ArrayField, HStoreField, PGJSONField, RangeField = ( + MissingType, + ) * 5 try: # JSONField is only available from Django 3.1 diff --git a/graphene_django/tests/test_query.py b/graphene_django/tests/test_query.py index 2ff81a3..699814d 100644 --- a/graphene_django/tests/test_query.py +++ b/graphene_django/tests/test_query.py @@ -11,7 +11,7 @@ from py.test import raises import graphene from graphene.relay import Node -from ..compat import JSONField, MissingType +from ..compat import IntegerRangeField, MissingType from ..fields import DjangoConnectionField from ..types import DjangoObjectType from ..utils import DJANGO_FILTER_INSTALLED @@ -114,7 +114,7 @@ def test_should_query_well(): assert result.data == expected -@pytest.mark.skipif(JSONField is MissingType, reason="RangeField should exist") +@pytest.mark.skipif(IntegerRangeField is MissingType, reason="RangeField should exist") def test_should_query_postgres_fields(): from django.contrib.postgres.fields import ( IntegerRangeField, diff --git a/setup.py b/setup.py index 3358ab1..4750671 100644 --- a/setup.py +++ b/setup.py @@ -20,7 +20,6 @@ tests_require = [ "coveralls", "mock", "pytz", - "psycopg2-binary", "django-filter>=2", "pytest-django>=3.3.2", ] + rest_framework_require