Add test

deepset-ai · Mar 18, 2024 · 98839da · 98839da
1 parent f4459db
commit 98839da
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/integrations/opensearch/tests/test_document_store.py b/integrations/opensearch/tests/test_document_store.py
@@ -324,3 +324,12 @@ def test_write_documents_different_embedding_sizes_fail(
 
         with pytest.raises(DocumentStoreError):
             document_store_embedding_dim_4.write_documents(docs)
+
+    @patch("haystack_integrations.document_stores.opensearch.document_store.bulk")
+    def test_write_documents_with_badly_formatted_bulk_errors(self, mock_bulk, document_store):
+        error = {"some_key": "some_value"}
+        mock_bulk.return_value = ([], [error])
+
+        with pytest.raises(DocumentStoreError) as e:
+            document_store.write_documents([Document(content="Hello world")])
+            e.match(f"{error}")