RHAIENG-2189: configure PIP_INDEX_URL

Author: jiridanek

codeserver/ubi9-python-3.12/Dockerfile.cpu

@@ -72,9 +72,10 @@ FROM ${BASE_IMAGE} AS cpu-base
 
 WORKDIR /opt/app-root/bin
 
-# RHAIENG-2189: this is AIPCC migration phase 1.5
+### BEGIN RHAIENG-2189: this is AIPCC migration phase 1.5
 ENV PIP_EXTRA_INDEX_URL=https://pypi.org/simple
 ENV UV_EXTRA_INDEX_URL=https://pypi.org/simple
+### END RHAIENG-2189: this is AIPCC migration phase 1.5
 
 # OS Packages needs to be installed as root
 USER 0

jupyter/minimal/ubi9-python-3.12/Dockerfile.cpu

@@ -11,9 +11,10 @@ FROM registry.access.redhat.com/ubi9/ubi AS ubi-repos
 ####################
 FROM ${BASE_IMAGE} AS cpu-base
 
-# RHAIENG-2189: this is AIPCC migration phase 1.5
+### BEGIN RHAIENG-2189: this is AIPCC migration phase 1.5
 ENV PIP_EXTRA_INDEX_URL=https://pypi.org/simple
 ENV UV_EXTRA_INDEX_URL=https://pypi.org/simple
+### END RHAIENG-2189: this is AIPCC migration phase 1.5
 
 WORKDIR /opt/app-root/bin
 

runtimes/datascience/ubi9-python-3.12/Dockerfile.cpu

@@ -10,9 +10,10 @@ FROM ${BASE_IMAGE} AS cpu-base
 
 ARG TARGETARCH
 
-# RHAIENG-2189: this is AIPCC migration phase 1.5
+### BEGIN RHAIENG-2189: this is AIPCC migration phase 1.5
 ENV PIP_EXTRA_INDEX_URL=https://pypi.org/simple
 ENV UV_EXTRA_INDEX_URL=https://pypi.org/simple
+### END RHAIENG-2189: this is AIPCC migration phase 1.5
 
 WORKDIR /opt/app-root/bin
 

runtimes/minimal/ubi9-python-3.12/Dockerfile.cpu

@@ -8,9 +8,10 @@ ARG BASE_IMAGE
 ####################
 FROM ${BASE_IMAGE} AS cpu-base
 
-# RHAIENG-2189: this is AIPCC migration phase 1.5
+### BEGIN RHAIENG-2189: this is AIPCC migration phase 1.5
 ENV PIP_EXTRA_INDEX_URL=https://pypi.org/simple
 ENV UV_EXTRA_INDEX_URL=https://pypi.org/simple
+### END RHAIENG-2189: this is AIPCC migration phase 1.5
 
 WORKDIR /opt/app-root/bin
 

scripts/dockerfile_fragments.py

@@ -61,6 +61,10 @@ def main():
     """)
 
     replacements = {
+        "RHAIENG-2189: this is AIPCC migration phase 1.5": textwrap.dedent(r"""
+            ENV PIP_EXTRA_INDEX_URL=https://pypi.org/simple
+            ENV UV_EXTRA_INDEX_URL=https://pypi.org/simple"""),
+
         "Subscribe with subscription manager": textwrap.dedent(subscription_manager_register_refresh),
         "upgrade first to avoid fixable vulnerabilities": textwrap.dedent(ntb.process_template_with_indents(rt"""
             {subscription_manager_register_refresh}

tests/containers/base_image_test.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import binascii
+import contextlib
 import inspect
 import json
 import logging
@@ -11,6 +12,7 @@
 import textwrap
 from typing import TYPE_CHECKING, Any
 
+import allure
 import pytest
 import testcontainers.core.container
 
@@ -20,7 +22,7 @@
 LOGGER = logging.getLogger(__name__)
 
 if TYPE_CHECKING:
-    from collections.abc import Callable
+    from collections.abc import Callable, Generator
 
     import pytest_subtests
 
@@ -43,6 +45,23 @@ def _run_test(self, image: str, test_fn: Callable[[testcontainers.core.container
         # If the return doesn't happen in the try block, fail the test
         pytest.fail("The test did not pass as expected.")
 
+    @contextlib.contextmanager
+    def _test_container(self, image: str) -> Generator[testcontainers.core.container.DockerContainer]:
+        """Context manager that starts a test container and yields it."""
+        container = testcontainers.core.container.DockerContainer(image=image, user=23456, group_add=[0])
+        container.with_command("/bin/sh -c 'sleep infinity'")
+        try:
+            container.start()
+            yield container
+            return
+        except Exception as e:
+            pytest.fail(f"Unexpected exception in test: {e}")
+        finally:
+            docker_utils.NotebookContainer(container).stop(timeout=0)
+
+        # If the return doesn't happen in the try block, fail the test
+        pytest.fail("The test did not pass as expected.")
+
     def test_elf_files_can_link_runtime_libs(self, subtests: pytest_subtests.SubTests, image):
         def test_fn(container: testcontainers.core.container.DockerContainer):
             def check_elf_file():
@@ -250,6 +269,28 @@ def test_fn(container: testcontainers.core.container.DockerContainer):
 
         self._run_test(image=image, test_fn=test_fn)
 
+    @allure.issue("RHAIENG-2189")
+    def test_python_package_index(self, image: str, subtests: pytest_subtests.SubTests):
+        """Checks that we use the Python Package Index we mean to use.
+        https://redhat-internal.slack.com/archives/C05TTTYG599/p1764240587118899?thread_ts=1764234802.564119&cid=C05TTTYG599
+        """
+
+        expected_env = {
+            "PIP_EXTRA_INDEX_URL": "https://pypi.org/simple",
+            "UV_EXTRA_INDEX_URL": "https://pypi.org/simple",
+        }
+
+        with self._test_container(image=image) as container:
+            _, output = container.exec(["env"])
+            output = output.decode().strip()
+
+            actual = {}
+            for line in output.splitlines():
+                key, value = line.split("=", maxsplit=1)
+                actual[key] = value
+
+            assert actual.items() >= expected_env.items()
+
 
 def encode_python_function_execution_command_interpreter(
     python: str, function: Callable[..., Any], *args: list[Any]