test: Fix the failing test for Azure AI (#2579)

* Add sleep to make the calls seaparate

* docs: update changelog for integrations/stackit (#2576)

* Update changelog for integrations/stackit

* Update CHANGELOG.md

---------

Co-authored-by: Amnah199 <[email protected]>
Co-authored-by: Amna Mubashar <[email protected]>

* chore: Weave - move type ignore comment (#2578)

* Update tests

* Update CHANGELOG.md

---------

Co-authored-by: Haystack Bot <[email protected]>
Co-authored-by: Amnah199 <[email protected]>
Co-authored-by: Stefano Fiorucci <[email protected]>

Author: 3 people

integrations/azure_ai_search/tests/conftest.py

@@ -48,6 +48,7 @@ def document_store(request):
     # Override some methods to wait for the documents to be available
     original_write_documents = store.write_documents
     original_delete_documents = store.delete_documents
+    original_delete_all_documents = store.delete_all_documents
 
     def write_documents_and_wait(documents, policy=DuplicatePolicy.OVERWRITE):
         written_docs = original_write_documents(documents, policy)
@@ -58,6 +59,10 @@ def delete_documents_and_wait(filters):
         original_delete_documents(filters)
         time.sleep(SLEEP_TIME_IN_SECONDS)
 
+    def delete_all_documents_and_wait():
+        original_delete_all_documents()
+        time.sleep(SLEEP_TIME_IN_SECONDS)
+
     # Helper function to wait for the index to be deleted, needed to cover latency
     def wait_for_index_deletion(client, index_name):
         start_time = time.time()
@@ -69,6 +74,7 @@ def wait_for_index_deletion(client, index_name):
 
     store.write_documents = write_documents_and_wait
     store.delete_documents = delete_documents_and_wait
+    store.delete_all_documents = delete_all_documents_and_wait
 
     yield store
     try:

integrations/azure_ai_search/tests/test_document_store.py

@@ -294,7 +294,6 @@ def test_delete_all_documents(self, document_store: AzureAISearchDocumentStore):
         docs = [Document(content="first doc"), Document(content="second doc")]
         document_store.write_documents(docs)
         assert document_store.count_documents() == 2
-
         document_store.delete_all_documents()
         assert document_store.count_documents() == 0