fix: review comments

Author: srini047

integrations/astra/src/haystack_integrations/document_stores/astra/document_store.py

@@ -5,7 +5,7 @@
 
 from haystack import default_from_dict, default_to_dict, logging
 from haystack.dataclasses import Document
-from haystack.document_stores.errors import DuplicateDocumentError, MissingDocumentError
+from haystack.document_stores.errors import DocumentStoreError, DuplicateDocumentError, MissingDocumentError
 from haystack.document_stores.types import DuplicatePolicy
 from haystack.utils import Secret, deserialize_secrets_inplace
 
@@ -424,7 +424,12 @@ def delete_all_documents(self) -> None:
         Deletes all documents from the document store.
         """
         deletion_counter = 0
-        deletion_counter = self.index.delete_all_documents()
+
+        try:
+            deletion_counter = self.index.delete_all_documents()
+        except Exception as e:
+            msg = f"Failed to delete all documents from Astra: {e!s}"
+            raise DocumentStoreError(msg) from e
 
         if deletion_counter == -1:
             logger.info("All documents deleted")

integrations/astra/tests/test_document_store.py

@@ -204,7 +204,7 @@ def test_filter_documents_by_in_operator(self, document_store):
         self.assert_documents_are_equal([result[0]], [docs[0]])
         self.assert_documents_are_equal([result[1]], [docs[1]])
 
-    def test_delete_all_documents_empty_store(self, document_store: AstraDocumentStore):
+    def test_delete_all_documents(self, document_store: AstraDocumentStore):
         """
         Test delete_all_documents() on an Astra.
         """