Skip to content

Commit b40ae44

Browse files
committed
fixing tests
1 parent 2b8d88f commit b40ae44

File tree

2 files changed

+29
-24
lines changed

2 files changed

+29
-24
lines changed

services/catalog/tests/unit/with_dbs/conftest.py

Lines changed: 15 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -247,22 +247,27 @@ async def services_db_tables_injector(
247247

248248
async def _inject_in_db(fake_catalog: list[tuple]):
249249
# [(service, ar1, ...), (service2, ar1, ...) ]
250+
iter_services = (items[0] for items in fake_catalog)
251+
iter_access_rights = itertools.chain(items[1:] for items in fake_catalog)
250252

251253
async with sqlalchemy_async_engine.begin() as conn:
252254
# NOTE: The 'default' dialect with current database version settings does not support in-place multirow inserts
253-
for service in [items[0] for items in fake_catalog]:
254-
insert_meta = pg_insert(services_meta_data).values(**service)
255-
upsert_meta = insert_meta.on_conflict_do_update(
256-
index_elements=[
257-
services_meta_data.c.key,
258-
services_meta_data.c.version,
259-
],
260-
set_=service,
255+
for service in iter_services:
256+
257+
insert_stmt = pg_insert(services_meta_data).values(**service)
258+
259+
update_stmt = insert_stmt.on_conflict_do_update(
260+
index_elements=["key", "version"],
261+
set_={
262+
column_name: insert_stmt.excluded[column_name]
263+
for column_name in service
264+
if column_name not in ("key", "version")
265+
},
261266
)
262-
await conn.execute(upsert_meta)
267+
await conn.execute(update_stmt)
263268
inserted_services.add((service["key"], service["version"]))
264269

265-
for access_rights in itertools.chain(items[1:] for items in fake_catalog):
270+
for access_rights in iter_access_rights:
266271
stmt_access = services_access_rights.insert().values(access_rights)
267272
await conn.execute(stmt_access)
268273

services/catalog/tests/unit/with_dbs/test_service_access_rights.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
import simcore_service_catalog.service.access_rights
88
from fastapi import FastAPI
9-
from models_library.groups import GroupAtDB, GroupID
9+
from models_library.groups import GroupID
1010
from models_library.products import ProductName
1111
from models_library.services import ServiceMetaDataPublished, ServiceVersion
1212
from pydantic import TypeAdapter
@@ -102,18 +102,18 @@ async def test_auto_upgrade_policy(
102102
return_value=False,
103103
)
104104
# Avoids creating a users + user_to_group table
105-
data = GroupAtDB.model_json_schema()["example"]
106-
data["gid"] = everyone_gid
107-
mocker.patch.object(
108-
simcore_service_catalog.service.access_rights.GroupsRepository,
109-
"get_everyone_group",
110-
return_value=GroupAtDB.model_validate(data),
111-
)
112-
mocker.patch.object(
113-
simcore_service_catalog.service.access_rights.GroupsRepository,
114-
"get_user_gid_from_email",
115-
return_value=user_gid,
116-
)
105+
# data = GroupAtDB.model_json_schema()["example"]
106+
# data["gid"] = everyone_gid
107+
# mocker.patch.object(
108+
# simcore_service_catalog.service.access_rights.GroupsRepository,
109+
# "get_everyone_group",
110+
# return_value=GroupAtDB.model_validate(data),
111+
# )
112+
# mocker.patch.object(
113+
# simcore_service_catalog.service.access_rights.GroupsRepository,
114+
# "get_user_gid_from_email",
115+
# return_value=user_gid,
116+
# )
117117

118118
# SETUP ---
119119
MOST_UPDATED_EXAMPLE = -1
@@ -144,7 +144,7 @@ async def test_auto_upgrade_policy(
144144
team_access="x",
145145
everyone_access=None,
146146
product=other_product, # <-- different product
147-
),
147+
)
148148
)
149149

150150
latest_release_in_other_product = (

0 commit comments

Comments
 (0)