Просмотр исходного кода

Remove most errors when running migration tests twice

Andrew Godwin 11 лет назад
Родитель
Сommit
f671fb9d11
1 измененных файлов с 10 добавлено и 0 удалено
  1. 10 0
      tests/migrations/test_operations.py

+ 10 - 0
tests/migrations/test_operations.py

@@ -16,6 +16,16 @@ class OperationTests(MigrationTestBase):
         """
         Creates a test model state and database table.
         """
+        # Delete the tables if they already exist
+        cursor = connection.cursor()
+        try:
+            cursor.execute("DROP TABLE %s_pony" % app_label)
+        except:
+            pass
+        try:
+            cursor.execute("DROP TABLE %s_stable" % app_label)
+        except:
+            pass
         # Make the "current" state
         operations = [migrations.CreateModel(
             "Pony",