support new restore statuses

parkerduckworth · parkerduckworth · commit a486fef9972f · 2026-01-26T15:26:31.000-05:00
diff --git a/integration/test_backup_v4.py b/integration/test_backup_v4.py
@@ -225,6 +225,7 @@ def test_create_and_restore_backup_without_waiting(
             BackupStatus.TRANSFERRED,
             BackupStatus.TRANSFERRING,
             BackupStatus.STARTED,
+            BackupStatus.FINALIZING,
         ]
         if restore_status.status == BackupStatus.SUCCESS:
             break
diff --git a/weaviate/backup/backup.py b/weaviate/backup/backup.py
@@ -41,6 +41,8 @@ class BackupStatus(str, Enum):
     STARTED = "STARTED"
     TRANSFERRING = "TRANSFERRING"
     TRANSFERRED = "TRANSFERRED"
+    CANCELLING = "CANCELLING"
+    FINALIZING = "FINALIZING"
     SUCCESS = "SUCCESS"
     FAILED = "FAILED"
     CANCELED = "CANCELED"

Original file line number	Diff line number	Diff line change
`@@ -225,6 +225,7 @@ def test_create_and_restore_backup_without_waiting(`
`225`	`225`	`BackupStatus.TRANSFERRED,`
`226`	`226`	`BackupStatus.TRANSFERRING,`
`227`	`227`	`BackupStatus.STARTED,`
	`228`	`+ BackupStatus.FINALIZING,`
`228`	`229`	`]`
`229`	`230`	`if restore_status.status == BackupStatus.SUCCESS:`
`230`	`231`	`break`