Skip to content

Commit 8eb7690

Browse files
committed
Refactor catalog iteration
1 parent b9484fe commit 8eb7690

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

test/integration_test.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -462,23 +462,24 @@ class Catalog:
462462
log.warning('Will skip indexing due to overriding IT flag.')
463463

464464
catalogs: list[Catalog] = []
465-
for catalog in config.integration_test_catalogs:
465+
for catalog in config.integration_test_catalogs.values():
466466
if index:
467-
public_source, _ = self._select_source(catalog, public=True)
468-
ma_source = self._select_source(catalog, public=False)
467+
public_source, _ = self._select_source(catalog.name, public=True)
468+
ma_source = self._select_source(catalog.name, public=False)
469469
if ma_source is not None:
470470
ma_source = ma_source[0]
471471
sources = alist(public_source, ma_source)
472-
notifications, fqids = self._prepare_notifications(catalog, sources)
472+
notifications, fqids = self._prepare_notifications(catalog.name, sources)
473473
else:
474474
with self._service_account_credentials:
475-
fqids = self._get_indexed_bundles(catalog)
475+
fqids = self._get_indexed_bundles(catalog.name)
476476
indexed_sources = {fqid.source for fqid in fqids}
477-
ma_source_ids = {s.id for s in self.managed_access_sources_by_catalog[catalog]}
477+
ma_sources = self.managed_access_sources_by_catalog[catalog.name]
478+
ma_source_ids = {s.id for s in ma_sources}
478479
public_source = one(s for s in indexed_sources if s.id not in ma_source_ids)
479480
ma_source = only(s for s in indexed_sources if s.id in ma_source_ids)
480481
notifications = []
481-
catalogs.append(Catalog(name=catalog,
482+
catalogs.append(Catalog(name=catalog.name,
482483
bundles=fqids,
483484
notifications=notifications,
484485
public_source=public_source,

0 commit comments

Comments
 (0)