Merge pull request #927 from rewenila/mcpo

Add test for mcpo container

Author: dirkmueller

bci_tester/data.py

@@ -1049,6 +1049,24 @@ def create_BCI(
     forwarded_ports=[PortForwarding(container_port=8080)],
 )
 
+MCPO_CONTAINER = create_BCI(
+    build_tag=f"{SAC_CONTAINER_PREFIX}/mcpo:0",
+    bci_type=ImageType.SAC_APPLICATION,
+    available_versions=["15.6-ai"],
+    custom_entry_point="mcpo",
+    extra_entrypoint_args=[
+        "--host",
+        "0.0.0.0",
+        "--port",
+        "8000",
+        "--",
+        "uvx",
+        "mcp-server-time",
+        "--local-timezone=America/New_York",
+    ],
+    forwarded_ports=[PortForwarding(container_port=8000)],
+)
+
 MILVUS_CONTAINER = create_BCI(
     build_tag=f"{SAC_CONTAINER_PREFIX}/milvus:2.4",
     bci_type=ImageType.SAC_APPLICATION,
@@ -1369,6 +1387,7 @@ def create_BCI(
     *KIOSK_XORG_CONTAINERS,
     *KIOSK_XORG_CLIENT_CONTAINERS,
     *KIOSK_PULSEAUDIO_CONTAINERS,
+    MCPO_CONTAINER,
     MICRO_CONTAINER,
     MICRO_FIPS_CONTAINER,
     MINIMAL_CONTAINER,

pyproject.toml

@@ -109,6 +109,7 @@ markers = [
     'git_latest',
     'helm_3',
     'helm_latest',
+    'mcpo_0',
     'milvus_2.4',
     'nginx_1.21',
     'nginx_latest',

tests/test_ai.py

@@ -8,6 +8,7 @@
 
 from bci_tester.data import LMCACHE_LMSTACK_ROUTER_CONTAINER
 from bci_tester.data import LMCACHE_VLLM_OPENAI_CONTAINER
+from bci_tester.data import MCPO_CONTAINER
 from bci_tester.data import MILVUS_CONTAINER
 from bci_tester.data import OLLAMA_CONTAINER
 from bci_tester.data import OPENWEBUI_CONTAINER
@@ -21,6 +22,7 @@
     OLLAMA_CONTAINER,
     OPENWEBUI_CONTAINER,
     MILVUS_CONTAINER,
+    MCPO_CONTAINER,
     PYTORCH_CONTAINER,
     SUSE_AI_OBSERVABILITY_EXTENSION_RUNTIME,
     SUSE_AI_OBSERVABILITY_EXTENSION_SETUP,
@@ -172,3 +174,25 @@ def test_lmstack_router_health(container):
     container.connection.check_output(
         "python3.11 -c 'import vllm_router._version; print(vllm_router._version.__version__)'"
     )
+
+
+@pytest.mark.parametrize(
+    "container_per_test",
+    [MCPO_CONTAINER],
+    indirect=["container_per_test"],
+)
+def test_mcpo_health(container_per_test):
+    """Test that we can reach the port 8000 successfully."""
+    host_port = container_per_test.forwarded_ports[0].host_port
+
+    # Retry 5 times with exponential backoff delay
+    @retry(
+        wait=wait_exponential(multiplier=1, min=4, max=10),
+        stop=stop_after_attempt(5),
+    )
+    def check_mcpo_response():
+        resp = requests.get(f"http://localhost:{host_port}/docs", timeout=30)
+        resp.raise_for_status()
+        assert "mcp-time" in resp.text
+
+    check_mcpo_response()

tests/test_metadata.py

@@ -68,6 +68,7 @@
 from bci_tester.data import LTSS_BASE_FIPS_CONTAINERS
 from bci_tester.data import MARIADB_CLIENT_CONTAINERS
 from bci_tester.data import MARIADB_CONTAINERS
+from bci_tester.data import MCPO_CONTAINER
 from bci_tester.data import MICRO_CONTAINER
 from bci_tester.data import MICRO_FIPS_CONTAINER
 from bci_tester.data import MILVUS_CONTAINER
@@ -338,6 +339,7 @@ def _get_container_ref(
             "lmcache-lmstack-router",
             ImageType.SAC_APPLICATION,
         ),
+        (MCPO_CONTAINER, "mcpo", ImageType.SAC_APPLICATION),
     ]
     + [(STUNNEL_CONTAINER, "stunnel", ImageType.APPLICATION)]
     + [