updated tests

franciscojavierarceo · franciscojavierarceo · commit 4c16c0b72aa6 · 2024-12-21T21:19:29.000-05:00
Signed-off-by: Francisco Javier Arceo &lt;farceo@redhat.com&gt;
diff --git a/Makefile b/Makefile
@@ -356,10 +356,10 @@ test-python-universal-elasticsearch-online:
 test-python-universal-milvus-online:
 	PYTHONPATH='.' \
 		FULL_REPO_CONFIGS_MODULE=sdk.python.feast.infra.online_stores.milvus_online_store.milvus_repo_configuration \
-		PYTEST_PLUGINS=sdk.python.tests.integration.feature_repos.universal.online_store.milvus\
+		PYTEST_PLUGINS=sdk.python.tests.integration.feature_repos.universal.online_store.milvus \
 		python -m pytest -n 8 --integration \
-		-k "test_retrieve_online_documents and not test_dqm" \
- 			sdk/python/tests
+		-k "test_retrieve_online_documents" \
+ 			sdk/python/tests --ignore=sdk/python/tests/integration/offline_store/test_dqm_validation.py
 
 test-python-universal-singlestore-online:
 	PYTHONPATH='.' \
diff --git a/sdk/python/tests/conftest.py b/sdk/python/tests/conftest.py
@@ -191,6 +191,7 @@ def environment(request, worker_id):
         request.param,
         worker_id=worker_id,
         fixture_request=request,
+        entity_key_serialization_version=3,
     )
 
     e.setup()
@@ -204,9 +205,20 @@ def environment(request, worker_id):
     e.teardown()
 
 
+from tests.integration.feature_repos.integration_test_repo_config import (
+    IntegrationTestRepoConfig,
+)
+from tests.integration.feature_repos.universal.online_store.milvus import MilvusOnlineStoreCreator
+
 @pytest.fixture
 def vectordb_environment(request, worker_id):
+    milvus_config = IntegrationTestRepoConfig(
+        provider="milvus",
+        online_store_creator=MilvusOnlineStoreCreator,
+        offline_store_creator='',
+        batch_engine={})
     e = construct_test_environment(
+        milvus_config,
         request.param,
         worker_id=worker_id,
         fixture_request=request,
