test: validate PostgreSQL database connection

- Add test to verify model catalog pod uses PostgreSQL database
- Enhance get_model_catalog_pod utility with configurable label selector

Author: fege

tests/model_registry/model_catalog/test_default_model_catalog.py

@@ -11,14 +11,15 @@
 from ocp_resources.config_map import ConfigMap
 from ocp_resources.route import Route
 from ocp_resources.service import Service
-from tests.model_registry.model_catalog.constants import DEFAULT_CATALOG_ID
+from tests.model_registry.model_catalog.constants import DEFAULT_CATALOG_ID, CATALOG_CONTAINER
 from tests.model_registry.model_catalog.utils import (
     validate_model_catalog_enabled,
     execute_get_command,
     validate_model_catalog_resource,
     validate_default_catalog,
     get_validate_default_model_catalog_source,
     extract_schema_fields,
+    get_model_catalog_pod,
 )
 from tests.model_registry.utils import get_rest_headers
 from utilities.user_utils import UserTestSession
@@ -100,6 +101,20 @@ def test_model_catalog_resources_exists(
     def test_operator_pod_enabled_model_catalog(self: Self, model_registry_operator_pod: Pod):
         assert validate_model_catalog_enabled(pod=model_registry_operator_pod)
 
+    def test_model_catalog_uses_postgres(self: Self, admin_client: DynamicClient, model_registry_namespace: str):
+        """
+        Validate that model catalog pod is using PostgreSQL database
+        """
+        model_catalog_pods = get_model_catalog_pod(
+            client=admin_client,
+            model_registry_namespace=model_registry_namespace,
+            label_selector="app.kubernetes.io/name=model-catalog",
+        )
+        assert len(model_catalog_pods) == 1
+        model_catalog_pod = model_catalog_pods[0]
+        model_catalog_pod_log = model_catalog_pod.log(container=CATALOG_CONTAINER)
+        assert "Successfully connected to PostgreSQL database" in model_catalog_pod_log
+
 
 @pytest.mark.parametrize(
     "user_token_for_api_calls,",

tests/model_registry/utils.py

@@ -649,10 +649,10 @@ def validate_mlmd_removal_in_model_registry_pod_log(
     assert not errors, f"Log validation failed with error(s): {errors}"
 
 
-def get_model_catalog_pod(client: DynamicClient, model_registry_namespace: str) -> list[Pod]:
-    return list(
-        Pod.get(namespace=model_registry_namespace, label_selector="component=model-catalog", dyn_client=client)
-    )
+def get_model_catalog_pod(
+    client: DynamicClient, model_registry_namespace: str, label_selector: str = "component=model-catalog"
+) -> list[Pod]:
+    return list(Pod.get(namespace=model_registry_namespace, label_selector=label_selector, dyn_client=client))
 
 
 def get_rest_headers(token: str) -> dict[str, str]: