From 040e120101ffbdd93dd686928d88a60849b68dfe Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Fri, 27 May 2022 07:32:26 -1000
Subject: [PATCH] Fix recorder system health when the db_url is lacking a
 hostname (#72612)

---
 .../recorder/system_health/__init__.py        |  5 ++-
 .../components/recorder/test_system_health.py | 31 +++++++++++++++++++
 2 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/recorder/system_health/__init__.py b/homeassistant/components/recorder/system_health/__init__.py
index 8ba68a1649b..c4bf2c3bb89 100644
--- a/homeassistant/components/recorder/system_health/__init__.py
+++ b/homeassistant/components/recorder/system_health/__init__.py
@@ -2,8 +2,7 @@
 from __future__ import annotations
 
 from typing import Any
-
-from yarl import URL
+from urllib.parse import urlparse
 
 from homeassistant.components import system_health
 from homeassistant.components.recorder.core import Recorder
@@ -60,7 +59,7 @@ async def system_health_info(hass: HomeAssistant) -> dict[str, Any]:
     instance = get_instance(hass)
 
     run_history = instance.run_history
-    database_name = URL(instance.db_url).path.lstrip("/")
+    database_name = urlparse(instance.db_url).path.lstrip("/")
     db_engine_info = _async_get_db_engine_info(instance)
     db_stats: dict[str, Any] = {}
 
diff --git a/tests/components/recorder/test_system_health.py b/tests/components/recorder/test_system_health.py
index 80997b9df36..b465ee89ebe 100644
--- a/tests/components/recorder/test_system_health.py
+++ b/tests/components/recorder/test_system_health.py
@@ -53,6 +53,37 @@ async def test_recorder_system_health_alternate_dbms(hass, recorder_mock, dialec
     }
 
 
+@pytest.mark.parametrize(
+    "dialect_name", [SupportedDialect.MYSQL, SupportedDialect.POSTGRESQL]
+)
+async def test_recorder_system_health_db_url_missing_host(
+    hass, recorder_mock, dialect_name
+):
+    """Test recorder system health with a db_url without a hostname."""
+    assert await async_setup_component(hass, "system_health", {})
+    await async_wait_recording_done(hass)
+
+    instance = get_instance(hass)
+    with patch(
+        "homeassistant.components.recorder.core.Recorder.dialect_name", dialect_name
+    ), patch.object(
+        instance,
+        "db_url",
+        "postgresql://homeassistant:blabla@/home_assistant?host=/config/socket",
+    ), patch(
+        "sqlalchemy.orm.session.Session.execute",
+        return_value=Mock(first=Mock(return_value=("1048576",))),
+    ):
+        info = await get_system_health_info(hass, "recorder")
+    assert info == {
+        "current_recorder_run": instance.run_history.current.start,
+        "oldest_recorder_run": instance.run_history.first.start,
+        "estimated_db_size": "1.00 MiB",
+        "database_engine": dialect_name.value,
+        "database_version": ANY,
+    }
+
+
 async def test_recorder_system_health_crashed_recorder_runs_table(
     hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT
 ):