fix: reuse of astra client

srini047 · srini047 · commit f6708013ed0d · 2025-10-06T23:00:33.000+05:30
diff --git a/integrations/astra/tests/test_document_store.py b/integrations/astra/tests/test_document_store.py
@@ -54,7 +54,7 @@ class TestDocumentStore(DocumentStoreBaseTests):
     you can add more to this class.
     """
 
-    @pytest.fixture
+    @pytest.fixture(scope="class")
     def document_store(self) -> AstraDocumentStore:
         return AstraDocumentStore(
             collection_name="haystack_integration",
diff --git a/integrations/astra/tests/test_embedding_retrieval.py b/integrations/astra/tests/test_embedding_retrieval.py
@@ -13,7 +13,7 @@
 )
 @pytest.mark.skipif(os.environ.get("ASTRA_DB_API_ENDPOINT", "") == "", reason="ASTRA_DB_API_ENDPOINT env var not set")
 class TestEmbeddingRetrieval:
-    @pytest.fixture
+    @pytest.fixture(scope="class")
     def document_store(self) -> AstraDocumentStore:
         return AstraDocumentStore(
             collection_name="haystack_integration",

Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,7 @@`
`13`	`13`	`)`
`14`	`14`	`@pytest.mark.skipif(os.environ.get("ASTRA_DB_API_ENDPOINT", "") == "", reason="ASTRA_DB_API_ENDPOINT env var not set")`
`15`	`15`	`class TestEmbeddingRetrieval:`
`16`		`- @pytest.fixture`
	`16`	`+ @pytest.fixture(scope="class")`
`17`	`17`	`def document_store(self) -> AstraDocumentStore:`
`18`	`18`	`return AstraDocumentStore(`
`19`	`19`	`collection_name="haystack_integration",`