Fix sql integration no recorder (#111263)

pull/111319/head
G Johansson 2024-02-24 22:40:49 +01:00 committed by GitHub
parent cd46cc6e80
commit 5073842514
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 16 additions and 2 deletions

View File

@ -184,10 +184,14 @@ async def async_setup_sensor(
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up the SQL sensor."""
instance = get_instance(hass)
try:
instance = get_instance(hass)
except KeyError: # No recorder loaded
uses_recorder_db = False
else:
uses_recorder_db = db_url == instance.db_url
sessmaker: scoped_session | None
sql_data = _async_get_or_init_domain_data(hass)
uses_recorder_db = db_url == instance.db_url
use_database_executor = False
if uses_recorder_db and instance.dialect_name == SupportedDialect.SQLITE:
use_database_executor = True

View File

@ -635,3 +635,13 @@ async def test_query_recover_from_rollback(
state = hass.states.get("sensor.select_value_sql_query")
assert state.state == "5"
assert state.attributes.get("value") == 5
async def test_setup_without_recorder(hass: HomeAssistant) -> None:
"""Test the SQL sensor without recorder."""
assert await async_setup_component(hass, DOMAIN, YAML_CONFIG)
await hass.async_block_till_done()
state = hass.states.get("sensor.get_value")
assert state.state == "5"