Fix some bad test running under PostgreSQL

This commit is contained in:
Andrew Godwin 2013-07-02 11:51:38 +01:00
parent f343cbf06c
commit 9ef715d256
1 changed files with 4 additions and 2 deletions

View File

@ -1,5 +1,6 @@
from django.test import TestCase from django.test import TestCase
from django.db import connection, models, migrations from django.db import connection, models, migrations
from django.db.transaction import atomic
from django.db.utils import IntegrityError from django.db.utils import IntegrityError
from django.db.migrations.state import ProjectState from django.db.migrations.state import ProjectState
@ -38,7 +39,7 @@ class OperationTests(TestCase):
"Pony", "Pony",
[ [
("id", models.AutoField(primary_key=True)), ("id", models.AutoField(primary_key=True)),
("pink", models.BooleanField(default=True)), ("pink", models.IntegerField(default=3)),
("weight", models.FloatField()), ("weight", models.FloatField()),
], ],
) )
@ -232,6 +233,7 @@ class OperationTests(TestCase):
operation.database_forwards("test_alunto", editor, project_state, new_state) operation.database_forwards("test_alunto", editor, project_state, new_state)
cursor.execute("INSERT INTO test_alunto_pony (id, pink, weight) VALUES (1, 1, 1)") cursor.execute("INSERT INTO test_alunto_pony (id, pink, weight) VALUES (1, 1, 1)")
with self.assertRaises(IntegrityError): with self.assertRaises(IntegrityError):
with atomic():
cursor.execute("INSERT INTO test_alunto_pony (id, pink, weight) VALUES (2, 1, 1)") cursor.execute("INSERT INTO test_alunto_pony (id, pink, weight) VALUES (2, 1, 1)")
cursor.execute("DELETE FROM test_alunto_pony") cursor.execute("DELETE FROM test_alunto_pony")
# And test reversal # And test reversal