Skip to content

Commit 5559d74

Browse files
committed
✨ Refactor service lifespan management: rename repository_lifespan to repository_lifespan_manager for improved clarity
1 parent 4de05d8 commit 5559d74

File tree

4 files changed

+21
-12
lines changed

4 files changed

+21
-12
lines changed

services/catalog/src/simcore_service_catalog/api/rpc/_services.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -210,17 +210,17 @@ async def batch_get_my_services(
210210
) -> list[MyServiceGet]:
211211
assert app.state.engine # nosec
212212

213-
services = await services.batch_get_my_services(
213+
services_batch = await services.batch_get_my_services(
214214
repo=ServicesRepository(app.state.engine),
215215
groups_repo=GroupsRepository(app.state.engine),
216216
product_name=product_name,
217217
user_id=user_id,
218218
ids=ids,
219219
)
220220

221-
assert [(sv.key, sv.release.version) for sv in services] == ids # nosec
221+
assert [(sv.key, sv.release.version) for sv in services_batch] == ids # nosec
222222

223-
return services
223+
return services_batch
224224

225225

226226
@router.expose(reraise_if_error_type=(ValidationError,))

services/catalog/src/simcore_service_catalog/core/events.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from ..api.rpc.events import rpc_api_lifespan
1515
from ..clients.director import director_lifespan
1616
from ..clients.rabbitmq import rabbitmq_lifespan
17-
from ..repository.events import repository_lifespan
17+
from ..repository.events import repository_lifespan_manager
1818
from ..service.function_services import function_services_lifespan
1919
from .background_tasks import background_task_lifespan
2020
from .settings import ApplicationSettings
@@ -61,7 +61,7 @@ def create_app_lifespan():
6161
app_lifespan.add(_main_lifespan)
6262

6363
# - postgres
64-
app_lifespan.add(repository_lifespan)
64+
app_lifespan.include(repository_lifespan_manager)
6565

6666
# - rabbitmq
6767
app_lifespan.add(rabbitmq_lifespan)

services/catalog/src/simcore_service_catalog/repository/events.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,11 @@
1313
_logger = logging.getLogger(__name__)
1414

1515

16-
repository_lifespan = LifespanManager()
17-
repository_lifespan.add(postgres_database_lifespan)
16+
repository_lifespan_manager = LifespanManager()
17+
repository_lifespan_manager.add(postgres_database_lifespan)
1818

1919

20-
@repository_lifespan.add
20+
@repository_lifespan_manager.add
2121
async def _database_lifespan(app: FastAPI, state: State) -> AsyncIterator[State]:
2222
app.state.engine = state[PostgresLifespanState.POSTGRES_ASYNC_ENGINE]
2323

services/catalog/tests/unit/conftest.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,9 @@ def service_caching_disabled(monkeypatch: pytest.MonkeyPatch) -> None:
220220
@pytest.fixture
221221
def repository_lifespan_disabled(mocker: MockerFixture):
222222
mocker.patch.object(
223-
simcore_service_catalog.repository.events, "repository_lifespan"
223+
simcore_service_catalog.core.events,
224+
"repository_lifespan_manager",
225+
autospec=True,
224226
)
225227

226228

@@ -244,6 +246,7 @@ async def __aexit__(self, exc_type, exc_value, traceback):
244246
mocker.patch.object(
245247
simcore_service_catalog.core.events,
246248
"background_task_lifespan",
249+
autospec=True,
247250
return_value=MockedBackgroundTaskContextManager(),
248251
)
249252

@@ -256,8 +259,12 @@ async def __aexit__(self, exc_type, exc_value, traceback):
256259
@pytest.fixture
257260
def rabbitmq_and_rpc_setup_disabled(mocker: MockerFixture):
258261
# The following services are affected if rabbitmq is not in place
259-
mocker.patch.object(simcore_service_catalog.core.events, "rabbitmq_lifespan")
260-
mocker.patch.object(simcore_service_catalog.core.events, "rpc_api_lifespan")
262+
mocker.patch.object(
263+
simcore_service_catalog.core.events, "rabbitmq_lifespan", autospec=True
264+
)
265+
mocker.patch.object(
266+
simcore_service_catalog.core.events, "rpc_api_lifespan", autospec=True
267+
)
261268

262269

263270
@pytest.fixture
@@ -274,7 +281,9 @@ async def rpc_client(
274281

275282
@pytest.fixture
276283
def director_setup_disabled(mocker: MockerFixture) -> None:
277-
mocker.patch.object(simcore_service_catalog.core.events, "director_lifespan")
284+
mocker.patch.object(
285+
simcore_service_catalog.core.events, "director_lifespan", autospec=True
286+
)
278287

279288

280289
@pytest.fixture

0 commit comments

Comments
 (0)