Fixed #1371: Keep the unique flag in DropIndexOp

sqlalchemy · Dec 1, 2023 · 5d6616c · 5d6616c
1 parent e8f6a85
commit 5d6616c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/alembic/operations/ops.py b/alembic/operations/ops.py
@@ -1054,6 +1054,7 @@ def from_index(cls, index: Index) -> DropIndexOp:
             table_name=index.table.name,
             schema=index.table.schema,
             _reverse=CreateIndexOp.from_index(index),
+            unique=index.unique,
             **index.kwargs,
         )
 

diff --git a/tests/test_autogen_diffs.py b/tests/test_autogen_diffs.py
@@ -1783,6 +1783,7 @@ def setUp(self):
             Column("x", Integer()),
         )
         self.ix = Index("ix1", t.c.id)
+        self.ix_unique = Index("ix2", t.c.id, unique=True)
         fk = ForeignKeyConstraint(["t_id"], ["t.id"])
         q = Table("q", m, Column("t_id", Integer()), fk)
         self.table = t
@@ -1905,6 +1906,11 @@ def test_create_index(self):
         eq_(op.to_index(), schemacompare.CompareIndex(self.ix))
         eq_(op.reverse().to_index(), schemacompare.CompareIndex(self.ix))
 
+    def test_create_unique_index(self):
+        op = ops.CreateIndexOp.from_index(self.ix_unique)
+        eq_(op.to_index(), schemacompare.CompareIndex(self.ix_unique))
+        eq_(op.reverse().to_index(), schemacompare.CompareIndex(self.ix_unique))
+
 
 class MultipleMetaDataTest(AutogenFixtureTest, TestBase):
     def test_multiple(self):