From d292d03ec98074b4d5dcb339da3090ab428e8548 Mon Sep 17 00:00:00 2001 From: The Dude <38289341+kartik4949@users.noreply.github.com> Date: Tue, 9 Jan 2024 15:52:12 +0530 Subject: [PATCH] Add env injection in test suite (#1657) --- test/conftest.py | 1 + test/integration/conftest.py | 9 +++++++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/test/conftest.py b/test/conftest.py index 762c54ff92..330d173841 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -104,6 +104,7 @@ def test_db(monkeypatch, request) -> Iterator[Datalayer]: # mongodb instead of localhost is required for CFG compatibility with docker-host db_name = "test_db" data_backend = f'mongodb://superduper:superduper@mongodb:27017/{db_name}' + data_backend = os.environ.get('SUPERDUPER_MONGO_URI', data_backend) monkeypatch.setattr(CFG, 'data_backend', data_backend) diff --git a/test/integration/conftest.py b/test/integration/conftest.py index fbb46d9e8b..908220ba71 100644 --- a/test/integration/conftest.py +++ b/test/integration/conftest.py @@ -1,3 +1,4 @@ +import os import random from pathlib import Path @@ -111,11 +112,14 @@ def dask_client(monkeypatch, request): db_name = "test_db" data_backend = f'mongodb://superduper:superduper@localhost:27017/{db_name}' + data_backend = os.environ.get('SUPERDUPER_MONGO_URI', data_backend) + address = os.environ.get('SUPERDUPER_DASK_URI', 'tcp://localhost:8786') + monkeypatch.setenv('SUPERDUPERDB_DATA_BACKEND', data_backend) # Change the default value client = DaskComputeBackend( - address='tcp://localhost:8786', + address=address, local=False, ) @@ -130,9 +134,10 @@ def ray_client(): from superduperdb.backends.ray.compute import RayComputeBackend working_dir = Path(__file__).parents[1] + address = os.environ.get('SUPERDUPER_RAY_URI', 'ray://127.0.0.1:10001') client = RayComputeBackend( - address='ray://127.0.0.1:10001', + address=address, runtime_env={"working_dir": working_dir, 'excludes': ['unittest']}, )