From 933c6af14922a368b0172ebcfb634eac2c8a4803 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lucas=20Br=C3=A9mond?= <lucas.bremond@gmail.com>
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