diff --git a/homeassistant/components/recorder/__init__.py b/homeassistant/components/recorder/__init__.py index a0d2f1c8702..1ff7c886c35 100644 --- a/homeassistant/components/recorder/__init__.py +++ b/homeassistant/components/recorder/__init__.py @@ -79,6 +79,7 @@ from .const import ( ) from .executor import DBInterruptibleThreadPoolExecutor from .models import ( + SCHEMA_VERSION, Base, Events, StateAttributes, @@ -634,6 +635,7 @@ class Recorder(threading.Thread): self.entity_filter = entity_filter self.exclude_t = exclude_t + self.schema_version = 0 self._commits_without_expire = 0 self._old_states: dict[str, States] = {} self._state_attributes_ids: LRU = LRU(STATE_ATTRIBUTES_ID_CACHE_SIZE) @@ -973,6 +975,8 @@ class Recorder(threading.Thread): self.hass.add_job(self.async_connection_failed) return + self.schema_version = current_version + schema_is_current = migration.schema_is_current(current_version) if schema_is_current: self._setup_run() @@ -994,6 +998,7 @@ class Recorder(threading.Thread): # with startup which is also cpu intensive if not schema_is_current: if self._migrate_schema_and_setup_run(current_version): + self.schema_version = SCHEMA_VERSION if not self._event_listener: # If the schema migration takes so long that the end # queue watcher safety kicks in because MAX_QUEUE_BACKLOG diff --git a/homeassistant/components/recorder/history.py b/homeassistant/components/recorder/history.py index 24fe21f101c..d221ced3a84 100644 --- a/homeassistant/components/recorder/history.py +++ b/homeassistant/components/recorder/history.py @@ -116,7 +116,7 @@ def query_and_join_attributes( # If we in the process of migrating schema we do # not want to join the state_attributes table as we # do not know if it will be there yet - if recorder.get_instance(hass).migration_in_progress: + if recorder.get_instance(hass).schema_version < 25: return QUERY_STATES_PRE_SCHEMA_25, False # Finally if no migration is in progress and no_attributes # was not requested, we query both attributes columns and @@ -146,7 +146,7 @@ def bake_query_and_join_attributes( # If we in the process of migrating schema we do # not want to join the state_attributes table as we # do not know if it will be there yet - if recorder.get_instance(hass).migration_in_progress: + if recorder.get_instance(hass).schema_version < 25: if include_last_updated: return ( bakery(lambda session: session.query(*QUERY_STATES_PRE_SCHEMA_25)), diff --git a/tests/components/recorder/test_history.py b/tests/components/recorder/test_history.py index 8f22447cd8e..c4952d8355b 100644 --- a/tests/components/recorder/test_history.py +++ b/tests/components/recorder/test_history.py @@ -628,7 +628,7 @@ async def test_state_changes_during_period_query_during_migration_to_schema_25( conn.execute(text("drop table state_attributes;")) conn.commit() - with patch.object(instance, "migration_in_progress", True): + with patch.object(instance, "schema_version", 24): no_attributes = True hist = history.state_changes_during_period( hass, start, end, entity_id, no_attributes, include_start_time_state=False @@ -674,7 +674,7 @@ async def test_get_states_query_during_migration_to_schema_25( conn.execute(text("drop table state_attributes;")) conn.commit() - with patch.object(instance, "migration_in_progress", True): + with patch.object(instance, "schema_version", 24): no_attributes = True hist = await _async_get_states( hass, end, [entity_id], no_attributes=no_attributes @@ -723,7 +723,7 @@ async def test_get_states_query_during_migration_to_schema_25_multiple_entities( conn.execute(text("drop table state_attributes;")) conn.commit() - with patch.object(instance, "migration_in_progress", True): + with patch.object(instance, "schema_version", 24): no_attributes = True hist = await _async_get_states( hass, end, entity_ids, no_attributes=no_attributes diff --git a/tests/components/recorder/test_init.py b/tests/components/recorder/test_init.py index 77bd3992e72..0a0194f4ddf 100644 --- a/tests/components/recorder/test_init.py +++ b/tests/components/recorder/test_init.py @@ -31,6 +31,7 @@ from homeassistant.components.recorder import ( ) from homeassistant.components.recorder.const import DATA_INSTANCE from homeassistant.components.recorder.models import ( + SCHEMA_VERSION, Events, RecorderRuns, StateAttributes, @@ -438,6 +439,12 @@ def _state_empty_context(hass, entity_id): return state +def test_setup_without_migration(hass_recorder): + """Verify the schema version without a migration.""" + hass = hass_recorder() + assert recorder.get_instance(hass).schema_version == SCHEMA_VERSION + + # pylint: disable=redefined-outer-name,invalid-name def test_saving_state_include_domains(hass_recorder): """Test saving and restoring a state.""" diff --git a/tests/components/recorder/test_migrate.py b/tests/components/recorder/test_migrate.py index 6b963941263..464c3a8b4a3 100644 --- a/tests/components/recorder/test_migrate.py +++ b/tests/components/recorder/test_migrate.py @@ -22,7 +22,11 @@ from homeassistant.bootstrap import async_setup_component from homeassistant.components import persistent_notification as pn, recorder from homeassistant.components.recorder import migration, models from homeassistant.components.recorder.const import DATA_INSTANCE -from homeassistant.components.recorder.models import RecorderRuns, States +from homeassistant.components.recorder.models import ( + SCHEMA_VERSION, + RecorderRuns, + States, +) from homeassistant.components.recorder.util import session_scope import homeassistant.util.dt as dt_util @@ -79,6 +83,7 @@ async def test_migration_in_progress(hass): await async_wait_recording_done(hass) assert recorder.util.async_migration_in_progress(hass) is False + assert recorder.get_instance(hass).schema_version == SCHEMA_VERSION async def test_database_migration_failed(hass):