summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAymeric Augustin <aymeric.augustin@m4x.org>2014-01-12 18:26:20 +0100
committerAymeric Augustin <aymeric.augustin@m4x.org>2014-01-12 18:28:51 +0100
commit1afe7488322f9b53ea00f47fb07c4374a5f2150e (patch)
treefa4a38370fbac2cf464bd685b5d546e6d766db2c
parent1c24096f1e860960fa44df937713528065b5f79a (diff)
Fixed #21452 -- Non-autocommit connections to PostgreSQL.
When settings.DATABASES['default']['AUTOCOMMIT'] = False, the connection wasn't in autocommit mode but Django pretended it was. Thanks Anssi for analysing this issue. Refs #17062.
-rw-r--r--django/db/backends/postgresql_psycopg2/base.py4
-rw-r--r--tests/backends/tests.py23
2 files changed, 21 insertions, 6 deletions
diff --git a/django/db/backends/postgresql_psycopg2/base.py b/django/db/backends/postgresql_psycopg2/base.py
index 91e16a1b76..c7fd98ae98 100644
--- a/django/db/backends/postgresql_psycopg2/base.py
+++ b/django/db/backends/postgresql_psycopg2/base.py
@@ -142,11 +142,11 @@ class DatabaseWrapper(BaseDatabaseWrapper):
conn_tz = get_parameter_status('TimeZone')
if conn_tz != tz:
- # Set the time zone in autocommit mode (see #17062)
- self.set_autocommit(True)
self.connection.cursor().execute(
self.ops.set_time_zone_sql(), [tz]
)
+ # Commit after setting the time zone (see #17062)
+ self.connection.commit()
self.connection.set_isolation_level(self.isolation_level)
def create_cursor(self):
diff --git a/tests/backends/tests.py b/tests/backends/tests.py
index 1b15b9a316..6aeea34860 100644
--- a/tests/backends/tests.py
+++ b/tests/backends/tests.py
@@ -373,10 +373,25 @@ class PostgresNewConnectionTests(TestCase):
tz = cursor.fetchone()[0]
self.assertEqual(new_tz, tz)
finally:
- try:
- new_connection.close()
- except DatabaseError:
- pass
+ new_connection.close()
+
+ @unittest.skipUnless(
+ connection.vendor == 'postgresql',
+ "This test applies only to PostgreSQL")
+ def test_connect_non_autocommit(self):
+ """
+ The connection wrapper shouldn't believe that autocommit is enabled
+ after setting the time zone when AUTOCOMMIT is False (#21452).
+ """
+ databases = copy.deepcopy(settings.DATABASES)
+ new_connections = ConnectionHandler(databases)
+ new_connection = new_connections[DEFAULT_DB_ALIAS]
+ try:
+ new_connection.settings_dict['AUTOCOMMIT'] = False
+ cursor = new_connection.cursor()
+ self.assertFalse(new_connection.get_autocommit())
+ finally:
+ new_connection.close()
# This test needs to run outside of a transaction, otherwise closing the