[1.10.x] Fixed a backends test with psycopg2 2.7.

Backport of 49a63d08d3 from master
This commit is contained in:
Tim Graham 2017-03-01 13:14:35 -05:00
parent fac22535ab
commit ead5b14d8c
1 changed files with 3 additions and 1 deletions

View File

@ -292,6 +292,7 @@ class PostgreSQLTests(TestCase):
"""
Regression test for #18130 and #24318.
"""
import psycopg2
from psycopg2.extensions import (
ISOLATION_LEVEL_READ_COMMITTED as read_committed,
ISOLATION_LEVEL_SERIALIZABLE as serializable,
@ -302,7 +303,8 @@ class PostgreSQLTests(TestCase):
# PostgreSQL is configured with the default isolation level.
# Check the level on the psycopg2 connection, not the Django wrapper.
self.assertEqual(connection.connection.isolation_level, read_committed)
default_level = read_committed if psycopg2.__version__ < '2.7' else None
self.assertEqual(connection.connection.isolation_level, default_level)
new_connection = connection.copy()
new_connection.settings_dict['OPTIONS']['isolation_level'] = serializable