add missing fixtures

ragstack-e2e-tests/e2e_tests/langchain/test_unstructured.py

@@ -7,14 +7,35 @@
 from langchain_core.runnables import RunnablePassthrough
 from langchain.chat_models import ChatOpenAI
 
+from e2e_tests.langchain.rag_application import BASIC_QA_PROMPT
+from e2e_tests.test_utils import get_local_resource_path
+
 from e2e_tests.conftest import (
     set_current_test_info,
     get_required_env,
+    get_vector_store_handler,
 )
 
-from e2e_tests.langchain.rag_application import BASIC_QA_PROMPT
-from e2e_tests.test_utils import get_local_resource_path
-from e2e_tests.test_utils.vector_store_handler import VectorStoreTestContext
+from e2e_tests.test_utils.vector_store_handler import (
+    VectorStoreImplementation,
+    VectorStoreTestContext,
+)
+
+
+@pytest.fixture
+def astra_db():
+    handler = get_vector_store_handler(VectorStoreImplementation.ASTRADB)
+    context = handler.before_test()
+    yield context
+    handler.after_test()
+
+
+@pytest.fixture
+def cassandra():
+    handler = get_vector_store_handler(VectorStoreImplementation.CASSANDRA)
+    context = handler.before_test()
+    yield context
+    handler.after_test()
 
 
 @pytest.mark.parametrize("vector_store", ["cassandra", "astra_db"])