Extract variable and rename method (#6691)

DataBiosphere · Nov 14, 2024 · 05f053a · 05f053a
1 parent 31a0443
commit 05f053a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test/integration_test.py b/test/integration_test.py
@@ -286,7 +286,7 @@ def managed_access_sources_by_catalog(self
                     managed_access_sources[catalog].add(ref)
         return managed_access_sources
 
-    def _choose_source(self,
+    def _select_source(self,
                        catalog: CatalogName,
                        *,
                        public: bool | None = None
@@ -407,10 +407,10 @@ def _wait_for_indexer():
         catalogs: list[Catalog] = []
         for catalog in config.integration_test_catalogs:
             if index:
-                public_source = self._choose_source(catalog, public=True)
-                ma_source = self._choose_source(catalog, public=False)
-                notifications, fqids = self._prepare_notifications(catalog,
-                                                                   sources=alist(public_source, ma_source))
+                public_source = self._select_source(catalog, public=True)
+                ma_source = self._select_source(catalog, public=False)
+                sources = alist(public_source, ma_source)
+                notifications, fqids = self._prepare_notifications(catalog, sources)
             else:
                 with self._service_account_credentials:
                     fqids = self._get_indexed_bundles(catalog)
@@ -1902,7 +1902,7 @@ def test_can_bundle_canned_repository(self):
             self._test_catalog(mock_catalog)
 
     def bundle_fqid(self, catalog: CatalogName) -> SourcedBundleFQID:
-        source = self._choose_source(catalog)
+        source = self._select_source(catalog)
         # The plugin will raise an exception if the source lacks a prefix
         source = source.with_prefix(Prefix.of_everything)
         bundle_fqids = self.azul_client.list_bundles(catalog, source, prefix='')