Skip to content

Commit

Permalink
Adding parameter ignore_bundle_ocp_version to merge-index API
Browse files Browse the repository at this point in the history
- If ignore_bundle_ocp_version parameter is set to true while calling merge-index API and target_index image is listed in iib_no_ocp_label_allow_list in config file then any bundle without "com.redhat.openshift.versions" set will be added to target_index.

[CLOUDDST-20869]
  • Loading branch information
lipoja committed Nov 23, 2023
1 parent 6fb9225 commit a58b5d4
Show file tree
Hide file tree
Showing 8 changed files with 93 additions and 12 deletions.
1 change: 1 addition & 0 deletions iib/web/api_v1.py
Original file line number Diff line number Diff line change
Expand Up @@ -1091,6 +1091,7 @@ def merge_index_image() -> Tuple[flask.Response, int]:
flask.current_app.config['IIB_BINARY_IMAGE_CONFIG'],
payload.get('build_tags', []),
payload.get('graph_update_mode'),
payload.get('ignore_bundle_ocp_version'),
]
safe_args = _get_safe_args(args, payload)

Expand Down
1 change: 1 addition & 0 deletions iib/web/iib_static_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -391,6 +391,7 @@ class MergeIndexImageRequestResponse(APIPartImageBuildRequestResponse):
deprecation_list: List[str]
distribution_scope: str
graph_update_mode: GRAPH_MODE_LITERAL
ignore_bundle_ocp_version: Optional[bool]
index_image: Optional[str]
source_from_index: str
source_from_index_resolved: Optional[str]
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
"""Adding ignore_bundle_ocp_version.
Revision ID: 1920ad83d0ab
Revises: 9e9d4f9730c8
Create Date: 2023-11-22 12:03:50.711489
"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '1920ad83d0ab'
down_revision = '9e9d4f9730c8'
branch_labels = None
depends_on = None


def upgrade():
with op.batch_alter_table('request_merge_index_image', schema=None) as batch_op:
batch_op.add_column(sa.Column('ignore_bundle_ocp_version', sa.Boolean(), nullable=True))


def downgrade():
with op.batch_alter_table('request_merge_index_image', schema=None) as batch_op:
batch_op.drop_column('ignore_bundle_ocp_version')
2 changes: 2 additions & 0 deletions iib/web/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -1511,6 +1511,7 @@ class RequestMergeIndexImage(Request):
)
distribution_scope: Mapped[Optional[str]]
graph_update_mode: Mapped[Optional[str]]
ignore_bundle_ocp_version: Mapped[Optional[bool]]

__mapper_args__ = {
'polymorphic_identity': RequestTypeMapping.__members__['merge_index_image'].value
Expand Down Expand Up @@ -1640,6 +1641,7 @@ def to_json(self, verbose: Optional[bool] = True) -> MergeIndexImageRequestRespo
)
rv['deprecation_list'] = [bundle.pull_specification for bundle in self.deprecation_list]
rv['graph_update_mode'] = self.graph_update_mode
rv['ignore_bundle_ocp_version'] = self.ignore_bundle_ocp_version
rv['index_image'] = getattr(self.index_image, 'pull_specification', None)
rv['source_from_index'] = self.source_from_index.pull_specification
rv['source_from_index_resolved'] = getattr(
Expand Down
8 changes: 8 additions & 0 deletions iib/web/static/api_v1.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -1347,6 +1347,14 @@ components:
image pull specs configured in IIB_GRAPH_MODE_INDEX_ALLOW_LIST in the IIB API Config. If not
specified, "--mode" will not be added to OPM commands to add the bundle(s) to the index.
default: None
ignore_bundle_ocp_version:
type: string
description: >
When `ignore_bundle_ocp_version` is set to `true` and image set as target_index
is listed in `iib_no_ocp_label_allow_list` config then bundles without
"com.redhat.openshift.versions" label set will be added in the result `index_image`.
example: 'true'
default: False
RequestCreateEmptyIndex:
type: object
properties:
Expand Down
20 changes: 16 additions & 4 deletions iib/workers/tasks/build_merge_index_image.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@ def _add_bundles_missing_in_source(
graph_update_mode: Optional[str] = None,
target_index=None,
overwrite_target_index_token: Optional[str] = None,
ignore_bundle_ocp_version: Optional[bool] = False,
) -> Tuple[List[BundleImage], List[BundleImage]]:
"""
Rebuild index image with bundles missing from source image but present in target image.
Expand All @@ -84,6 +85,9 @@ def _add_bundles_missing_in_source(
:param str overwrite_target_index_token: the token used for overwriting the input
``source_from_index`` image. This is required to use ``overwrite_target_index``.
The format of the token must be in the format "user:password".
:param bool ignore_bundle_ocp_version: When set to `true` and image set as target_index is
listed in `iib_no_ocp_label_allow_list` config then bundles without
"com.redhat.openshift.versions" label set will be added in the result `index_image`.
:return: tuple where the first value is a list of bundles which were added to the index image
and the second value is a list of bundles in the new index whose ocp_version range does not
satisfy the ocp_version value of the target index.
Expand Down Expand Up @@ -124,10 +128,13 @@ def _add_bundles_missing_in_source(
missing_bundles.append(bundle)
missing_bundle_paths.append(bundle['bundlePath'])

allow_no_ocp_version = any(
target_index.startswith(index)
for index in get_worker_config()['iib_no_ocp_label_allow_list']
)
if ignore_bundle_ocp_version and target_index is not None:
allow_no_ocp_version = any(
target_index.startswith(index)
for index in get_worker_config()['iib_no_ocp_label_allow_list']
)
else:
allow_no_ocp_version = False

for bundle in itertools.chain(missing_bundles, source_index_bundles):
if not is_bundle_version_valid(bundle['bundlePath'], ocp_version, allow_no_ocp_version):
Expand Down Expand Up @@ -190,6 +197,7 @@ def handle_merge_request(
binary_image_config: Optional[str] = None,
build_tags: Optional[List[str]] = None,
graph_update_mode: Optional[str] = None,
ignore_bundle_ocp_version: Optional[bool] = False,
) -> None:
"""
Coordinate the work needed to merge old (N) index image with new (N+1) index image.
Expand All @@ -212,6 +220,9 @@ def handle_merge_request(
:param build_tags: list of extra tag to use for intermetdiate index image
:param str graph_update_mode: Graph update mode that defines how channel graphs are updated
in the index.
:param bool ignore_bundle_ocp_version: When set to `true` and image set as target_index is
listed in `iib_no_ocp_label_allow_list` config then bundles without
"com.redhat.openshift.versions" label set will be added in the result `index_image`.
:raises IIBError: if the index image merge fails.
"""
_cleanup()
Expand Down Expand Up @@ -282,6 +293,7 @@ def handle_merge_request(
target_index=target_index,
overwrite_target_index_token=overwrite_target_index_token,
distribution_scope=prebuild_info['distribution_scope'],
ignore_bundle_ocp_version=ignore_bundle_ocp_version,
)

missing_bundle_paths = [bundle['bundlePath'] for bundle in missing_bundles]
Expand Down
2 changes: 2 additions & 0 deletions tests/test_web/test_api_v1.py
Original file line number Diff line number Diff line change
Expand Up @@ -1963,6 +1963,7 @@ def test_merge_index_image_success(
'distribution_scope': distribution_scope,
'graph_update_mode': 'semver',
'id': 1,
'ignore_bundle_ocp_version': None,
'index_image': None,
'logs': {
'expiration': '2020-02-15T17:03:00Z',
Expand Down Expand Up @@ -2067,6 +2068,7 @@ def test_merge_index_image_custom_user_queue(
'source_from_index': 'source_index:image',
'target_index': 'target_index:image',
'graph_update_mode': 'replaces',
'ignore_bundle_ocp_version': True,
}
if overwrite_from_index:
data['overwrite_target_index'] = True
Expand Down
45 changes: 37 additions & 8 deletions tests/test_workers/test_tasks/test_build_merge_index_image.py
Original file line number Diff line number Diff line change
Expand Up @@ -331,6 +331,7 @@ def test_handle_merge_request_no_deprecate(
mock_run_cmd.assert_not_called()


@mock.patch('iib.workers.config.get_worker_config')
@mock.patch('iib.workers.tasks.build_merge_index_image.is_image_fbc')
@mock.patch('iib.workers.tasks.build_merge_index_image.get_image_label')
@mock.patch('iib.workers.tasks.build_merge_index_image._create_and_push_manifest_list')
Expand All @@ -340,7 +341,7 @@ def test_handle_merge_request_no_deprecate(
@mock.patch('iib.workers.tasks.build_merge_index_image._opm_index_add')
@mock.patch('iib.workers.tasks.build_merge_index_image.set_request_state')
def test_add_bundles_missing_in_source(
mock_srs, mock_oia, mock_aolti, mock_bi, mock_pi, mock_capml, mock_gil, mock_iifbc
mock_srs, mock_oia, mock_aolti, mock_bi, mock_pi, mock_capml, mock_gil, mock_iifbc, mock_gwc
):
source_bundles = [
{
Expand Down Expand Up @@ -387,8 +388,19 @@ def test_add_bundles_missing_in_source(
'bundlePath': 'quay.io/bundle4@sha256:569854',
'csvName': 'bundle5-5.0',
},
{
'packageName': 'bundle6-ignore-ocp-failed',
'version': '14.0',
'bundlePath': 'quay.io/ignore-ocp-failed-bundle4@sha256:567890',
'csvName': 'bundle6-ignore-ocp-failed-14.0',
},
]
mock_gil.side_effect = ['=v4.5', '=v4.6', 'v4.7', 'v4.5-v4.7', 'v4.5,v4.6']

mock_gwc.iib_api_url.return_value = {
'iib_no_ocp_label_allow_list': ['quay.io/bundle'],
}

mock_gil.side_effect = ['=v4.5', '=v4.6', 'v4.7', 'v4.5-v4.7', 'v4.5,v4.6', '']
mock_iifbc.return_value = False
missing_bundles, invalid_bundles = build_merge_index_image._add_bundles_missing_in_source(
source_bundles,
Expand All @@ -401,6 +413,7 @@ def test_add_bundles_missing_in_source(
'4.6',
'dev',
'replaces',
ignore_bundle_ocp_version=True,
)
assert missing_bundles == [
{
Expand All @@ -415,6 +428,12 @@ def test_add_bundles_missing_in_source(
'bundlePath': 'quay.io/bundle4@sha256:567890',
'csvName': 'bundle4-4.0',
},
{
'bundlePath': 'quay.io/ignore-ocp-failed-bundle4@sha256:567890',
'csvName': 'bundle6-ignore-ocp-failed-14.0',
'packageName': 'bundle6-ignore-ocp-failed',
'version': '14.0',
},
]
assert invalid_bundles == [
{
Expand All @@ -424,22 +443,32 @@ def test_add_bundles_missing_in_source(
'csvName': 'bundle3-3.0',
},
{
'packageName': 'bundle1',
'version': '1.0',
'bundlePath': 'quay.io/bundle1@sha256:123456',
'csvName': 'bundle1-1.0',
'packageName': 'bundle6-ignore-ocp-failed',
'version': '14.0',
'bundlePath': 'quay.io/ignore-ocp-failed-bundle4@sha256:567890',
'csvName': 'bundle6-ignore-ocp-failed-14.0',
},
{
'packageName': 'bundle5',
'version': '5.0-2',
'bundlePath': 'quay.io/bundle2@sha256:456132',
'csvName': 'bundle5-5.0',
},
]
mock_srs.assert_called_once()
mock_oia.assert_called_once_with(
base_dir='some_dir',
bundles=['quay.io/bundle3@sha256:456789', 'quay.io/bundle4@sha256:567890'],
bundles=[
'quay.io/bundle3@sha256:456789',
'quay.io/bundle4@sha256:567890',
'quay.io/ignore-ocp-failed-bundle4@sha256:567890',
],
binary_image='binary-image:4.5',
from_index='index-image:4.6',
container_tool='podman',
graph_update_mode='replaces',
)
assert mock_gil.call_count == 5
assert mock_gil.call_count == 6
assert mock_aolti.call_count == 2
mock_bi.assert_called_once()
mock_pi.assert_called_once()
Expand Down

0 comments on commit a58b5d4

Please sign in to comment.