Skip to content

Commit ca946b7

Browse files
committed
added a test for SchemaDiffer.new_table_columns
1 parent 9d5206a commit ca946b7

File tree

1 file changed

+25
-1
lines changed

1 file changed

+25
-1
lines changed

tests/apps/migrations/auto/test_schema_differ.py

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,31 @@ def test_add_table(self):
1414
"""
1515
Test adding a new table.
1616
"""
17-
pass
17+
name_column = Varchar()
18+
name_column._meta.name = "name"
19+
schema: t.List[DiffableTable] = [
20+
DiffableTable(
21+
class_name="Band", tablename="band", columns=[name_column]
22+
)
23+
]
24+
schema_snapshot: t.List[DiffableTable] = []
25+
schema_differ = SchemaDiffer(
26+
schema=schema, schema_snapshot=schema_snapshot, auto_input="y"
27+
)
28+
29+
create_tables = schema_differ.create_tables
30+
self.assertTrue(len(create_tables.statements) == 1)
31+
self.assertEqual(
32+
create_tables.statements[0],
33+
"manager.add_table('Band', tablename='band')",
34+
)
35+
36+
new_table_columns = schema_differ.new_table_columns
37+
self.assertTrue(len(new_table_columns.statements) == 1)
38+
self.assertEqual(
39+
new_table_columns.statements[0],
40+
"manager.add_column(table_class_name='Band', tablename='band', column_name='name', column_class_name='Varchar', column_class=Varchar, params={'length': 255, 'default': '', 'null': False, 'primary': False, 'key': False, 'unique': False, 'index': False})", # noqa
41+
)
1842

1943
def test_drop_table(self):
2044
"""

0 commit comments

Comments
 (0)