Commit 925a6936 authored by Anssi Kääriäinen's avatar Anssi Kääriäinen
Browse files

Stylistic cleanup of Postgres autocommit tests 

Cleaned up tests introduced in f572ee0c.

Thanks to Claude Paroz for suggesting the changes.
parent 784d0c26

tests/regressiontests/transactions_regress/tests.py

+5 −8
Original line number Diff line number Diff line
@@ -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):