Add CANCELED status and include it in the test.

weaviate · Sep 13, 2024 · 1ed127a · 1ed127a
1 parent 2a00148
commit 1ed127a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/integration_v3/test_backup_v4.py b/integration_v3/test_backup_v4.py
@@ -476,7 +476,9 @@ def test_cancel_backup(client: weaviate.WeaviateClient) -> None:
 
     with pytest.raises(UnexpectedStatusCodeError):
         assert client.backup.cancel_backup(backup_id=backup_id, backend=BACKEND)
-
+
+    status_resp = client.backup.get_create_status(backup_id=backup_id, backend=BACKEND)
+    assert status_resp.status == BackupStatus.CANCELED
     _ = client.backup.list_backups(backend=BACKEND)
     # depends on the final implementation
     # assert len(backups) == 0
diff --git a/weaviate/backup/backup.py b/weaviate/backup/backup.py
@@ -55,6 +55,7 @@ class BackupStatus(str, Enum):
     TRANSFERRED = "TRANSFERRED"
     SUCCESS = "SUCCESS"
     FAILED = "FAILED"
+    CANCELED = "CANCELED"
 
 
 class _BackupConfigBase(BaseModel):