Add test_services.test_async_error_executing_query

Signed-off-by: David Rapan <[email protected]>

Author: davidrapan

homeassistant/components/sql/util.py

@@ -215,7 +215,9 @@ async def _async_validate_and_get_session_maker_for_db_url(
     """Validate the db_url and return a async session maker."""
     try:
         maker = async_scoped_session(
-            async_sessionmaker(bind=create_async_engine(db_url)),
+            async_sessionmaker(
+                bind=create_async_engine(db_url, future=True), future=True
+            ),
             scopefunc=asyncio.current_task,
         )
         # Run a dummy query just to test the db_url

tests/components/sql/test_services.py

@@ -7,6 +7,7 @@
 from unittest.mock import patch
 
 import pytest
+from sqlalchemy.exc import SQLAlchemyError
 import voluptuous as vol
 from voluptuous import MultipleInvalid
 
@@ -55,6 +56,31 @@ async def test_query_service_recorder_db(
     }
 
 
+async def test_async_error_executing_query(
+    hass: HomeAssistant,
+    caplog: pytest.LogCaptureFixture,
+) -> None:
+    """Test the query service."""
+    db_url = "sqlite+aiosqlite:///"
+
+    await async_setup_component(hass, DOMAIN, {})
+    await hass.async_block_till_done()
+
+    with patch(
+        "sqlalchemy.ext.asyncio.session.AsyncSession.execute",
+        side_effect=SQLAlchemyError("Error executing query"),
+    ):
+        await hass.services.async_call(
+            DOMAIN,
+            SERVICE_QUERY,
+            {"query": "SELECT name, age FROM users ORDER BY age", "db_url": db_url},
+            blocking=True,
+            return_response=True,
+        )
+
+    assert "Error executing query" in caplog.text
+
+
 async def test_query_service_external_db(hass: HomeAssistant, tmp_path: Path) -> None:
     """Test the query service with an external database via db_url."""
     db_path = tmp_path / "test.db"