diff options
| author | Jacob Walls <jacobtylerwalls@gmail.com> | 2024-05-25 17:17:15 -0400 |
|---|---|---|
| committer | Sarah Boyce <42296566+sarahboyce@users.noreply.github.com> | 2024-05-28 12:44:07 +0200 |
| commit | 99f23eaabd8da653f046dc1d19f5008c030a4f79 (patch) | |
| tree | d19eaf625c29a846294a016dff8f288e48f3e75f /tests/backends | |
| parent | d3a7ed5bcc45000a6c3dd55d85a4caaa83299f83 (diff) | |
Fixed #35469 -- Removed deferred SQL to create index removed by AlterField operation.
Diffstat (limited to 'tests/backends')
| -rw-r--r-- | tests/backends/test_ddl_references.py | 31 |
1 files changed, 24 insertions, 7 deletions
diff --git a/tests/backends/test_ddl_references.py b/tests/backends/test_ddl_references.py index 86984ed3e8..8975b97124 100644 --- a/tests/backends/test_ddl_references.py +++ b/tests/backends/test_ddl_references.py @@ -166,10 +166,13 @@ class ForeignKeyNameTests(IndexNameTests): class MockReference: - def __init__(self, representation, referenced_tables, referenced_columns): + def __init__( + self, representation, referenced_tables, referenced_columns, referenced_indexes + ): self.representation = representation self.referenced_tables = referenced_tables self.referenced_columns = referenced_columns + self.referenced_indexes = referenced_indexes def references_table(self, table): return table in self.referenced_tables @@ -177,6 +180,9 @@ class MockReference: def references_column(self, table, column): return (table, column) in self.referenced_columns + def references_index(self, table, index): + return (table, index) in self.referenced_indexes + def rename_table_references(self, old_table, new_table): if old_table in self.referenced_tables: self.referenced_tables.remove(old_table) @@ -195,32 +201,43 @@ class MockReference: class StatementTests(SimpleTestCase): def test_references_table(self): statement = Statement( - "", reference=MockReference("", {"table"}, {}), non_reference="" + "", reference=MockReference("", {"table"}, {}, {}), non_reference="" ) self.assertIs(statement.references_table("table"), True) self.assertIs(statement.references_table("other"), False) def test_references_column(self): statement = Statement( - "", reference=MockReference("", {}, {("table", "column")}), non_reference="" + "", + reference=MockReference("", {}, {("table", "column")}, {}), + non_reference="", ) self.assertIs(statement.references_column("table", "column"), True) self.assertIs(statement.references_column("other", "column"), False) + def test_references_index(self): + statement = Statement( + "", + reference=MockReference("", {}, {}, {("table", "index")}), + non_reference="", + ) + self.assertIs(statement.references_index("table", "index"), True) + self.assertIs(statement.references_index("other", "index"), False) + def test_rename_table_references(self): - reference = MockReference("", {"table"}, {}) + reference = MockReference("", {"table"}, {}, {}) statement = Statement("", reference=reference, non_reference="") statement.rename_table_references("table", "other") self.assertEqual(reference.referenced_tables, {"other"}) def test_rename_column_references(self): - reference = MockReference("", {}, {("table", "column")}) + reference = MockReference("", {}, {("table", "column")}, {}) statement = Statement("", reference=reference, non_reference="") statement.rename_column_references("table", "column", "other") self.assertEqual(reference.referenced_columns, {("table", "other")}) def test_repr(self): - reference = MockReference("reference", {}, {}) + reference = MockReference("reference", {}, {}, {}) statement = Statement( "%(reference)s - %(non_reference)s", reference=reference, @@ -229,7 +246,7 @@ class StatementTests(SimpleTestCase): self.assertEqual(repr(statement), "<Statement 'reference - non_reference'>") def test_str(self): - reference = MockReference("reference", {}, {}) + reference = MockReference("reference", {}, {}, {}) statement = Statement( "%(reference)s - %(non_reference)s", reference=reference, |