# -*- coding: utf-8 -*-
"""
Integration tests for /version and the Version tab in the Admin UI.
Copyright 2025
SPDX-License-Identifier: Apache-2.0
Author: Mihai Criveti
"""
# Future
from __future__ import annotations
# Standard
import base64
from typing import Dict
# Third-Party
import pytest
from starlette.testclient import TestClient
# First-Party
from mcpgateway.config import settings
from mcpgateway.main import app
# --------------------------------------------------------------------------- #
# Fixtures
# --------------------------------------------------------------------------- #
@pytest.fixture(scope="session")
def test_client() -> TestClient:
"""Spin up the FastAPI test client once for the whole session."""
return TestClient(app)
@pytest.fixture()
def auth_headers() -> Dict[str, str]:
"""
Build the auth headers expected by the gateway:
* Authorization: Basic <base64(user:pw)>
* X-API-Key: user:pw (plain text)
"""
creds = f"{settings.basic_auth_user}:{settings.basic_auth_password}"
basic_b64 = base64.b64encode(creds.encode()).decode()
return {
"Authorization": f"Basic {basic_b64}",
"X-API-Key": creds,
}
# --------------------------------------------------------------------------- #
# Tests
# --------------------------------------------------------------------------- #
# def test_version_partial_html(test_client: TestClient, auth_headers: Dict[str, str]):
# """
# /version?partial=true must return an HTML fragment with core meta-info.
# """
# resp = test_client.get("/version?partial=true", headers=auth_headers)
# assert resp.status_code == 200
# assert "text/html" in resp.headers["content-type"]
# html = resp.text
# # Very loose sanity checks - we only care that it is an HTML fragment
# # and that some well-known marker exists.
# assert "<div" in html
# assert "App:" in html or "Application:" in html
@pytest.mark.skipif(not settings.mcpgateway_ui_enabled, reason="Admin UI tests require MCPGATEWAY_UI_ENABLED=true")
def test_admin_ui_contains_version_tab(test_client: TestClient, auth_headers: Dict[str, str]):
"""The Admin dashboard must contain the "Version & Environment Info" tab."""
resp = test_client.get("/admin", headers=auth_headers)
assert resp.status_code == 200
assert 'id="tab-version-info"' in resp.text
assert "Version and Environment Info" in resp.text
# def test_version_partial_htmx_load(test_client: TestClient, auth_headers: Dict[str, str]):
# """
# A second call (mimicking an HTMX swap) should yield the same fragment.
# """
# resp = test_client.get("/version?partial=true", headers=auth_headers)
# assert resp.status_code == 200
# html = resp.text
# assert "<div" in html
# assert "App:" in html or "Application:" in html
