File tree Expand file tree Collapse file tree 2 files changed +14
-2
lines changed
piccolo/apps/migrations/auto
tests/apps/migrations/auto Expand file tree Collapse file tree 2 files changed +14
-2
lines changed Original file line number Diff line number Diff line change @@ -241,7 +241,7 @@ def drop_tables(self) -> AlterStatements:
241241
242242 return AlterStatements (
243243 statements = [
244- f"manager.drop_table(tablename='{ i .tablename } ')"
244+ f"manager.drop_table(class_name=' { i . class_name } ', tablename='{ i .tablename } ')" # noqa
245245 for i in drop_tables
246246 ]
247247 )
Original file line number Diff line number Diff line change @@ -20,7 +20,19 @@ def test_drop_table(self):
2020 """
2121 Test dropping an existing table.
2222 """
23- pass
23+ schema : t .List [DiffableTable ] = []
24+ schema_snapshot : t .List [DiffableTable ] = [
25+ DiffableTable (class_name = "Band" , tablename = "band" , columns = [])
26+ ]
27+ schema_differ = SchemaDiffer (
28+ schema = schema , schema_snapshot = schema_snapshot , auto_input = "y"
29+ )
30+
31+ self .assertTrue (len (schema_differ .drop_tables .statements ) == 1 )
32+ self .assertEqual (
33+ schema_differ .drop_tables .statements [0 ],
34+ "manager.drop_table(class_name='Band', tablename='band')" ,
35+ )
2436
2537 def test_rename_table (self ):
2638 """
You can’t perform that action at this time.
0 commit comments