Commit c37c6dc4 authored by Carl Meyer's avatar Carl Meyer
Browse files

[1.8.x] Improved isolation of applied-migrations table in migration tests.

Backport of 335fc44f from master.
parent de62c774
Loading
Loading
Loading
Loading
+6 −0
Original line number Diff line number Diff line
import os

from django.db import connection
from django.db.migrations.recorder import MigrationRecorder
from django.test import TransactionTestCase
from django.utils._os import upath

@@ -13,6 +14,11 @@ class MigrationTestBase(TransactionTestCase):
    available_apps = ["migrations"]
    test_dir = os.path.abspath(os.path.dirname(upath(__file__)))

    def tearDown(self):
        # Reset applied-migrations state.
        recorder = MigrationRecorder(connection)
        recorder.migration_qs.filter(app='migrations').delete()

    def get_table_description(self, table):
        with connection.cursor() as cursor:
            return connection.introspection.get_table_description(cursor, table)