test0908
/
django
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/migrations/test_operations.py
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,7 +233,8 @@ 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):
cursor.execute("INSERT INTO test_alunto_pony (id, pink, weight) VALUES (2, 1, 1)") with atomic():
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
with connection.schema_editor() as editor: with connection.schema_editor() as editor: