task: set status betterstack page conditionally for staging and production

Author: melifaro

src/aignostics/gui/_frame.py

@@ -14,6 +14,7 @@
 from loguru import logger
 
 from aignostics.constants import WINDOW_TITLE
+from aignostics.platform import API_ROOT_PRODUCTION, API_ROOT_STAGING
 from aignostics.utils import __version__, open_user_data_directory
 
 from ._theme import theme
@@ -29,6 +30,24 @@
 CLASSES_FULL_SIZE = f"{CLASSES_FULL_WIDTH} {CLASSES_FULL_HEIGHT}"
 
 
+def get_status_page_url(api_root: str) -> str | None:
+    """Get the status page URL based on the API root environment.
+
+    Args:
+        api_root: The API root URL to determine the environment
+
+    Returns:
+        The status page URL for production/staging, or None for dev/test
+    """
+    if api_root == API_ROOT_PRODUCTION:
+        return "https://status.platform.aignostics.com"
+    elif api_root == API_ROOT_STAGING:
+        return "https://status.platform-staging.aignostics.com"
+    else:
+        # No status page for dev and test environments
+        return None
+
+
 @contextmanager
 def frame(  # noqa: C901, PLR0915
     navigation_title: str,
@@ -213,7 +232,9 @@ def _update_health() -> None:
             coroutine=_health_load_and_render(),
             name="_health_load_and_render",
         )
-        ui.run_javascript("document.getElementById('betterstack').src = document.getElementById('betterstack').src;")
+        # Only refresh the status iframe if it exists (production/staging)
+        if get_status_page_url(settings().api_root):
+            ui.run_javascript("document.getElementById('betterstack').src = document.getElementById('betterstack').src;")
 
     ui.timer(interval=HEALTH_UPDATE_INTERVAL, callback=_update_health, immediate=True)
 
@@ -342,13 +363,15 @@ def toggle_dark_mode() -> None:
                     ui.link("Get Support", "https://platform.aignostics.com/support", new_tab=True).mark(
                         "LINK_DOCUMENTATION"
                     )
-            with ui.item().props("clickable"):
-                with ui.item_section().props("avatar"):
-                    ui.icon("check_circle", color="primary")
-                with ui.item_section():
-                    ui.link("Check Platform Status", "https://status.aignostics.com", new_tab=True).mark(
-                        "LINK_DOCUMENTATION"
-                    )
+            status_url = get_status_page_url(settings().api_root)
+            if status_url:
+                with ui.item().props("clickable"):
+                    with ui.item_section().props("avatar"):
+                        ui.icon("check_circle", color="primary")
+                    with ui.item_section():
+                        ui.link("Check Platform Status", status_url, new_tab=True).mark(
+                            "LINK_DOCUMENTATION"
+                        )
             with ui.item().props("clickable"):
                 with ui.item_section().props("avatar"):
                     ui.icon("handshake", color="primary")
@@ -368,14 +391,16 @@ def toggle_dark_mode() -> None:
         ui.row(align_items="center").classes("justify-start w-full"),
     ):
         health_link()
-        with ui.row().style("padding: 0"):
-            ui.html(
-                '<iframe id="betterstack" src="https://status.aignostics.com/badge?theme=dark" '
-                'width="250" height="30" frameborder="0" scrolling="no" '
-                'style="color-scheme: dark"></iframe>',
-                sanitize=False,
-            ).style("margin-left: 0px;")
-            ui.tooltip("Check Platform Status")
+        status_url = get_status_page_url(settings().api_root)
+        if status_url:
+            with ui.row().style("padding: 0"):
+                ui.html(
+                    f'<iframe id="betterstack" src="{status_url}/badge?theme=dark" '
+                    'width="250" height="30" frameborder="0" scrolling="no" '
+                    'style="color-scheme: dark"></iframe>',
+                    sanitize=False,
+                ).style("margin-left: 0px;")
+                ui.tooltip("Check Platform Status")
         ui.space()
         with ui.row():
             flavor = " (native)" if getattr(sys, "frozen", False) else ""

tests/aignostics/gui/__init__.py

@@ -0,0 +1 @@
+"""Tests for GUI module."""

tests/aignostics/gui/frame_test.py

@@ -0,0 +1,71 @@
+"""Tests for GUI frame module."""
+
+import pytest
+
+from aignostics.gui._frame import get_status_page_url
+from aignostics.platform import (
+    API_ROOT_DEV,
+    API_ROOT_PRODUCTION,
+    API_ROOT_STAGING,
+    API_ROOT_TEST,
+)
+
+
+@pytest.mark.unit
+def test_get_status_page_url_production(record_property) -> None:
+    """Test that production environment returns correct status page URL.
+
+    Args:
+        record_property: pytest record_property fixture
+    """
+    record_property("tested-item-id", "SPEC-GUI-FRAME")
+    url = get_status_page_url(API_ROOT_PRODUCTION)
+    assert url == "https://status.platform.aignostics.com"
+
+
+@pytest.mark.unit
+def test_get_status_page_url_staging(record_property) -> None:
+    """Test that staging environment returns correct status page URL.
+
+    Args:
+        record_property: pytest record_property fixture
+    """
+    record_property("tested-item-id", "SPEC-GUI-FRAME")
+    url = get_status_page_url(API_ROOT_STAGING)
+    assert url == "https://status.platform-staging.aignostics.com"
+
+
+@pytest.mark.unit
+def test_get_status_page_url_dev(record_property) -> None:
+    """Test that dev environment returns None (no status page).
+
+    Args:
+        record_property: pytest record_property fixture
+    """
+    record_property("tested-item-id", "SPEC-GUI-FRAME")
+    url = get_status_page_url(API_ROOT_DEV)
+    assert url is None
+
+
+@pytest.mark.unit
+def test_get_status_page_url_test(record_property) -> None:
+    """Test that test environment returns None (no status page).
+
+    Args:
+        record_property: pytest record_property fixture
+    """
+    record_property("tested-item-id", "SPEC-GUI-FRAME")
+    url = get_status_page_url(API_ROOT_TEST)
+    assert url is None
+
+
+@pytest.mark.unit
+def test_get_status_page_url_unknown(record_property) -> None:
+    """Test that unknown environment returns None (no status page).
+
+    Args:
+        record_property: pytest record_property fixture
+    """
+    record_property("tested-item-id", "SPEC-GUI-FRAME")
+    url = get_status_page_url("https://custom.example.com")
+    assert url is None