updating all tests to use new test class from haystack core

Author: davidsbatista

integrations/astra/tests/test_document_store.py

@@ -11,6 +11,7 @@
 from haystack.document_stores.errors import MissingDocumentError
 from haystack.document_stores.types import DuplicatePolicy
 from haystack.testing.document_store import DocumentStoreBaseTests
+from haystack.testing.document_store import DocumentStoreBaseExtendedTests
 
 from haystack_integrations.document_stores.astra import AstraDocumentStore
 
@@ -48,7 +49,7 @@ def test_to_dict(mock_auth):  # noqa
     os.environ.get("ASTRA_DB_APPLICATION_TOKEN", "") == "", reason="ASTRA_DB_APPLICATION_TOKEN env var not set"
 )
 @pytest.mark.skipif(os.environ.get("ASTRA_DB_API_ENDPOINT", "") == "", reason="ASTRA_DB_API_ENDPOINT env var not set")
-class TestDocumentStore(DocumentStoreBaseTests):
+class TestDocumentStore(DocumentStoreBaseTests, DocumentStoreBaseExtendedTests):
     """
     Common test cases will be provided by `DocumentStoreBaseTests` but
     you can add more to this class.

integrations/azure_ai_search/tests/test_document_store.py

@@ -14,8 +14,8 @@
 from haystack.testing.document_store import (
     CountDocumentsTest,
     DeleteDocumentsTest,
+    DocumentStoreBaseExtendedTests,
     FilterDocumentsTest,
-    UpdateByFilterTest,
     WriteDocumentsTest,
 )
 from haystack.utils.auth import EnvVarSecret, Secret
@@ -257,7 +257,7 @@ def _assert_documents_are_equal(received: list[Document], expected: list[Documen
     not os.environ.get("AZURE_AI_SEARCH_ENDPOINT", None) and not os.environ.get("AZURE_AI_SEARCH_API_KEY", None),
     reason="Missing AZURE_AI_SEARCH_ENDPOINT or AZURE_AI_SEARCH_API_KEY.",
 )
-class TestDocumentStore(CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, UpdateByFilterTest):
+class TestDocumentStore(CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, DocumentStoreBaseExtendedTests):
     def assert_documents_are_equal(self, received: list[Document], expected: list[Document]):
         _assert_documents_are_equal(received, expected)
 

integrations/chroma/tests/test_document_store.py

@@ -16,7 +16,7 @@
     CountDocumentsTest,
     DeleteDocumentsTest,
     FilterDocumentsTest,
-    UpdateByFilterTest,
+    DocumentStoreBaseExtendedTests,
 )
 
 from haystack_integrations.document_stores.chroma import ChromaDocumentStore
@@ -35,7 +35,7 @@ def clear_chroma_system_cache():
     SharedSystemClient.clear_system_cache()
 
 
-class TestDocumentStore(CountDocumentsTest, DeleteDocumentsTest, FilterDocumentsTest, UpdateByFilterTest):
+class TestDocumentStore(CountDocumentsTest, DeleteDocumentsTest, FilterDocumentsTest, DocumentStoreBaseExtendedTests):
     """
     Common test cases will be provided by `DocumentStoreBaseTests` but
     you can add more to this class.

integrations/elasticsearch/tests/test_document_store.py

@@ -10,7 +10,7 @@
 from haystack.dataclasses.document import Document
 from haystack.document_stores.errors import DocumentStoreError, DuplicateDocumentError
 from haystack.document_stores.types import DuplicatePolicy
-from haystack.testing.document_store import DocumentStoreBaseTests
+from haystack.testing.document_store import DocumentStoreBaseTests, DocumentStoreBaseExtendedTests
 from haystack.utils import Secret
 from haystack.utils.auth import TokenSecret
 
@@ -199,7 +199,7 @@ def test_client_initialization_with_api_key_string(_mock_async_es, _mock_es):
 
 
 @pytest.mark.integration
-class TestDocumentStore(DocumentStoreBaseTests):
+class TestDocumentStore(DocumentStoreBaseTests, DocumentStoreBaseExtendedTests):
     """
     Common test cases will be provided by `DocumentStoreBaseTests` but
     you can add more to this class.

integrations/mongodb_atlas/tests/test_document_store.py

@@ -10,7 +10,7 @@
 from haystack.dataclasses.document import ByteStream, Document
 from haystack.document_stores.errors import DuplicateDocumentError
 from haystack.document_stores.types import DuplicatePolicy
-from haystack.testing.document_store import DocumentStoreBaseTests
+from haystack.testing.document_store import DocumentStoreBaseTests, DocumentStoreBaseExtendedTests
 from haystack.utils import Secret
 from pymongo import MongoClient
 from pymongo.driver_info import DriverInfo
@@ -116,7 +116,7 @@ def test_document_conversion_methods_with_custom_field_names(self, _mock_client)
     reason="No MongoDB Atlas connection string provided",
 )
 @pytest.mark.integration
-class TestDocumentStore(DocumentStoreBaseTests):
+class TestDocumentStore(DocumentStoreBaseTests, DocumentStoreBaseExtendedTests):
     @pytest.fixture
     def document_store(self):
         database_name = "haystack_integration_test"

integrations/opensearch/tests/test_document_store.py

@@ -12,7 +12,7 @@
 from haystack.testing.document_store import (
     CountDocumentsTest,
     DeleteDocumentsTest,
-    UpdateByFilterTest,
+    DocumentStoreBaseExtendedTests,
     WriteDocumentsTest,
 )
 from opensearchpy.exceptions import RequestError
@@ -156,7 +156,7 @@ def test_routing_in_delete(mock_bulk, document_store):
 
 
 @pytest.mark.integration
-class TestDocumentStore(CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, UpdateByFilterTest):
+class TestDocumentStore(CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, DocumentStoreBaseExtendedTests):
     """
     Common test cases will be provided by `DocumentStoreBaseTests` but
     you can add more to this class.

integrations/pgvector/tests/test_document_store.py

@@ -12,7 +12,7 @@
 from haystack.testing.document_store import (
     CountDocumentsTest,
     DeleteDocumentsTest,
-    UpdateByFilterTest,
+    DocumentStoreBaseExtendedTests,
     WriteDocumentsTest,
 )
 from haystack.utils import Secret
@@ -21,7 +21,7 @@
 
 
 @pytest.mark.integration
-class TestDocumentStore(CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, UpdateByFilterTest):
+class TestDocumentStore(CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, DocumentStoreBaseExtendedTests):
     def test_write_documents(self, document_store: PgvectorDocumentStore):
         docs = [Document(id="1")]
         assert document_store.write_documents(docs) == 1

integrations/pinecone/tests/test_document_store.py

@@ -14,7 +14,7 @@
 from haystack.testing.document_store import (
     CountDocumentsTest,
     DeleteDocumentsTest,
-    UpdateByFilterTest,
+    DocumentStoreBaseExtendedTests,
     WriteDocumentsTest,
 )
 from haystack.utils import Secret
@@ -265,7 +265,7 @@ def test_serverless_index_creation_from_scratch(delete_sleep_time):
 
 @pytest.mark.integration
 @pytest.mark.skipif(not os.environ.get("PINECONE_API_KEY"), reason="PINECONE_API_KEY not set")
-class TestDocumentStore(CountDocumentsTest, DeleteDocumentsTest, WriteDocumentsTest, UpdateByFilterTest):
+class TestDocumentStore(CountDocumentsTest, DeleteDocumentsTest, WriteDocumentsTest, DocumentStoreBaseExtendedTests):
     def test_write_documents(self, document_store: PineconeDocumentStore):
         docs = [Document(id="1")]
         assert document_store.write_documents(docs) == 1

integrations/qdrant/tests/test_document_store.py

@@ -8,7 +8,7 @@
 from haystack.testing.document_store import (
     CountDocumentsTest,
     DeleteDocumentsTest,
-    UpdateByFilterTest,
+    DocumentStoreBaseExtendedTests,
     WriteDocumentsTest,
     _random_embeddings,
 )
@@ -23,7 +23,7 @@
 )
 
 
-class TestQdrantDocumentStore(CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, UpdateByFilterTest):
+class TestQdrantDocumentStore(CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, DocumentStoreBaseExtendedTests):
     @pytest.fixture
     def document_store(self) -> QdrantDocumentStore:
         return QdrantDocumentStore(

integrations/weaviate/tests/test_document_store.py

@@ -16,7 +16,7 @@
     CountDocumentsTest,
     DeleteDocumentsTest,
     FilterDocumentsTest,
-    UpdateByFilterTest,
+    DocumentStoreBaseExtendedTests,
     WriteDocumentsTest,
     create_filterable_docs,
 )
@@ -49,7 +49,7 @@ def test_init_is_lazy(_mock_client):
 
 @pytest.mark.integration
 class TestWeaviateDocumentStore(
-    CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, FilterDocumentsTest, UpdateByFilterTest
+    CountDocumentsTest, WriteDocumentsTest, DeleteDocumentsTest, FilterDocumentsTest, DocumentStoreBaseExtendedTests
 ):
     @pytest.fixture
     def document_store(self, request) -> WeaviateDocumentStore: