Updated tests for Django 1.8.

tomatohater · Jul 23, 2015 · 06ce0f5 · 06ce0f5
1 parent e493c48
commit 06ce0f5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pgcryptoauth/tests.py b/pgcryptoauth/tests.py
@@ -18,9 +18,9 @@ class PgCryptoAuthTests(TestCase):
     encoded = u'pgcrypto$%s' % pghash
 
     def setUp(self):
-        cursor = connection.cursor()
-        cursor.execute("CREATE EXTENSION pgcrypto")
-        transaction.commit_unless_managed()
+        with transaction.atomic():
+            cursor = connection.cursor()
+            cursor.execute("CREATE EXTENSION pgcrypto")
         self.user = User.objects.create_user(username=self.username)
         self.user.password = self.encoded
         self.user.save()