summaryrefslogtreecommitdiff
path: root/tests/backends/postgresql
diff options
context:
space:
mode:
Diffstat (limited to 'tests/backends/postgresql')
-rw-r--r--tests/backends/postgresql/tests.py20
1 files changed, 19 insertions, 1 deletions
diff --git a/tests/backends/postgresql/tests.py b/tests/backends/postgresql/tests.py
index af08f6f286..ce8ed7b4d5 100644
--- a/tests/backends/postgresql/tests.py
+++ b/tests/backends/postgresql/tests.py
@@ -4,7 +4,13 @@ from io import StringIO
from unittest import mock
from django.core.exceptions import ImproperlyConfigured
-from django.db import DEFAULT_DB_ALIAS, DatabaseError, connection, connections
+from django.db import (
+ DEFAULT_DB_ALIAS,
+ DatabaseError,
+ NotSupportedError,
+ connection,
+ connections,
+)
from django.db.backends.base.base import BaseDatabaseWrapper
from django.test import TestCase, override_settings
@@ -303,3 +309,15 @@ class Tests(TestCase):
[q["sql"] for q in connection.queries],
[copy_expert_sql, "COPY django_session TO STDOUT"],
)
+
+ def test_get_database_version(self):
+ new_connection = connection.copy()
+ new_connection.pg_version = 110009
+ self.assertEqual(new_connection.get_database_version(), (11, 9))
+
+ @mock.patch.object(connection, "get_database_version", return_value=(9, 6))
+ def test_check_database_version_supported(self, mocked_get_database_version):
+ msg = "PostgreSQL 10 or later is required (found 9.6)."
+ with self.assertRaisesMessage(NotSupportedError, msg):
+ connection.check_database_version_supported()
+ self.assertTrue(mocked_get_database_version.called)