From e3813d4134aca938b6e9d35803c449695967a354 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Mon, 12 Feb 2024 01:30:38 -0600 Subject: [PATCH] Migrate blink to use async_update_entry to alter config entries (#110303) --- tests/components/blink/test_init.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/components/blink/test_init.py b/tests/components/blink/test_init.py index 32c424aae60..5521303ac8b 100644 --- a/tests/components/blink/test_init.py +++ b/tests/components/blink/test_init.py @@ -86,7 +86,7 @@ async def test_migrate_V0( ) -> None: """Test migration script version 0.""" - mock_config_entry.version = 0 + hass.config_entries.async_update_entry(mock_config_entry, version=0) mock_config_entry.add_to_hass(hass) assert await hass.config_entries.async_setup(mock_config_entry.entry_id) @@ -105,8 +105,11 @@ async def test_migrate( ) -> None: """Test migration scripts.""" - mock_config_entry.version = version - mock_config_entry.data = {**mock_config_entry.data, "login_response": "Blah"} + hass.config_entries.async_update_entry( + mock_config_entry, + version=version, + data={**mock_config_entry.data, "login_response": "Blah"}, + ) mock_config_entry.add_to_hass(hass) assert not await hass.config_entries.async_setup(mock_config_entry.entry_id)