diff --git a/iib/workers/tasks/build.py b/iib/workers/tasks/build.py index ab4c9e00f..b917680a3 100644 --- a/iib/workers/tasks/build.py +++ b/iib/workers/tasks/build.py @@ -951,7 +951,10 @@ def handle_add_request( msg = 'Checking if bundles are already present in index image' log.info(msg) set_request_state(request_id, 'in_progress', msg) - present_bundles = _get_present_bundles(from_index, temp_dir) + + with set_registry_token(overwrite_from_index_token, from_index): + present_bundles = _get_present_bundles(from_index, temp_dir) + filtered_bundles = _get_missing_bundles(present_bundles, resolved_bundles) excluded_bundles = [ bundle for bundle in resolved_bundles if bundle not in filtered_bundles diff --git a/tests/test_workers/test_tasks/test_build.py b/tests/test_workers/test_tasks/test_build.py index 264928d5d..a4a420975 100644 --- a/tests/test_workers/test_tasks/test_build.py +++ b/tests/test_workers/test_tasks/test_build.py @@ -674,7 +674,9 @@ def test_skopeo_copy_fail_max_retries(mock_run_cmd): @mock.patch('iib.workers.tasks.build._add_ocp_label_to_index') @mock.patch('iib.workers.tasks.build._get_present_bundles') @mock.patch('iib.workers.tasks.build._get_missing_bundles') +@mock.patch('iib.workers.tasks.build.set_registry_token') def test_handle_add_request( + mock_srt, mock_gmb, mock_gpb, mock_aolti, @@ -737,6 +739,7 @@ def test_handle_add_request( filter_args = mock_gmb.call_args[0] assert ['some-bundle@sha'] in filter_args mock_oia.assert_called_once() + mock_srt.assert_called_once() assert mock_bi.call_count == len(arches) assert mock_pi.call_count == len(arches) @@ -833,7 +836,9 @@ def test_handle_add_request_bundle_resolution_failure(mock_grb, mock_srs, mock_c @mock.patch('iib.workers.tasks.build._add_ocp_label_to_index') @mock.patch('iib.workers.tasks.build._get_present_bundles') @mock.patch('iib.workers.tasks.build._get_missing_bundles') +@mock.patch('iib.workers.tasks.build.set_registry_token') def test_handle_add_request_backport_failure_no_overwrite( + mock_srt, mock_gmb, mock_gpb, mock_aolti,