Skip to content

Commit 3cb6e9a

Browse files
author
Andrei Neagu
committed
extended tests
1 parent f626572 commit 3cb6e9a

File tree

1 file changed

+22
-1
lines changed

1 file changed

+22
-1
lines changed

services/dynamic-scheduler/tests/unit/api_rpc/test_api_rpc__services.py

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
from models_library.users import UserID
2121
from pydantic import TypeAdapter
2222
from pytest_mock import MockerFixture
23+
from pytest_simcore.helpers.monkeypatch_envs import setenvs_from_dict
2324
from pytest_simcore.helpers.typing_env import EnvVarsDict
2425
from servicelib.rabbitmq import RabbitMQRPCClient, RPCServerError
2526
from servicelib.rabbitmq.rpc_interfaces.dynamic_scheduler import services
@@ -29,6 +30,7 @@
2930
)
3031
from settings_library.rabbit import RabbitSettings
3132
from settings_library.redis import RedisSettings
33+
from simcore_service_dynamic_scheduler.core.settings import SchedulingMode
3234

3335
pytest_simcore_core_services_selection = [
3436
"redis",
@@ -133,13 +135,32 @@ def mock_director_v2_service_state(
133135
yield None
134136

135137

138+
@pytest.fixture(
139+
params=[
140+
SchedulingMode.VIA_DIRECTOR_V2,
141+
# NOTE: enable below when INTERNAL scheduler is impelmented
142+
# SchedulingMode.INTERNAL,
143+
]
144+
)
145+
def scheduling_mode(request: pytest.FixtureRequest) -> SchedulingMode:
146+
return request.param
147+
148+
136149
@pytest.fixture
137150
def app_environment(
151+
monkeypatch: pytest.MonkeyPatch,
138152
app_environment: EnvVarsDict,
139153
rabbit_service: RabbitSettings,
140154
redis_service: RedisSettings,
155+
scheduling_mode: SchedulingMode,
141156
) -> EnvVarsDict:
142-
return app_environment
157+
return setenvs_from_dict(
158+
monkeypatch,
159+
{
160+
**app_environment,
161+
"DYNAMIC_SCHEDULER_SCHEDULING_MODE": scheduling_mode,
162+
},
163+
)
143164

144165

145166
@pytest.fixture

0 commit comments

Comments
 (0)