Skip to content

Commit e9be2ee

Browse files
author
Andrei Neagu
committed
rename
1 parent 5b67d46 commit e9be2ee

File tree

15 files changed

+28
-24
lines changed

15 files changed

+28
-24
lines changed

packages/models-library/src/models_library/services_types.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ class ServiceRunID(str):
6161
__slots__ = ()
6262

6363
@classmethod
64-
def create_for_dynamic_sidecar(cls) -> Self:
64+
def get_resource_tracking_run_id_for_dynamic(cls) -> Self:
6565
"""used for dynamic services"""
6666
# NOTE: there was a legacy version of this RunID
6767
# legacy version:
@@ -73,7 +73,7 @@ def create_for_dynamic_sidecar(cls) -> Self:
7373
return cls(run_id_format)
7474

7575
@classmethod
76-
def get_resource_tracking_run_id(
76+
def get_resource_tracking_run_id_for_computational(
7777
cls,
7878
user_id: UserID,
7979
project_id: "ProjectID",

packages/models-library/tests/test_services_types.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,12 +25,16 @@ def test_run_id_get_resource_tracking_run_id(
2525
iteration: PositiveInt,
2626
expected_result: str,
2727
):
28-
resource_tracking_service_run_id = ServiceRunID.get_resource_tracking_run_id(
29-
user_id, project_id, node_id, iteration
28+
resource_tracking_service_run_id = (
29+
ServiceRunID.get_resource_tracking_run_id_for_computational(
30+
user_id, project_id, node_id, iteration
31+
)
3032
)
3133
assert isinstance(resource_tracking_service_run_id, ServiceRunID)
3234
assert resource_tracking_service_run_id == expected_result
3335

3436

35-
def test_run_id_create_for_dynamic_sidecar():
36-
assert isinstance(ServiceRunID.create_for_dynamic_sidecar(), ServiceRunID)
37+
def test_get_resource_tracking_run_id_for_dynamic():
38+
assert isinstance(
39+
ServiceRunID.get_resource_tracking_run_id_for_dynamic(), ServiceRunID
40+
)

services/agent/tests/unit/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def test_client(initialized_app: FastAPI) -> TestClient:
5757

5858
@pytest.fixture
5959
def service_run_id() -> ServiceRunID:
60-
return ServiceRunID.create_for_dynamic_sidecar()
60+
return ServiceRunID.get_resource_tracking_run_id_for_dynamic()
6161

6262

6363
@pytest.fixture

services/director-v2/src/simcore_service_director_v2/models/dynamic_services_scheduler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -377,7 +377,7 @@ class SchedulerData(CommonServiceDetails, DynamicSidecarServiceLabels):
377377
description="Name of the current dynamic-sidecar being observed",
378378
)
379379
run_id: ServiceRunID = Field(
380-
default_factory=ServiceRunID.create_for_dynamic_sidecar,
380+
default_factory=ServiceRunID.get_resource_tracking_run_id_for_dynamic,
381381
description=(
382382
"Uniquely identify the dynamic sidecar session (a.k.a. 2 "
383383
"subsequent exact same services will have a different run_id)"

services/director-v2/src/simcore_service_director_v2/modules/comp_scheduler/_scheduler_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -295,7 +295,7 @@ def _need_heartbeat(task: CompTaskAtDB) -> bool:
295295
*(
296296
publish_service_resource_tracking_heartbeat(
297297
self.rabbitmq_client,
298-
ServiceRunID.get_resource_tracking_run_id(
298+
ServiceRunID.get_resource_tracking_run_id_for_computational(
299299
user_id, t.project_id, t.node_id, iteration
300300
),
301301
)
@@ -348,7 +348,7 @@ async def _process_started_tasks(
348348
*(
349349
publish_service_resource_tracking_started(
350350
self.rabbitmq_client,
351-
service_run_id=ServiceRunID.get_resource_tracking_run_id(
351+
service_run_id=ServiceRunID.get_resource_tracking_run_id_for_computational(
352352
user_id, t.project_id, t.node_id, iteration
353353
),
354354
wallet_id=run_metadata.get("wallet_id"),

services/director-v2/src/simcore_service_director_v2/modules/comp_scheduler/_scheduler_dask.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ async def _start_tasks(
129129
hardware_info=task.hardware_info,
130130
callback=wake_up_callback,
131131
metadata=comp_run.metadata,
132-
resource_tracking_run_id=ServiceRunID.get_resource_tracking_run_id(
132+
resource_tracking_run_id=ServiceRunID.get_resource_tracking_run_id_for_computational(
133133
user_id, project_id, node_id, comp_run.iteration
134134
),
135135
)
@@ -322,7 +322,7 @@ async def _process_task_result(
322322
# resource tracking
323323
await publish_service_resource_tracking_stopped(
324324
self.rabbitmq_client,
325-
ServiceRunID.get_resource_tracking_run_id(
325+
ServiceRunID.get_resource_tracking_run_id_for_computational(
326326
user_id, project_id, node_id, iteration
327327
),
328328
simcore_platform_status=simcore_platform_status,

services/director-v2/src/simcore_service_director_v2/modules/dynamic_sidecar/scheduler/_core/_event_create_sidecars.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ async def action(cls, app: FastAPI, scheduler_data: SchedulerData) -> None:
237237

238238
# Each time a new dynamic-sidecar service is created
239239
# generate a new `run_id` to avoid resource collisions
240-
scheduler_data.run_id = ServiceRunID.create_for_dynamic_sidecar()
240+
scheduler_data.run_id = ServiceRunID.get_resource_tracking_run_id_for_dynamic()
241241

242242
rpc_client: RabbitMQRPCClient = app.state.rabbitmq_rpc_client
243243

services/director-v2/tests/unit/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,14 +80,14 @@ def dynamic_sidecar_port() -> PortInt:
8080

8181
@pytest.fixture
8282
def service_run_id() -> ServiceRunID:
83-
return ServiceRunID.create_for_dynamic_sidecar()
83+
return ServiceRunID.get_resource_tracking_run_id_for_dynamic()
8484

8585

8686
@pytest.fixture
8787
def resource_tracking_run_id(
8888
user_id: UserID, project_id: ProjectID, node_id: NodeID
8989
) -> ServiceRunID:
90-
return ServiceRunID.get_resource_tracking_run_id(
90+
return ServiceRunID.get_resource_tracking_run_id_for_computational(
9191
user_id, project_id, node_id, iteration=42
9292
)
9393

services/director-v2/tests/unit/test_modules_dynamic_sidecar_volumes_resolver.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def state_paths() -> list[Path]:
3737

3838
@pytest.fixture
3939
def service_run_id() -> ServiceRunID:
40-
return ServiceRunID.create_for_dynamic_sidecar()
40+
return ServiceRunID.get_resource_tracking_run_id_for_dynamic()
4141

4242

4343
@pytest.fixture
@@ -142,15 +142,15 @@ async def test_unique_name_creation_and_removal(faker: Faker):
142142
unique_volume_name = DynamicSidecarVolumesPathsResolver.source(
143143
path=Path("/some/random/path/to/a/workspace/folder"),
144144
node_uuid=faker.uuid4(cast_to=None),
145-
service_run_id=ServiceRunID.create_for_dynamic_sidecar(),
145+
service_run_id=ServiceRunID.get_resource_tracking_run_id_for_dynamic(),
146146
)
147147

148148
await assert_creation_and_removal(unique_volume_name)
149149

150150

151151
def test_volumes_get_truncated_as_expected(faker: Faker):
152152
node_uuid = faker.uuid4(cast_to=None)
153-
service_run_id = ServiceRunID.create_for_dynamic_sidecar()
153+
service_run_id = ServiceRunID.get_resource_tracking_run_id_for_dynamic()
154154
assert node_uuid != service_run_id
155155
unique_volume_name = DynamicSidecarVolumesPathsResolver.source(
156156
path=Path(

services/dynamic-sidecar/tests/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ def node_id(faker: Faker) -> NodeID:
125125

126126
@pytest.fixture
127127
def service_run_id() -> ServiceRunID:
128-
return ServiceRunID.create_for_dynamic_sidecar()
128+
return ServiceRunID.get_resource_tracking_run_id_for_dynamic()
129129

130130

131131
@pytest.fixture

0 commit comments

Comments
 (0)