Avoided direct mocking of psycopg2.__version__ in test_correct_extraction_psycopg2_version().

This commit is contained in:
Mariusz Felisiak 2022-12-08 14:31:35 +01:00 committed by GitHub
parent cbc0fb3705
commit 9da2210f12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions

View File

@ -291,12 +291,14 @@ class Tests(TestCase):
"::citext", do.lookup_cast(lookup, internal_type=field_type)
)
def test_correct_extraction_psycopg2_version(self):
from django.db.backends.postgresql.base import psycopg2_version
def test_correct_extraction_psycopg_version(self):
from django.db.backends.postgresql.base import Database, psycopg2_version
with mock.patch("psycopg2.__version__", "4.2.1 (dt dec pq3 ext lo64)"):
with mock.patch.object(Database, "__version__", "4.2.1 (dt dec pq3 ext lo64)"):
self.assertEqual(psycopg2_version(), (4, 2, 1))
with mock.patch("psycopg2.__version__", "4.2b0.dev1 (dt dec pq3 ext lo64)"):
with mock.patch.object(
Database, "__version__", "4.2b0.dev1 (dt dec pq3 ext lo64)"
):
self.assertEqual(psycopg2_version(), (4, 2))
@override_settings(DEBUG=True)