0
0
mirror of https://github.com/django/django.git synced 2024-11-28 21:43:13 +01:00
django/tests/migrations/test_migrations_atomic_operation/0001_initial.py
2022-02-07 20:37:05 +01:00

23 lines
639 B
Python

from django.db import migrations, models
def raise_error(apps, schema_editor):
# Test atomic operation in non-atomic migration is wrapped in transaction
Editor = apps.get_model("migrations", "Editor")
Editor.objects.create(name="Test Editor")
raise RuntimeError("Abort migration")
class Migration(migrations.Migration):
atomic = False
operations = [
migrations.CreateModel(
"Editor",
[
("name", models.CharField(primary_key=True, max_length=255)),
],
),
migrations.RunPython(raise_error, reverse_code=raise_error, atomic=True),
]