mirror of https://github.com/django/django.git
Fix some bad test running under PostgreSQL
This commit is contained in:
parent
f343cbf06c
commit
9ef715d256
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue