jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-26 02:07:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Separate recorder test fixtures disabling context id migration (#125324)

Browse Source 
* Separate recorder test fixtures disabling context id migration

* Fix test
This commit is contained in:
Erik Montnemery 2024-09-09 08:16:30 +02:00 committed by GitHub
parent d88487e30b
commit 2a1df2063d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 24 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/components/recorder/test_migration_from_schema_32.py
View File

@ -118,7 +118,7 @@ def db_schema_32():
yield yield
@pytest.mark.parametrize("enable_migrate_context_ids", [True]) @pytest.mark.parametrize("enable_migrate_event_context_ids", [True])
@pytest.mark.usefixtures("db_schema_32") @pytest.mark.usefixtures("db_schema_32")
async def test_migrate_events_context_ids( async def test_migrate_events_context_ids(
hass: HomeAssistant, recorder_mock: Recorder hass: HomeAssistant, recorder_mock: Recorder
@ -338,7 +338,7 @@ async def test_migrate_events_context_ids(
assert get_index_by_name(session, "events", "ix_events_context_id") is None assert get_index_by_name(session, "events", "ix_events_context_id") is None
@pytest.mark.parametrize("enable_migrate_context_ids", [True]) @pytest.mark.parametrize("enable_migrate_state_context_ids", [True])
@pytest.mark.usefixtures("db_schema_32") @pytest.mark.usefixtures("db_schema_32")
async def test_migrate_states_context_ids( async def test_migrate_states_context_ids(
hass: HomeAssistant, recorder_mock: Recorder hass: HomeAssistant, recorder_mock: Recorder

6
tests/components/recorder/test_migration_run_time_migrations_remember.py
View File

@ -72,7 +72,7 @@ def _create_engine_test(*args, **kwargs):
return engine return engine
@pytest.mark.parametrize("enable_migrate_context_ids", [True]) @pytest.mark.parametrize("enable_migrate_state_context_ids", [True])
@pytest.mark.parametrize("persistent_database", [True]) @pytest.mark.parametrize("persistent_database", [True])
@pytest.mark.usefixtures("hass_storage") # Prevent test hass from writing to storage @pytest.mark.usefixtures("hass_storage") # Prevent test hass from writing to storage
async def test_migration_changes_prevent_trying_to_migrate_again( async def test_migration_changes_prevent_trying_to_migrate_again(
@ -173,4 +173,6 @@ async def test_migration_changes_prevent_trying_to_migrate_again(
await hass.async_stop() await hass.async_stop()
for task in tasks: for task in tasks:
assert not isinstance(task, MigrationTask) if not isinstance(task, MigrationTask):
continue
assert not isinstance(task.migrator, migration.StatesContextIDMigration)

3
tests/components/recorder/test_v32_migration.py
View File

@ -60,7 +60,8 @@ def _create_engine_test(schema_module: str) -> Callable:
return _create_engine_test return _create_engine_test
@pytest.mark.parametrize("enable_migrate_context_ids", [True]) @pytest.mark.parametrize("enable_migrate_event_context_ids", [True])
@pytest.mark.parametrize("enable_migrate_state_context_ids", [True])
@pytest.mark.parametrize("enable_migrate_event_type_ids", [True]) @pytest.mark.parametrize("enable_migrate_event_type_ids", [True])
@pytest.mark.parametrize("enable_migrate_entity_ids", [True]) @pytest.mark.parametrize("enable_migrate_entity_ids", [True])
@pytest.mark.parametrize("persistent_database", [True]) @pytest.mark.parametrize("persistent_database", [True])

21
tests/conftest.py
View File

@ -1293,11 +1293,21 @@ def enable_nightly_purge() -> bool:
@pytest.fixture @pytest.fixture
def enable_migrate_context_ids() -> bool: def enable_migrate_event_context_ids() -> bool:
"""Fixture to control enabling of recorder's context id migration. """Fixture to control enabling of recorder's context id migration.
To enable context id migration, tests can be marked with: To enable context id migration, tests can be marked with:
@pytest.mark.parametrize("enable_migrate_context_ids", [True]) @pytest.mark.parametrize("enable_migrate_event_context_ids", [True])
"""
return False
@pytest.fixture
def enable_migrate_state_context_ids() -> bool:
"""Fixture to control enabling of recorder's context id migration.
To enable context id migration, tests can be marked with:
@pytest.mark.parametrize("enable_migrate_state_context_ids", [True])
""" """
return False return False
@ -1465,7 +1475,8 @@ async def async_test_recorder(
enable_statistics: bool, enable_statistics: bool,
enable_missing_statistics: bool, enable_missing_statistics: bool,
enable_schema_validation: bool, enable_schema_validation: bool,
enable_migrate_context_ids: bool, enable_migrate_event_context_ids: bool,
enable_migrate_state_context_ids: bool,
enable_migrate_event_type_ids: bool, enable_migrate_event_type_ids: bool,
enable_migrate_entity_ids: bool, enable_migrate_entity_ids: bool,
enable_migrate_event_ids: bool, enable_migrate_event_ids: bool,
@ -1527,12 +1538,12 @@ async def async_test_recorder(
) )
migrate_states_context_ids = ( migrate_states_context_ids = (
migration.StatesContextIDMigration.migrate_data migration.StatesContextIDMigration.migrate_data
if enable_migrate_context_ids if enable_migrate_state_context_ids
else None else None
) )
migrate_events_context_ids = ( migrate_events_context_ids = (
migration.EventsContextIDMigration.migrate_data migration.EventsContextIDMigration.migrate_data
if enable_migrate_context_ids if enable_migrate_event_context_ids
else None else None
) )
migrate_event_type_ids = ( migrate_event_type_ids = (