Stylistic cleanup of Postgres autocommit tests

Cleaned up tests introduced in f572ee0c65.

Thanks to Claude Paroz for suggesting the changes.
This commit is contained in:
Anssi Kääriäinen 2012-07-02 15:35:59 +03:00
parent 784d0c261c
commit 925a6936b9
1 changed files with 5 additions and 8 deletions

View File

@ -175,6 +175,8 @@ class TestTransactionClosing(TransactionTestCase):
self.test_failing_query_transaction_closed()
@skipUnless(connection.vendor == 'postgresql',
"This test only valid for PostgreSQL")
class TestPostgresAutocommit(TransactionTestCase):
"""
Tests to make sure psycopg2's autocommit mode is restored after entering
@ -196,6 +198,9 @@ class TestPostgresAutocommit(TransactionTestCase):
new_backend = self._old_backend.__class__(settings, DEFAULT_DB_ALIAS)
connections[DEFAULT_DB_ALIAS] = new_backend
def tearDown(self):
connections[DEFAULT_DB_ALIAS] = self._old_backend
def test_initial_autocommit_state(self):
self.assertTrue(connection.features.uses_autocommit)
self.assertEqual(connection.isolation_level, self._autocommit)
@ -222,14 +227,6 @@ class TestPostgresAutocommit(TransactionTestCase):
transaction.leave_transaction_management()
self.assertEqual(connection.isolation_level, self._autocommit)
def tearDown(self):
connections[DEFAULT_DB_ALIAS] = self._old_backend
TestPostgresAutocommit = skipUnless(connection.vendor == 'postgresql',
"This test only valid for PostgreSQL")(TestPostgresAutocommit)
TestPostgresAutoCommit = skipUnlessDBFeature('supports_transactions')(
TestPostgresAutocommit)
class TestManyToManyAddTransaction(TransactionTestCase):
def test_manyrelated_add_commit(self):