Test with psycopg2 and psycopg3

Author: mikicz

tests/requirements.txt

@@ -13,7 +13,7 @@
 
 DATABASES = {
     "default": {
-        "ENGINE": "django.db.backends.postgresql_psycopg2",  # Add 'postgresql_psycopg2', 'mysql', 'sqlite3' or 'oracle'.
+        "ENGINE": "django.db.backends.postgresql",  # Add 'postgresql', 'mysql', 'sqlite3' or 'oracle'.
         "NAME": "django_pgviews",  # Or path to database file if using sqlite3.
         # The following settings are not used with sqlite3:
         "USER": "django_pgviews",
@@ -22,7 +22,7 @@
         "PORT": "",  # Set to empty string for default.
     },
     "weather_db": {
-        "ENGINE": "django.db.backends.postgresql_psycopg2",  # Add 'postgresql_psycopg2', 'mysql', 'sqlite3' or 'oracle'.
+        "ENGINE": "django.db.backends.postgresql",  # Add 'postgresql', 'mysql', 'sqlite3' or 'oracle'.
         "NAME": "django_pgviews_weatherdb",  # Or path to database file if using sqlite3.
         # The following settings are not used with sqlite3:
         "USER": "django_pgviews",

tests/test_project/test_project/settings/base.py

@@ -5,15 +5,15 @@
 
 DATABASES = {
     "default": {
-        "ENGINE": "django.db.backends.postgresql_psycopg2",
+        "ENGINE": "django.db.backends.postgresql",
         "NAME": os.environ.get("DB_NAME", "circle_test"),
         "USER": os.environ.get("DB_USER", "postgres"),
         "PASSWORD": os.environ.get("DB_PASSWORD", "testing"),
         "HOST": "localhost",
         "PORT": "5432",
     },
     "weather_db": {
-        "ENGINE": "django.db.backends.postgresql_psycopg2",
+        "ENGINE": "django.db.backends.postgresql",
         "NAME": os.environ.get("DB_NAME_WEATHER", "weatherdb"),
         "USER": os.environ.get("DB_USER", "postgres"),
         "PASSWORD": os.environ.get("DB_PASSWORD", "testing"),

tests/test_project/test_project/settings/ci.py

@@ -1,6 +1,8 @@
 from datetime import timedelta
 
-from django.db import models
+from django.db import models, connections
+from django.db.models import signals
+from django.dispatch import receiver
 from django.utils import timezone
 
 from django_pgviews import view
@@ -92,3 +94,10 @@ class MaterializedRelatedViewWithNoData(view.ReadOnlyMaterializedView):
     sql = """SELECT id AS model_id, id FROM viewtest_testmodel"""
     model = models.ForeignKey(TestModel, on_delete=models.DO_NOTHING)
     with_data = False
+
+
+@receiver(signals.post_migrate)
+def create_test_schema(sender, app_config, using, **kwargs):
+    command = "CREATE SCHEMA IF NOT EXISTS {};".format("test_schema")
+    with connections[using].cursor() as cursor:
+        cursor.execute(command)

tests/test_project/test_project/viewtest/models.py

@@ -3,30 +3,25 @@
 from contextlib import closing
 from datetime import timedelta
 
-import psycopg2.errors
 from django.conf import settings
 from django.contrib import auth
 from django.contrib.auth.models import User
 from django.core.management import call_command
-from django.db import connection, connections, DEFAULT_DB_ALIAS
+from django.db import connection, DEFAULT_DB_ALIAS
 from django.db.utils import OperationalError
-from django.db.models import signals
 from django.dispatch import receiver
 from django.test import TestCase
 from django.utils import timezone
 
 from django_pgviews.signals import view_synced, all_views_synced
 from django_pgviews.view import _schema_and_name
-
 from . import models
 from .models import LatestSuperusers
 
-
-@receiver(signals.post_migrate)
-def create_test_schema(sender, app_config, using, **kwargs):
-    command = "CREATE SCHEMA IF NOT EXISTS {};".format("test_schema")
-    with connections[using].cursor() as cursor:
-        cursor.execute(command)
+try:
+    from psycopg.errors import UndefinedTable
+except ImportError:
+    from psycopg2.errors import UndefinedTable
 
 
 def get_list_of_indexes(cursor, cls):
@@ -145,7 +140,7 @@ def test_refresh_missing(self):
         with connection.cursor() as cursor:
             cursor.execute("DROP MATERIALIZED VIEW viewtest_materializedrelatedview CASCADE;")
 
-        with self.assertRaises(psycopg2.errors.UndefinedTable):
+        with self.assertRaises(UndefinedTable):
             models.MaterializedRelatedView.refresh()
 
     def test_materialized_view_indexes(self):

tests/test_project/test_project/viewtest/tests.py

@@ -1,7 +1,8 @@
 [tox]
 envlist =
-    py{37,38,39,310}-dj{32},
-    py{38,39,310,311}-dj{40,41,42}
+    py{37,38,39,310}-dj{32}-pg2
+    py{38,39,310,311}-dj{40,41}-pg2
+    py{38,39,310,311}-dj{42}-pg{2,3}
 
 [gh]
 python =
@@ -17,7 +18,8 @@ setenv =
     DJANGO_SETTINGS_MODULE = test_project.settings.ci
 changedir = {toxinidir}/tests/test_project
 deps=
-    -rtests/requirements.txt
+    pg2: psycopg2>2.9
+    pg3: psycopg>3.1
     dj32: https://github.com/django/django/archive/stable/3.2.x.tar.gz#egg=django
     dj40: https://github.com/django/django/archive/stable/4.0.x.tar.gz#egg=django
     dj41: https://github.com/django/django/archive/stable/4.1.x.tar.gz#egg=django